Show More
@@ -0,0 +1,41 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | """ | |||
|
3 | vcs | |||
|
4 | ~~~ | |||
|
5 | ||||
|
6 | Various version Control System (vcs) management abstraction layer for | |||
|
7 | Python. | |||
|
8 | ||||
|
9 | :created_on: Apr 8, 2010 | |||
|
10 | :copyright: (c) 2010-2011 by Marcin Kuzminski, Lukasz Balcerzak. | |||
|
11 | """ | |||
|
12 | ||||
|
13 | VERSION = (0, 2, 3, 'dev') | |||
|
14 | ||||
|
15 | __version__ = '.'.join((str(each) for each in VERSION[:4])) | |||
|
16 | ||||
|
17 | __all__ = [ | |||
|
18 | 'get_version', 'get_repo', 'get_backend', | |||
|
19 | 'VCSError', 'RepositoryError', 'ChangesetError'] | |||
|
20 | ||||
|
21 | import sys | |||
|
22 | from rhodecode.lib.vcs.backends import get_repo, get_backend | |||
|
23 | from rhodecode.lib.vcs.exceptions import VCSError, RepositoryError, ChangesetError | |||
|
24 | ||||
|
25 | ||||
|
26 | def get_version(): | |||
|
27 | """ | |||
|
28 | Returns shorter version (digit parts only) as string. | |||
|
29 | """ | |||
|
30 | return '.'.join((str(each) for each in VERSION[:3])) | |||
|
31 | ||||
|
32 | def main(argv=None): | |||
|
33 | if argv is None: | |||
|
34 | argv = sys.argv | |||
|
35 | from rhodecode.lib.vcs.cli import ExecutionManager | |||
|
36 | manager = ExecutionManager(argv) | |||
|
37 | manager.execute() | |||
|
38 | return 0 | |||
|
39 | ||||
|
40 | if __name__ == '__main__': | |||
|
41 | sys.exit(main(sys.argv)) |
@@ -0,0 +1,63 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | """ | |||
|
3 | vcs.backends | |||
|
4 | ~~~~~~~~~~~~ | |||
|
5 | ||||
|
6 | Main package for scm backends | |||
|
7 | ||||
|
8 | :created_on: Apr 8, 2010 | |||
|
9 | :copyright: (c) 2010-2011 by Marcin Kuzminski, Lukasz Balcerzak. | |||
|
10 | """ | |||
|
11 | import os | |||
|
12 | from pprint import pformat | |||
|
13 | from rhodecode.lib.vcs.conf import settings | |||
|
14 | from rhodecode.lib.vcs.exceptions import VCSError | |||
|
15 | from rhodecode.lib.vcs.utils.helpers import get_scm | |||
|
16 | from rhodecode.lib.vcs.utils.paths import abspath | |||
|
17 | from rhodecode.lib.vcs.utils.imports import import_class | |||
|
18 | ||||
|
19 | ||||
|
20 | def get_repo(path=None, alias=None, create=False): | |||
|
21 | """ | |||
|
22 | Returns ``Repository`` object of type linked with given ``alias`` at | |||
|
23 | the specified ``path``. If ``alias`` is not given it will try to guess it | |||
|
24 | using get_scm method | |||
|
25 | """ | |||
|
26 | if create: | |||
|
27 | if not (path or alias): | |||
|
28 | raise TypeError("If create is specified, we need path and scm type") | |||
|
29 | return get_backend(alias)(path, create=True) | |||
|
30 | if path is None: | |||
|
31 | path = abspath(os.path.curdir) | |||
|
32 | try: | |||
|
33 | scm, path = get_scm(path, search_recursively=True) | |||
|
34 | path = abspath(path) | |||
|
35 | alias = scm | |||
|
36 | except VCSError: | |||
|
37 | raise VCSError("No scm found at %s" % path) | |||
|
38 | if alias is None: | |||
|
39 | alias = get_scm(path)[0] | |||
|
40 | ||||
|
41 | backend = get_backend(alias) | |||
|
42 | repo = backend(path, create=create) | |||
|
43 | return repo | |||
|
44 | ||||
|
45 | ||||
|
46 | def get_backend(alias): | |||
|
47 | """ | |||
|
48 | Returns ``Repository`` class identified by the given alias or raises | |||
|
49 | VCSError if alias is not recognized or backend class cannot be imported. | |||
|
50 | """ | |||
|
51 | if alias not in settings.BACKENDS: | |||
|
52 | raise VCSError("Given alias '%s' is not recognized! Allowed aliases:\n" | |||
|
53 | "%s" % (alias, pformat(settings.BACKENDS.keys()))) | |||
|
54 | backend_path = settings.BACKENDS[alias] | |||
|
55 | klass = import_class(backend_path) | |||
|
56 | return klass | |||
|
57 | ||||
|
58 | ||||
|
59 | def get_supported_backends(): | |||
|
60 | """ | |||
|
61 | Returns list of aliases of supported backends. | |||
|
62 | """ | |||
|
63 | return settings.BACKENDS.keys() |
This diff has been collapsed as it changes many lines, (911 lines changed) Show them Hide them | |||||
@@ -0,0 +1,911 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | """ | |||
|
3 | vcs.backends.base | |||
|
4 | ~~~~~~~~~~~~~~~~~ | |||
|
5 | ||||
|
6 | Base for all available scm backends | |||
|
7 | ||||
|
8 | :created_on: Apr 8, 2010 | |||
|
9 | :copyright: (c) 2010-2011 by Marcin Kuzminski, Lukasz Balcerzak. | |||
|
10 | """ | |||
|
11 | ||||
|
12 | ||||
|
13 | from itertools import chain | |||
|
14 | from rhodecode.lib.vcs.utils import author_name, author_email | |||
|
15 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |||
|
16 | from rhodecode.lib.vcs.utils.helpers import get_dict_for_attrs | |||
|
17 | from rhodecode.lib.vcs.conf import settings | |||
|
18 | ||||
|
19 | from rhodecode.lib.vcs.exceptions import ChangesetError, EmptyRepositoryError, \ | |||
|
20 | NodeAlreadyAddedError, NodeAlreadyChangedError, NodeAlreadyExistsError, \ | |||
|
21 | NodeAlreadyRemovedError, NodeDoesNotExistError, NodeNotChangedError, \ | |||
|
22 | RepositoryError | |||
|
23 | ||||
|
24 | ||||
|
25 | class BaseRepository(object): | |||
|
26 | """ | |||
|
27 | Base Repository for final backends | |||
|
28 | ||||
|
29 | **Attributes** | |||
|
30 | ||||
|
31 | ``DEFAULT_BRANCH_NAME`` | |||
|
32 | name of default branch (i.e. "trunk" for svn, "master" for git etc. | |||
|
33 | ||||
|
34 | ``scm`` | |||
|
35 | alias of scm, i.e. *git* or *hg* | |||
|
36 | ||||
|
37 | ``repo`` | |||
|
38 | object from external api | |||
|
39 | ||||
|
40 | ``revisions`` | |||
|
41 | list of all available revisions' ids, in ascending order | |||
|
42 | ||||
|
43 | ``changesets`` | |||
|
44 | storage dict caching returned changesets | |||
|
45 | ||||
|
46 | ``path`` | |||
|
47 | absolute path to the repository | |||
|
48 | ||||
|
49 | ``branches`` | |||
|
50 | branches as list of changesets | |||
|
51 | ||||
|
52 | ``tags`` | |||
|
53 | tags as list of changesets | |||
|
54 | """ | |||
|
55 | scm = None | |||
|
56 | DEFAULT_BRANCH_NAME = None | |||
|
57 | EMPTY_CHANGESET = '0' * 40 | |||
|
58 | ||||
|
59 | def __init__(self, repo_path, create=False, **kwargs): | |||
|
60 | """ | |||
|
61 | Initializes repository. Raises RepositoryError if repository could | |||
|
62 | not be find at the given ``repo_path`` or directory at ``repo_path`` | |||
|
63 | exists and ``create`` is set to True. | |||
|
64 | ||||
|
65 | :param repo_path: local path of the repository | |||
|
66 | :param create=False: if set to True, would try to craete repository. | |||
|
67 | :param src_url=None: if set, should be proper url from which repository | |||
|
68 | would be cloned; requires ``create`` parameter to be set to True - | |||
|
69 | raises RepositoryError if src_url is set and create evaluates to | |||
|
70 | False | |||
|
71 | """ | |||
|
72 | raise NotImplementedError | |||
|
73 | ||||
|
74 | def __str__(self): | |||
|
75 | return '<%s at %s>' % (self.__class__.__name__, self.path) | |||
|
76 | ||||
|
77 | def __repr__(self): | |||
|
78 | return self.__str__() | |||
|
79 | ||||
|
80 | def __len__(self): | |||
|
81 | return self.count() | |||
|
82 | ||||
|
83 | @LazyProperty | |||
|
84 | def alias(self): | |||
|
85 | for k, v in settings.BACKENDS.items(): | |||
|
86 | if v.split('.')[-1] == str(self.__class__.__name__): | |||
|
87 | return k | |||
|
88 | ||||
|
89 | @LazyProperty | |||
|
90 | def name(self): | |||
|
91 | raise NotImplementedError | |||
|
92 | ||||
|
93 | @LazyProperty | |||
|
94 | def owner(self): | |||
|
95 | raise NotImplementedError | |||
|
96 | ||||
|
97 | @LazyProperty | |||
|
98 | def description(self): | |||
|
99 | raise NotImplementedError | |||
|
100 | ||||
|
101 | @LazyProperty | |||
|
102 | def size(self): | |||
|
103 | """ | |||
|
104 | Returns combined size in bytes for all repository files | |||
|
105 | """ | |||
|
106 | ||||
|
107 | size = 0 | |||
|
108 | try: | |||
|
109 | tip = self.get_changeset() | |||
|
110 | for topnode, dirs, files in tip.walk('/'): | |||
|
111 | for f in files: | |||
|
112 | size += tip.get_file_size(f.path) | |||
|
113 | for dir in dirs: | |||
|
114 | for f in files: | |||
|
115 | size += tip.get_file_size(f.path) | |||
|
116 | ||||
|
117 | except RepositoryError, e: | |||
|
118 | pass | |||
|
119 | return size | |||
|
120 | ||||
|
121 | def is_valid(self): | |||
|
122 | """ | |||
|
123 | Validates repository. | |||
|
124 | """ | |||
|
125 | raise NotImplementedError | |||
|
126 | ||||
|
127 | def get_last_change(self): | |||
|
128 | self.get_changesets() | |||
|
129 | ||||
|
130 | #========================================================================== | |||
|
131 | # CHANGESETS | |||
|
132 | #========================================================================== | |||
|
133 | ||||
|
134 | def get_changeset(self, revision=None): | |||
|
135 | """ | |||
|
136 | Returns instance of ``Changeset`` class. If ``revision`` is None, most | |||
|
137 | recent changeset is returned. | |||
|
138 | ||||
|
139 | :raises ``EmptyRepositoryError``: if there are no revisions | |||
|
140 | """ | |||
|
141 | raise NotImplementedError | |||
|
142 | ||||
|
143 | def __iter__(self): | |||
|
144 | """ | |||
|
145 | Allows Repository objects to be iterated. | |||
|
146 | ||||
|
147 | *Requires* implementation of ``__getitem__`` method. | |||
|
148 | """ | |||
|
149 | for revision in self.revisions: | |||
|
150 | yield self.get_changeset(revision) | |||
|
151 | ||||
|
152 | def get_changesets(self, start=None, end=None, start_date=None, | |||
|
153 | end_date=None, branch_name=None, reverse=False): | |||
|
154 | """ | |||
|
155 | Returns iterator of ``MercurialChangeset`` objects from start to end | |||
|
156 | not inclusive This should behave just like a list, ie. end is not | |||
|
157 | inclusive | |||
|
158 | ||||
|
159 | :param start: None or str | |||
|
160 | :param end: None or str | |||
|
161 | :param start_date: | |||
|
162 | :param end_date: | |||
|
163 | :param branch_name: | |||
|
164 | :param reversed: | |||
|
165 | """ | |||
|
166 | raise NotImplementedError | |||
|
167 | ||||
|
168 | def __getslice__(self, i, j): | |||
|
169 | """ | |||
|
170 | Returns a iterator of sliced repository | |||
|
171 | """ | |||
|
172 | for rev in self.revisions[i:j]: | |||
|
173 | yield self.get_changeset(rev) | |||
|
174 | ||||
|
175 | def __getitem__(self, key): | |||
|
176 | return self.get_changeset(key) | |||
|
177 | ||||
|
178 | def count(self): | |||
|
179 | return len(self.revisions) | |||
|
180 | ||||
|
181 | def tag(self, name, user, revision=None, message=None, date=None, **opts): | |||
|
182 | """ | |||
|
183 | Creates and returns a tag for the given ``revision``. | |||
|
184 | ||||
|
185 | :param name: name for new tag | |||
|
186 | :param user: full username, i.e.: "Joe Doe <joe.doe@example.com>" | |||
|
187 | :param revision: changeset id for which new tag would be created | |||
|
188 | :param message: message of the tag's commit | |||
|
189 | :param date: date of tag's commit | |||
|
190 | ||||
|
191 | :raises TagAlreadyExistError: if tag with same name already exists | |||
|
192 | """ | |||
|
193 | raise NotImplementedError | |||
|
194 | ||||
|
195 | def remove_tag(self, name, user, message=None, date=None): | |||
|
196 | """ | |||
|
197 | Removes tag with the given ``name``. | |||
|
198 | ||||
|
199 | :param name: name of the tag to be removed | |||
|
200 | :param user: full username, i.e.: "Joe Doe <joe.doe@example.com>" | |||
|
201 | :param message: message of the tag's removal commit | |||
|
202 | :param date: date of tag's removal commit | |||
|
203 | ||||
|
204 | :raises TagDoesNotExistError: if tag with given name does not exists | |||
|
205 | """ | |||
|
206 | raise NotImplementedError | |||
|
207 | ||||
|
208 | def get_diff(self, rev1, rev2, path=None, ignore_whitespace=False, | |||
|
209 | context=3): | |||
|
210 | """ | |||
|
211 | Returns (git like) *diff*, as plain text. Shows changes introduced by | |||
|
212 | ``rev2`` since ``rev1``. | |||
|
213 | ||||
|
214 | :param rev1: Entry point from which diff is shown. Can be | |||
|
215 | ``self.EMPTY_CHANGESET`` - in this case, patch showing all | |||
|
216 | the changes since empty state of the repository until ``rev2`` | |||
|
217 | :param rev2: Until which revision changes should be shown. | |||
|
218 | :param ignore_whitespace: If set to ``True``, would not show whitespace | |||
|
219 | changes. Defaults to ``False``. | |||
|
220 | :param context: How many lines before/after changed lines should be | |||
|
221 | shown. Defaults to ``3``. | |||
|
222 | """ | |||
|
223 | raise NotImplementedError | |||
|
224 | ||||
|
225 | # ========== # | |||
|
226 | # COMMIT API # | |||
|
227 | # ========== # | |||
|
228 | ||||
|
229 | @LazyProperty | |||
|
230 | def in_memory_changeset(self): | |||
|
231 | """ | |||
|
232 | Returns ``InMemoryChangeset`` object for this repository. | |||
|
233 | """ | |||
|
234 | raise NotImplementedError | |||
|
235 | ||||
|
236 | def add(self, filenode, **kwargs): | |||
|
237 | """ | |||
|
238 | Commit api function that will add given ``FileNode`` into this | |||
|
239 | repository. | |||
|
240 | ||||
|
241 | :raises ``NodeAlreadyExistsError``: if there is a file with same path | |||
|
242 | already in repository | |||
|
243 | :raises ``NodeAlreadyAddedError``: if given node is already marked as | |||
|
244 | *added* | |||
|
245 | """ | |||
|
246 | raise NotImplementedError | |||
|
247 | ||||
|
248 | def remove(self, filenode, **kwargs): | |||
|
249 | """ | |||
|
250 | Commit api function that will remove given ``FileNode`` into this | |||
|
251 | repository. | |||
|
252 | ||||
|
253 | :raises ``EmptyRepositoryError``: if there are no changesets yet | |||
|
254 | :raises ``NodeDoesNotExistError``: if there is no file with given path | |||
|
255 | """ | |||
|
256 | raise NotImplementedError | |||
|
257 | ||||
|
258 | def commit(self, message, **kwargs): | |||
|
259 | """ | |||
|
260 | Persists current changes made on this repository and returns newly | |||
|
261 | created changeset. | |||
|
262 | ||||
|
263 | :raises ``NothingChangedError``: if no changes has been made | |||
|
264 | """ | |||
|
265 | raise NotImplementedError | |||
|
266 | ||||
|
267 | def get_state(self): | |||
|
268 | """ | |||
|
269 | Returns dictionary with ``added``, ``changed`` and ``removed`` lists | |||
|
270 | containing ``FileNode`` objects. | |||
|
271 | """ | |||
|
272 | raise NotImplementedError | |||
|
273 | ||||
|
274 | def get_config_value(self, section, name, config_file=None): | |||
|
275 | """ | |||
|
276 | Returns configuration value for a given [``section``] and ``name``. | |||
|
277 | ||||
|
278 | :param section: Section we want to retrieve value from | |||
|
279 | :param name: Name of configuration we want to retrieve | |||
|
280 | :param config_file: A path to file which should be used to retrieve | |||
|
281 | configuration from (might also be a list of file paths) | |||
|
282 | """ | |||
|
283 | raise NotImplementedError | |||
|
284 | ||||
|
285 | def get_user_name(self, config_file=None): | |||
|
286 | """ | |||
|
287 | Returns user's name from global configuration file. | |||
|
288 | ||||
|
289 | :param config_file: A path to file which should be used to retrieve | |||
|
290 | configuration from (might also be a list of file paths) | |||
|
291 | """ | |||
|
292 | raise NotImplementedError | |||
|
293 | ||||
|
294 | def get_user_email(self, config_file=None): | |||
|
295 | """ | |||
|
296 | Returns user's email from global configuration file. | |||
|
297 | ||||
|
298 | :param config_file: A path to file which should be used to retrieve | |||
|
299 | configuration from (might also be a list of file paths) | |||
|
300 | """ | |||
|
301 | raise NotImplementedError | |||
|
302 | ||||
|
303 | # =========== # | |||
|
304 | # WORKDIR API # | |||
|
305 | # =========== # | |||
|
306 | ||||
|
307 | @LazyProperty | |||
|
308 | def workdir(self): | |||
|
309 | """ | |||
|
310 | Returns ``Workdir`` instance for this repository. | |||
|
311 | """ | |||
|
312 | raise NotImplementedError | |||
|
313 | ||||
|
314 | ||||
|
315 | class BaseChangeset(object): | |||
|
316 | """ | |||
|
317 | Each backend should implement it's changeset representation. | |||
|
318 | ||||
|
319 | **Attributes** | |||
|
320 | ||||
|
321 | ``repository`` | |||
|
322 | repository object within which changeset exists | |||
|
323 | ||||
|
324 | ``id`` | |||
|
325 | may be ``raw_id`` or i.e. for mercurial's tip just ``tip`` | |||
|
326 | ||||
|
327 | ``raw_id`` | |||
|
328 | raw changeset representation (i.e. full 40 length sha for git | |||
|
329 | backend) | |||
|
330 | ||||
|
331 | ``short_id`` | |||
|
332 | shortened (if apply) version of ``raw_id``; it would be simple | |||
|
333 | shortcut for ``raw_id[:12]`` for git/mercurial backends or same | |||
|
334 | as ``raw_id`` for subversion | |||
|
335 | ||||
|
336 | ``revision`` | |||
|
337 | revision number as integer | |||
|
338 | ||||
|
339 | ``files`` | |||
|
340 | list of ``FileNode`` (``Node`` with NodeKind.FILE) objects | |||
|
341 | ||||
|
342 | ``dirs`` | |||
|
343 | list of ``DirNode`` (``Node`` with NodeKind.DIR) objects | |||
|
344 | ||||
|
345 | ``nodes`` | |||
|
346 | combined list of ``Node`` objects | |||
|
347 | ||||
|
348 | ``author`` | |||
|
349 | author of the changeset, as unicode | |||
|
350 | ||||
|
351 | ``message`` | |||
|
352 | message of the changeset, as unicode | |||
|
353 | ||||
|
354 | ``parents`` | |||
|
355 | list of parent changesets | |||
|
356 | ||||
|
357 | ``last`` | |||
|
358 | ``True`` if this is last changeset in repository, ``False`` | |||
|
359 | otherwise; trying to access this attribute while there is no | |||
|
360 | changesets would raise ``EmptyRepositoryError`` | |||
|
361 | """ | |||
|
362 | def __str__(self): | |||
|
363 | return '<%s at %s:%s>' % (self.__class__.__name__, self.revision, | |||
|
364 | self.short_id) | |||
|
365 | ||||
|
366 | def __repr__(self): | |||
|
367 | return self.__str__() | |||
|
368 | ||||
|
369 | def __unicode__(self): | |||
|
370 | return u'%s:%s' % (self.revision, self.short_id) | |||
|
371 | ||||
|
372 | def __eq__(self, other): | |||
|
373 | return self.raw_id == other.raw_id | |||
|
374 | ||||
|
375 | @LazyProperty | |||
|
376 | def last(self): | |||
|
377 | if self.repository is None: | |||
|
378 | raise ChangesetError("Cannot check if it's most recent revision") | |||
|
379 | return self.raw_id == self.repository.revisions[-1] | |||
|
380 | ||||
|
381 | @LazyProperty | |||
|
382 | def parents(self): | |||
|
383 | """ | |||
|
384 | Returns list of parents changesets. | |||
|
385 | """ | |||
|
386 | raise NotImplementedError | |||
|
387 | ||||
|
388 | @LazyProperty | |||
|
389 | def id(self): | |||
|
390 | """ | |||
|
391 | Returns string identifying this changeset. | |||
|
392 | """ | |||
|
393 | raise NotImplementedError | |||
|
394 | ||||
|
395 | @LazyProperty | |||
|
396 | def raw_id(self): | |||
|
397 | """ | |||
|
398 | Returns raw string identifying this changeset. | |||
|
399 | """ | |||
|
400 | raise NotImplementedError | |||
|
401 | ||||
|
402 | @LazyProperty | |||
|
403 | def short_id(self): | |||
|
404 | """ | |||
|
405 | Returns shortened version of ``raw_id`` attribute, as string, | |||
|
406 | identifying this changeset, useful for web representation. | |||
|
407 | """ | |||
|
408 | raise NotImplementedError | |||
|
409 | ||||
|
410 | @LazyProperty | |||
|
411 | def revision(self): | |||
|
412 | """ | |||
|
413 | Returns integer identifying this changeset. | |||
|
414 | ||||
|
415 | """ | |||
|
416 | raise NotImplementedError | |||
|
417 | ||||
|
418 | @LazyProperty | |||
|
419 | def author(self): | |||
|
420 | """ | |||
|
421 | Returns Author for given commit | |||
|
422 | """ | |||
|
423 | ||||
|
424 | raise NotImplementedError | |||
|
425 | ||||
|
426 | @LazyProperty | |||
|
427 | def author_name(self): | |||
|
428 | """ | |||
|
429 | Returns Author name for given commit | |||
|
430 | """ | |||
|
431 | ||||
|
432 | return author_name(self.author) | |||
|
433 | ||||
|
434 | @LazyProperty | |||
|
435 | def author_email(self): | |||
|
436 | """ | |||
|
437 | Returns Author email address for given commit | |||
|
438 | """ | |||
|
439 | ||||
|
440 | return author_email(self.author) | |||
|
441 | ||||
|
442 | def get_file_mode(self, path): | |||
|
443 | """ | |||
|
444 | Returns stat mode of the file at the given ``path``. | |||
|
445 | """ | |||
|
446 | raise NotImplementedError | |||
|
447 | ||||
|
448 | def get_file_content(self, path): | |||
|
449 | """ | |||
|
450 | Returns content of the file at the given ``path``. | |||
|
451 | """ | |||
|
452 | raise NotImplementedError | |||
|
453 | ||||
|
454 | def get_file_size(self, path): | |||
|
455 | """ | |||
|
456 | Returns size of the file at the given ``path``. | |||
|
457 | """ | |||
|
458 | raise NotImplementedError | |||
|
459 | ||||
|
460 | def get_file_changeset(self, path): | |||
|
461 | """ | |||
|
462 | Returns last commit of the file at the given ``path``. | |||
|
463 | """ | |||
|
464 | raise NotImplementedError | |||
|
465 | ||||
|
466 | def get_file_history(self, path): | |||
|
467 | """ | |||
|
468 | Returns history of file as reversed list of ``Changeset`` objects for | |||
|
469 | which file at given ``path`` has been modified. | |||
|
470 | """ | |||
|
471 | raise NotImplementedError | |||
|
472 | ||||
|
473 | def get_nodes(self, path): | |||
|
474 | """ | |||
|
475 | Returns combined ``DirNode`` and ``FileNode`` objects list representing | |||
|
476 | state of changeset at the given ``path``. | |||
|
477 | ||||
|
478 | :raises ``ChangesetError``: if node at the given ``path`` is not | |||
|
479 | instance of ``DirNode`` | |||
|
480 | """ | |||
|
481 | raise NotImplementedError | |||
|
482 | ||||
|
483 | def get_node(self, path): | |||
|
484 | """ | |||
|
485 | Returns ``Node`` object from the given ``path``. | |||
|
486 | ||||
|
487 | :raises ``NodeDoesNotExistError``: if there is no node at the given | |||
|
488 | ``path`` | |||
|
489 | """ | |||
|
490 | raise NotImplementedError | |||
|
491 | ||||
|
492 | def fill_archive(self, stream=None, kind='tgz', prefix=None): | |||
|
493 | """ | |||
|
494 | Fills up given stream. | |||
|
495 | ||||
|
496 | :param stream: file like object. | |||
|
497 | :param kind: one of following: ``zip``, ``tar``, ``tgz`` | |||
|
498 | or ``tbz2``. Default: ``tgz``. | |||
|
499 | :param prefix: name of root directory in archive. | |||
|
500 | Default is repository name and changeset's raw_id joined with dash. | |||
|
501 | ||||
|
502 | repo-tip.<kind> | |||
|
503 | """ | |||
|
504 | ||||
|
505 | raise NotImplementedError | |||
|
506 | ||||
|
507 | def get_chunked_archive(self, **kwargs): | |||
|
508 | """ | |||
|
509 | Returns iterable archive. Tiny wrapper around ``fill_archive`` method. | |||
|
510 | ||||
|
511 | :param chunk_size: extra parameter which controls size of returned | |||
|
512 | chunks. Default:8k. | |||
|
513 | """ | |||
|
514 | ||||
|
515 | chunk_size = kwargs.pop('chunk_size', 8192) | |||
|
516 | stream = kwargs.get('stream') | |||
|
517 | self.fill_archive(**kwargs) | |||
|
518 | while True: | |||
|
519 | data = stream.read(chunk_size) | |||
|
520 | if not data: | |||
|
521 | break | |||
|
522 | yield data | |||
|
523 | ||||
|
524 | @LazyProperty | |||
|
525 | def root(self): | |||
|
526 | """ | |||
|
527 | Returns ``RootNode`` object for this changeset. | |||
|
528 | """ | |||
|
529 | return self.get_node('') | |||
|
530 | ||||
|
531 | def next(self, branch=None): | |||
|
532 | """ | |||
|
533 | Returns next changeset from current, if branch is gives it will return | |||
|
534 | next changeset belonging to this branch | |||
|
535 | ||||
|
536 | :param branch: show changesets within the given named branch | |||
|
537 | """ | |||
|
538 | raise NotImplementedError | |||
|
539 | ||||
|
540 | def prev(self, branch=None): | |||
|
541 | """ | |||
|
542 | Returns previous changeset from current, if branch is gives it will | |||
|
543 | return previous changeset belonging to this branch | |||
|
544 | ||||
|
545 | :param branch: show changesets within the given named branch | |||
|
546 | """ | |||
|
547 | raise NotImplementedError | |||
|
548 | ||||
|
549 | @LazyProperty | |||
|
550 | def added(self): | |||
|
551 | """ | |||
|
552 | Returns list of added ``FileNode`` objects. | |||
|
553 | """ | |||
|
554 | raise NotImplementedError | |||
|
555 | ||||
|
556 | @LazyProperty | |||
|
557 | def changed(self): | |||
|
558 | """ | |||
|
559 | Returns list of modified ``FileNode`` objects. | |||
|
560 | """ | |||
|
561 | raise NotImplementedError | |||
|
562 | ||||
|
563 | @LazyProperty | |||
|
564 | def removed(self): | |||
|
565 | """ | |||
|
566 | Returns list of removed ``FileNode`` objects. | |||
|
567 | """ | |||
|
568 | raise NotImplementedError | |||
|
569 | ||||
|
570 | @LazyProperty | |||
|
571 | def size(self): | |||
|
572 | """ | |||
|
573 | Returns total number of bytes from contents of all filenodes. | |||
|
574 | """ | |||
|
575 | return sum((node.size for node in self.get_filenodes_generator())) | |||
|
576 | ||||
|
577 | def walk(self, topurl=''): | |||
|
578 | """ | |||
|
579 | Similar to os.walk method. Insted of filesystem it walks through | |||
|
580 | changeset starting at given ``topurl``. Returns generator of tuples | |||
|
581 | (topnode, dirnodes, filenodes). | |||
|
582 | """ | |||
|
583 | topnode = self.get_node(topurl) | |||
|
584 | yield (topnode, topnode.dirs, topnode.files) | |||
|
585 | for dirnode in topnode.dirs: | |||
|
586 | for tup in self.walk(dirnode.path): | |||
|
587 | yield tup | |||
|
588 | ||||
|
589 | def get_filenodes_generator(self): | |||
|
590 | """ | |||
|
591 | Returns generator that yields *all* file nodes. | |||
|
592 | """ | |||
|
593 | for topnode, dirs, files in self.walk(): | |||
|
594 | for node in files: | |||
|
595 | yield node | |||
|
596 | ||||
|
597 | def as_dict(self): | |||
|
598 | """ | |||
|
599 | Returns dictionary with changeset's attributes and their values. | |||
|
600 | """ | |||
|
601 | data = get_dict_for_attrs(self, ['id', 'raw_id', 'short_id', | |||
|
602 | 'revision', 'date', 'message']) | |||
|
603 | data['author'] = {'name': self.author_name, 'email': self.author_email} | |||
|
604 | data['added'] = [node.path for node in self.added] | |||
|
605 | data['changed'] = [node.path for node in self.changed] | |||
|
606 | data['removed'] = [node.path for node in self.removed] | |||
|
607 | return data | |||
|
608 | ||||
|
609 | ||||
|
610 | class BaseWorkdir(object): | |||
|
611 | """ | |||
|
612 | Working directory representation of single repository. | |||
|
613 | ||||
|
614 | :attribute: repository: repository object of working directory | |||
|
615 | """ | |||
|
616 | ||||
|
617 | def __init__(self, repository): | |||
|
618 | self.repository = repository | |||
|
619 | ||||
|
620 | def get_branch(self): | |||
|
621 | """ | |||
|
622 | Returns name of current branch. | |||
|
623 | """ | |||
|
624 | raise NotImplementedError | |||
|
625 | ||||
|
626 | def get_changeset(self): | |||
|
627 | """ | |||
|
628 | Returns current changeset. | |||
|
629 | """ | |||
|
630 | raise NotImplementedError | |||
|
631 | ||||
|
632 | def get_added(self): | |||
|
633 | """ | |||
|
634 | Returns list of ``FileNode`` objects marked as *new* in working | |||
|
635 | directory. | |||
|
636 | """ | |||
|
637 | raise NotImplementedError | |||
|
638 | ||||
|
639 | def get_changed(self): | |||
|
640 | """ | |||
|
641 | Returns list of ``FileNode`` objects *changed* in working directory. | |||
|
642 | """ | |||
|
643 | raise NotImplementedError | |||
|
644 | ||||
|
645 | def get_removed(self): | |||
|
646 | """ | |||
|
647 | Returns list of ``RemovedFileNode`` objects marked as *removed* in | |||
|
648 | working directory. | |||
|
649 | """ | |||
|
650 | raise NotImplementedError | |||
|
651 | ||||
|
652 | def get_untracked(self): | |||
|
653 | """ | |||
|
654 | Returns list of ``FileNode`` objects which are present within working | |||
|
655 | directory however are not tracked by repository. | |||
|
656 | """ | |||
|
657 | raise NotImplementedError | |||
|
658 | ||||
|
659 | def get_status(self): | |||
|
660 | """ | |||
|
661 | Returns dict with ``added``, ``changed``, ``removed`` and ``untracked`` | |||
|
662 | lists. | |||
|
663 | """ | |||
|
664 | raise NotImplementedError | |||
|
665 | ||||
|
666 | def commit(self, message, **kwargs): | |||
|
667 | """ | |||
|
668 | Commits local (from working directory) changes and returns newly | |||
|
669 | created | |||
|
670 | ``Changeset``. Updates repository's ``revisions`` list. | |||
|
671 | ||||
|
672 | :raises ``CommitError``: if any error occurs while committing | |||
|
673 | """ | |||
|
674 | raise NotImplementedError | |||
|
675 | ||||
|
676 | def update(self, revision=None): | |||
|
677 | """ | |||
|
678 | Fetches content of the given revision and populates it within working | |||
|
679 | directory. | |||
|
680 | """ | |||
|
681 | raise NotImplementedError | |||
|
682 | ||||
|
683 | def checkout_branch(self, branch=None): | |||
|
684 | """ | |||
|
685 | Checks out ``branch`` or the backend's default branch. | |||
|
686 | ||||
|
687 | Raises ``BranchDoesNotExistError`` if the branch does not exist. | |||
|
688 | """ | |||
|
689 | raise NotImplementedError | |||
|
690 | ||||
|
691 | ||||
|
692 | class BaseInMemoryChangeset(object): | |||
|
693 | """ | |||
|
694 | Represents differences between repository's state (most recent head) and | |||
|
695 | changes made *in place*. | |||
|
696 | ||||
|
697 | **Attributes** | |||
|
698 | ||||
|
699 | ``repository`` | |||
|
700 | repository object for this in-memory-changeset | |||
|
701 | ||||
|
702 | ``added`` | |||
|
703 | list of ``FileNode`` objects marked as *added* | |||
|
704 | ||||
|
705 | ``changed`` | |||
|
706 | list of ``FileNode`` objects marked as *changed* | |||
|
707 | ||||
|
708 | ``removed`` | |||
|
709 | list of ``FileNode`` or ``RemovedFileNode`` objects marked to be | |||
|
710 | *removed* | |||
|
711 | ||||
|
712 | ``parents`` | |||
|
713 | list of ``Changeset`` representing parents of in-memory changeset. | |||
|
714 | Should always be 2-element sequence. | |||
|
715 | ||||
|
716 | """ | |||
|
717 | ||||
|
718 | def __init__(self, repository): | |||
|
719 | self.repository = repository | |||
|
720 | self.added = [] | |||
|
721 | self.changed = [] | |||
|
722 | self.removed = [] | |||
|
723 | self.parents = [] | |||
|
724 | ||||
|
725 | def add(self, *filenodes): | |||
|
726 | """ | |||
|
727 | Marks given ``FileNode`` objects as *to be committed*. | |||
|
728 | ||||
|
729 | :raises ``NodeAlreadyExistsError``: if node with same path exists at | |||
|
730 | latest changeset | |||
|
731 | :raises ``NodeAlreadyAddedError``: if node with same path is already | |||
|
732 | marked as *added* | |||
|
733 | """ | |||
|
734 | # Check if not already marked as *added* first | |||
|
735 | for node in filenodes: | |||
|
736 | if node.path in (n.path for n in self.added): | |||
|
737 | raise NodeAlreadyAddedError("Such FileNode %s is already " | |||
|
738 | "marked for addition" % node.path) | |||
|
739 | for node in filenodes: | |||
|
740 | self.added.append(node) | |||
|
741 | ||||
|
742 | def change(self, *filenodes): | |||
|
743 | """ | |||
|
744 | Marks given ``FileNode`` objects to be *changed* in next commit. | |||
|
745 | ||||
|
746 | :raises ``EmptyRepositoryError``: if there are no changesets yet | |||
|
747 | :raises ``NodeAlreadyExistsError``: if node with same path is already | |||
|
748 | marked to be *changed* | |||
|
749 | :raises ``NodeAlreadyRemovedError``: if node with same path is already | |||
|
750 | marked to be *removed* | |||
|
751 | :raises ``NodeDoesNotExistError``: if node doesn't exist in latest | |||
|
752 | changeset | |||
|
753 | :raises ``NodeNotChangedError``: if node hasn't really be changed | |||
|
754 | """ | |||
|
755 | for node in filenodes: | |||
|
756 | if node.path in (n.path for n in self.removed): | |||
|
757 | raise NodeAlreadyRemovedError("Node at %s is already marked " | |||
|
758 | "as removed" % node.path) | |||
|
759 | try: | |||
|
760 | self.repository.get_changeset() | |||
|
761 | except EmptyRepositoryError: | |||
|
762 | raise EmptyRepositoryError("Nothing to change - try to *add* new " | |||
|
763 | "nodes rather than changing them") | |||
|
764 | for node in filenodes: | |||
|
765 | if node.path in (n.path for n in self.changed): | |||
|
766 | raise NodeAlreadyChangedError("Node at '%s' is already " | |||
|
767 | "marked as changed" % node.path) | |||
|
768 | self.changed.append(node) | |||
|
769 | ||||
|
770 | def remove(self, *filenodes): | |||
|
771 | """ | |||
|
772 | Marks given ``FileNode`` (or ``RemovedFileNode``) objects to be | |||
|
773 | *removed* in next commit. | |||
|
774 | ||||
|
775 | :raises ``NodeAlreadyRemovedError``: if node has been already marked to | |||
|
776 | be *removed* | |||
|
777 | :raises ``NodeAlreadyChangedError``: if node has been already marked to | |||
|
778 | be *changed* | |||
|
779 | """ | |||
|
780 | for node in filenodes: | |||
|
781 | if node.path in (n.path for n in self.removed): | |||
|
782 | raise NodeAlreadyRemovedError("Node is already marked to " | |||
|
783 | "for removal at %s" % node.path) | |||
|
784 | if node.path in (n.path for n in self.changed): | |||
|
785 | raise NodeAlreadyChangedError("Node is already marked to " | |||
|
786 | "be changed at %s" % node.path) | |||
|
787 | # We only mark node as *removed* - real removal is done by | |||
|
788 | # commit method | |||
|
789 | self.removed.append(node) | |||
|
790 | ||||
|
791 | def reset(self): | |||
|
792 | """ | |||
|
793 | Resets this instance to initial state (cleans ``added``, ``changed`` | |||
|
794 | and ``removed`` lists). | |||
|
795 | """ | |||
|
796 | self.added = [] | |||
|
797 | self.changed = [] | |||
|
798 | self.removed = [] | |||
|
799 | self.parents = [] | |||
|
800 | ||||
|
801 | def get_ipaths(self): | |||
|
802 | """ | |||
|
803 | Returns generator of paths from nodes marked as added, changed or | |||
|
804 | removed. | |||
|
805 | """ | |||
|
806 | for node in chain(self.added, self.changed, self.removed): | |||
|
807 | yield node.path | |||
|
808 | ||||
|
809 | def get_paths(self): | |||
|
810 | """ | |||
|
811 | Returns list of paths from nodes marked as added, changed or removed. | |||
|
812 | """ | |||
|
813 | return list(self.get_ipaths()) | |||
|
814 | ||||
|
815 | def check_integrity(self, parents=None): | |||
|
816 | """ | |||
|
817 | Checks in-memory changeset's integrity. Also, sets parents if not | |||
|
818 | already set. | |||
|
819 | ||||
|
820 | :raises CommitError: if any error occurs (i.e. | |||
|
821 | ``NodeDoesNotExistError``). | |||
|
822 | """ | |||
|
823 | if not self.parents: | |||
|
824 | parents = parents or [] | |||
|
825 | if len(parents) == 0: | |||
|
826 | try: | |||
|
827 | parents = [self.repository.get_changeset(), None] | |||
|
828 | except EmptyRepositoryError: | |||
|
829 | parents = [None, None] | |||
|
830 | elif len(parents) == 1: | |||
|
831 | parents += [None] | |||
|
832 | self.parents = parents | |||
|
833 | ||||
|
834 | # Local parents, only if not None | |||
|
835 | parents = [p for p in self.parents if p] | |||
|
836 | ||||
|
837 | # Check nodes marked as added | |||
|
838 | for p in parents: | |||
|
839 | for node in self.added: | |||
|
840 | try: | |||
|
841 | p.get_node(node.path) | |||
|
842 | except NodeDoesNotExistError: | |||
|
843 | pass | |||
|
844 | else: | |||
|
845 | raise NodeAlreadyExistsError("Node at %s already exists " | |||
|
846 | "at %s" % (node.path, p)) | |||
|
847 | ||||
|
848 | # Check nodes marked as changed | |||
|
849 | missing = set(self.changed) | |||
|
850 | not_changed = set(self.changed) | |||
|
851 | if self.changed and not parents: | |||
|
852 | raise NodeDoesNotExistError(str(self.changed[0].path)) | |||
|
853 | for p in parents: | |||
|
854 | for node in self.changed: | |||
|
855 | try: | |||
|
856 | old = p.get_node(node.path) | |||
|
857 | missing.remove(node) | |||
|
858 | if old.content != node.content: | |||
|
859 | not_changed.remove(node) | |||
|
860 | except NodeDoesNotExistError: | |||
|
861 | pass | |||
|
862 | if self.changed and missing: | |||
|
863 | raise NodeDoesNotExistError("Node at %s is missing " | |||
|
864 | "(parents: %s)" % (node.path, parents)) | |||
|
865 | ||||
|
866 | if self.changed and not_changed: | |||
|
867 | raise NodeNotChangedError("Node at %s wasn't actually changed " | |||
|
868 | "since parents' changesets: %s" % (not_changed.pop().path, | |||
|
869 | parents) | |||
|
870 | ) | |||
|
871 | ||||
|
872 | # Check nodes marked as removed | |||
|
873 | if self.removed and not parents: | |||
|
874 | raise NodeDoesNotExistError("Cannot remove node at %s as there " | |||
|
875 | "were no parents specified" % self.removed[0].path) | |||
|
876 | really_removed = set() | |||
|
877 | for p in parents: | |||
|
878 | for node in self.removed: | |||
|
879 | try: | |||
|
880 | p.get_node(node.path) | |||
|
881 | really_removed.add(node) | |||
|
882 | except ChangesetError: | |||
|
883 | pass | |||
|
884 | not_removed = set(self.removed) - really_removed | |||
|
885 | if not_removed: | |||
|
886 | raise NodeDoesNotExistError("Cannot remove node at %s from " | |||
|
887 | "following parents: %s" % (not_removed[0], parents)) | |||
|
888 | ||||
|
889 | def commit(self, message, author, parents=None, branch=None, date=None, | |||
|
890 | **kwargs): | |||
|
891 | """ | |||
|
892 | Performs in-memory commit (doesn't check workdir in any way) and | |||
|
893 | returns newly created ``Changeset``. Updates repository's | |||
|
894 | ``revisions``. | |||
|
895 | ||||
|
896 | .. note:: | |||
|
897 | While overriding this method each backend's should call | |||
|
898 | ``self.check_integrity(parents)`` in the first place. | |||
|
899 | ||||
|
900 | :param message: message of the commit | |||
|
901 | :param author: full username, i.e. "Joe Doe <joe.doe@example.com>" | |||
|
902 | :param parents: single parent or sequence of parents from which commit | |||
|
903 | would be derieved | |||
|
904 | :param date: ``datetime.datetime`` instance. Defaults to | |||
|
905 | ``datetime.datetime.now()``. | |||
|
906 | :param branch: branch name, as string. If none given, default backend's | |||
|
907 | branch would be used. | |||
|
908 | ||||
|
909 | :raises ``CommitError``: if any error occurs while committing | |||
|
910 | """ | |||
|
911 | raise NotImplementedError |
@@ -0,0 +1,9 b'' | |||||
|
1 | from .repository import GitRepository | |||
|
2 | from .changeset import GitChangeset | |||
|
3 | from .inmemory import GitInMemoryChangeset | |||
|
4 | from .workdir import GitWorkdir | |||
|
5 | ||||
|
6 | ||||
|
7 | __all__ = [ | |||
|
8 | 'GitRepository', 'GitChangeset', 'GitInMemoryChangeset', 'GitWorkdir', | |||
|
9 | ] |
@@ -0,0 +1,450 b'' | |||||
|
1 | import re | |||
|
2 | from itertools import chain | |||
|
3 | from dulwich import objects | |||
|
4 | from subprocess import Popen, PIPE | |||
|
5 | from rhodecode.lib.vcs.conf import settings | |||
|
6 | from rhodecode.lib.vcs.exceptions import RepositoryError | |||
|
7 | from rhodecode.lib.vcs.exceptions import ChangesetError | |||
|
8 | from rhodecode.lib.vcs.exceptions import NodeDoesNotExistError | |||
|
9 | from rhodecode.lib.vcs.exceptions import VCSError | |||
|
10 | from rhodecode.lib.vcs.exceptions import ChangesetDoesNotExistError | |||
|
11 | from rhodecode.lib.vcs.exceptions import ImproperArchiveTypeError | |||
|
12 | from rhodecode.lib.vcs.backends.base import BaseChangeset | |||
|
13 | from rhodecode.lib.vcs.nodes import FileNode, DirNode, NodeKind, RootNode, RemovedFileNode | |||
|
14 | from rhodecode.lib.vcs.utils import safe_unicode | |||
|
15 | from rhodecode.lib.vcs.utils import date_fromtimestamp | |||
|
16 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |||
|
17 | ||||
|
18 | ||||
|
19 | class GitChangeset(BaseChangeset): | |||
|
20 | """ | |||
|
21 | Represents state of the repository at single revision. | |||
|
22 | """ | |||
|
23 | ||||
|
24 | def __init__(self, repository, revision): | |||
|
25 | self._stat_modes = {} | |||
|
26 | self.repository = repository | |||
|
27 | self.raw_id = revision | |||
|
28 | self.revision = repository.revisions.index(revision) | |||
|
29 | ||||
|
30 | self.short_id = self.raw_id[:12] | |||
|
31 | self.id = self.raw_id | |||
|
32 | try: | |||
|
33 | commit = self.repository._repo.get_object(self.raw_id) | |||
|
34 | except KeyError: | |||
|
35 | raise RepositoryError("Cannot get object with id %s" % self.raw_id) | |||
|
36 | self._commit = commit | |||
|
37 | self._tree_id = commit.tree | |||
|
38 | ||||
|
39 | try: | |||
|
40 | self.message = safe_unicode(commit.message[:-1]) | |||
|
41 | # Always strip last eol | |||
|
42 | except UnicodeDecodeError: | |||
|
43 | self.message = commit.message[:-1].decode(commit.encoding | |||
|
44 | or 'utf-8') | |||
|
45 | #self.branch = None | |||
|
46 | self.tags = [] | |||
|
47 | #tree = self.repository.get_object(self._tree_id) | |||
|
48 | self.nodes = {} | |||
|
49 | self._paths = {} | |||
|
50 | ||||
|
51 | @LazyProperty | |||
|
52 | def author(self): | |||
|
53 | return safe_unicode(self._commit.committer) | |||
|
54 | ||||
|
55 | @LazyProperty | |||
|
56 | def date(self): | |||
|
57 | return date_fromtimestamp(self._commit.commit_time, | |||
|
58 | self._commit.commit_timezone) | |||
|
59 | ||||
|
60 | @LazyProperty | |||
|
61 | def status(self): | |||
|
62 | """ | |||
|
63 | Returns modified, added, removed, deleted files for current changeset | |||
|
64 | """ | |||
|
65 | return self.changed, self.added, self.removed | |||
|
66 | ||||
|
67 | @LazyProperty | |||
|
68 | def branch(self): | |||
|
69 | # TODO: Cache as we walk (id <-> branch name mapping) | |||
|
70 | refs = self.repository._repo.get_refs() | |||
|
71 | heads = [(key[len('refs/heads/'):], val) for key, val in refs.items() | |||
|
72 | if key.startswith('refs/heads/')] | |||
|
73 | ||||
|
74 | for name, id in heads: | |||
|
75 | walker = self.repository._repo.object_store.get_graph_walker([id]) | |||
|
76 | while True: | |||
|
77 | id = walker.next() | |||
|
78 | if not id: | |||
|
79 | break | |||
|
80 | if id == self.id: | |||
|
81 | return safe_unicode(name) | |||
|
82 | raise ChangesetError("This should not happen... Have you manually " | |||
|
83 | "change id of the changeset?") | |||
|
84 | ||||
|
85 | def _fix_path(self, path): | |||
|
86 | """ | |||
|
87 | Paths are stored without trailing slash so we need to get rid off it if | |||
|
88 | needed. | |||
|
89 | """ | |||
|
90 | if path.endswith('/'): | |||
|
91 | path = path.rstrip('/') | |||
|
92 | return path | |||
|
93 | ||||
|
94 | def _get_id_for_path(self, path): | |||
|
95 | # FIXME: Please, spare a couple of minutes and make those codes cleaner; | |||
|
96 | if not path in self._paths: | |||
|
97 | path = path.strip('/') | |||
|
98 | # set root tree | |||
|
99 | tree = self.repository._repo[self._commit.tree] | |||
|
100 | if path == '': | |||
|
101 | self._paths[''] = tree.id | |||
|
102 | return tree.id | |||
|
103 | splitted = path.split('/') | |||
|
104 | dirs, name = splitted[:-1], splitted[-1] | |||
|
105 | curdir = '' | |||
|
106 | for dir in dirs: | |||
|
107 | if curdir: | |||
|
108 | curdir = '/'.join((curdir, dir)) | |||
|
109 | else: | |||
|
110 | curdir = dir | |||
|
111 | #if curdir in self._paths: | |||
|
112 | ## This path have been already traversed | |||
|
113 | ## Update tree and continue | |||
|
114 | #tree = self.repository._repo[self._paths[curdir]] | |||
|
115 | #continue | |||
|
116 | dir_id = None | |||
|
117 | for item, stat, id in tree.iteritems(): | |||
|
118 | if curdir: | |||
|
119 | item_path = '/'.join((curdir, item)) | |||
|
120 | else: | |||
|
121 | item_path = item | |||
|
122 | self._paths[item_path] = id | |||
|
123 | self._stat_modes[item_path] = stat | |||
|
124 | if dir == item: | |||
|
125 | dir_id = id | |||
|
126 | if dir_id: | |||
|
127 | # Update tree | |||
|
128 | tree = self.repository._repo[dir_id] | |||
|
129 | if not isinstance(tree, objects.Tree): | |||
|
130 | raise ChangesetError('%s is not a directory' % curdir) | |||
|
131 | else: | |||
|
132 | raise ChangesetError('%s have not been found' % curdir) | |||
|
133 | for item, stat, id in tree.iteritems(): | |||
|
134 | if curdir: | |||
|
135 | name = '/'.join((curdir, item)) | |||
|
136 | else: | |||
|
137 | name = item | |||
|
138 | self._paths[name] = id | |||
|
139 | self._stat_modes[name] = stat | |||
|
140 | if not path in self._paths: | |||
|
141 | raise NodeDoesNotExistError("There is no file nor directory " | |||
|
142 | "at the given path %r at revision %r" | |||
|
143 | % (path, self.short_id)) | |||
|
144 | return self._paths[path] | |||
|
145 | ||||
|
146 | def _get_kind(self, path): | |||
|
147 | id = self._get_id_for_path(path) | |||
|
148 | obj = self.repository._repo[id] | |||
|
149 | if isinstance(obj, objects.Blob): | |||
|
150 | return NodeKind.FILE | |||
|
151 | elif isinstance(obj, objects.Tree): | |||
|
152 | return NodeKind.DIR | |||
|
153 | ||||
|
154 | def _get_file_nodes(self): | |||
|
155 | return chain(*(t[2] for t in self.walk())) | |||
|
156 | ||||
|
157 | @LazyProperty | |||
|
158 | def parents(self): | |||
|
159 | """ | |||
|
160 | Returns list of parents changesets. | |||
|
161 | """ | |||
|
162 | return [self.repository.get_changeset(parent) | |||
|
163 | for parent in self._commit.parents] | |||
|
164 | ||||
|
165 | def next(self, branch=None): | |||
|
166 | ||||
|
167 | if branch and self.branch != branch: | |||
|
168 | raise VCSError('Branch option used on changeset not belonging ' | |||
|
169 | 'to that branch') | |||
|
170 | ||||
|
171 | def _next(changeset, branch): | |||
|
172 | try: | |||
|
173 | next_ = changeset.revision + 1 | |||
|
174 | next_rev = changeset.repository.revisions[next_] | |||
|
175 | except IndexError: | |||
|
176 | raise ChangesetDoesNotExistError | |||
|
177 | cs = changeset.repository.get_changeset(next_rev) | |||
|
178 | ||||
|
179 | if branch and branch != cs.branch: | |||
|
180 | return _next(cs, branch) | |||
|
181 | ||||
|
182 | return cs | |||
|
183 | ||||
|
184 | return _next(self, branch) | |||
|
185 | ||||
|
186 | def prev(self, branch=None): | |||
|
187 | if branch and self.branch != branch: | |||
|
188 | raise VCSError('Branch option used on changeset not belonging ' | |||
|
189 | 'to that branch') | |||
|
190 | ||||
|
191 | def _prev(changeset, branch): | |||
|
192 | try: | |||
|
193 | prev_ = changeset.revision - 1 | |||
|
194 | if prev_ < 0: | |||
|
195 | raise IndexError | |||
|
196 | prev_rev = changeset.repository.revisions[prev_] | |||
|
197 | except IndexError: | |||
|
198 | raise ChangesetDoesNotExistError | |||
|
199 | ||||
|
200 | cs = changeset.repository.get_changeset(prev_rev) | |||
|
201 | ||||
|
202 | if branch and branch != cs.branch: | |||
|
203 | return _prev(cs, branch) | |||
|
204 | ||||
|
205 | return cs | |||
|
206 | ||||
|
207 | return _prev(self, branch) | |||
|
208 | ||||
|
209 | def get_file_mode(self, path): | |||
|
210 | """ | |||
|
211 | Returns stat mode of the file at the given ``path``. | |||
|
212 | """ | |||
|
213 | # ensure path is traversed | |||
|
214 | self._get_id_for_path(path) | |||
|
215 | return self._stat_modes[path] | |||
|
216 | ||||
|
217 | def get_file_content(self, path): | |||
|
218 | """ | |||
|
219 | Returns content of the file at given ``path``. | |||
|
220 | """ | |||
|
221 | id = self._get_id_for_path(path) | |||
|
222 | blob = self.repository._repo[id] | |||
|
223 | return blob.as_pretty_string() | |||
|
224 | ||||
|
225 | def get_file_size(self, path): | |||
|
226 | """ | |||
|
227 | Returns size of the file at given ``path``. | |||
|
228 | """ | |||
|
229 | id = self._get_id_for_path(path) | |||
|
230 | blob = self.repository._repo[id] | |||
|
231 | return blob.raw_length() | |||
|
232 | ||||
|
233 | def get_file_changeset(self, path): | |||
|
234 | """ | |||
|
235 | Returns last commit of the file at the given ``path``. | |||
|
236 | """ | |||
|
237 | node = self.get_node(path) | |||
|
238 | return node.history[0] | |||
|
239 | ||||
|
240 | def get_file_history(self, path): | |||
|
241 | """ | |||
|
242 | Returns history of file as reversed list of ``Changeset`` objects for | |||
|
243 | which file at given ``path`` has been modified. | |||
|
244 | ||||
|
245 | TODO: This function now uses os underlying 'git' and 'grep' commands | |||
|
246 | which is generally not good. Should be replaced with algorithm | |||
|
247 | iterating commits. | |||
|
248 | """ | |||
|
249 | cmd = 'log --name-status -p %s -- "%s" | grep "^commit"' \ | |||
|
250 | % (self.id, path) | |||
|
251 | so, se = self.repository.run_git_command(cmd) | |||
|
252 | ids = re.findall(r'\w{40}', so) | |||
|
253 | return [self.repository.get_changeset(id) for id in ids] | |||
|
254 | ||||
|
255 | def get_file_annotate(self, path): | |||
|
256 | """ | |||
|
257 | Returns a list of three element tuples with lineno,changeset and line | |||
|
258 | ||||
|
259 | TODO: This function now uses os underlying 'git' command which is | |||
|
260 | generally not good. Should be replaced with algorithm iterating | |||
|
261 | commits. | |||
|
262 | """ | |||
|
263 | cmd = 'blame -l --root -r %s -- "%s"' % (self.id, path) | |||
|
264 | # -l ==> outputs long shas (and we need all 40 characters) | |||
|
265 | # --root ==> doesn't put '^' character for bounderies | |||
|
266 | # -r sha ==> blames for the given revision | |||
|
267 | so, se = self.repository.run_git_command(cmd) | |||
|
268 | annotate = [] | |||
|
269 | for i, blame_line in enumerate(so.split('\n')[:-1]): | |||
|
270 | ln_no = i + 1 | |||
|
271 | id, line = re.split(r' \(.+?\) ', blame_line, 1) | |||
|
272 | annotate.append((ln_no, self.repository.get_changeset(id), line)) | |||
|
273 | return annotate | |||
|
274 | ||||
|
275 | def fill_archive(self, stream=None, kind='tgz', prefix=None, | |||
|
276 | subrepos=False): | |||
|
277 | """ | |||
|
278 | Fills up given stream. | |||
|
279 | ||||
|
280 | :param stream: file like object. | |||
|
281 | :param kind: one of following: ``zip``, ``tgz`` or ``tbz2``. | |||
|
282 | Default: ``tgz``. | |||
|
283 | :param prefix: name of root directory in archive. | |||
|
284 | Default is repository name and changeset's raw_id joined with dash | |||
|
285 | (``repo-tip.<KIND>``). | |||
|
286 | :param subrepos: include subrepos in this archive. | |||
|
287 | ||||
|
288 | :raise ImproperArchiveTypeError: If given kind is wrong. | |||
|
289 | :raise VcsError: If given stream is None | |||
|
290 | ||||
|
291 | """ | |||
|
292 | allowed_kinds = settings.ARCHIVE_SPECS.keys() | |||
|
293 | if kind not in allowed_kinds: | |||
|
294 | raise ImproperArchiveTypeError('Archive kind not supported use one' | |||
|
295 | 'of %s', allowed_kinds) | |||
|
296 | ||||
|
297 | if prefix is None: | |||
|
298 | prefix = '%s-%s' % (self.repository.name, self.short_id) | |||
|
299 | elif prefix.startswith('/'): | |||
|
300 | raise VCSError("Prefix cannot start with leading slash") | |||
|
301 | elif prefix.strip() == '': | |||
|
302 | raise VCSError("Prefix cannot be empty") | |||
|
303 | ||||
|
304 | if kind == 'zip': | |||
|
305 | frmt = 'zip' | |||
|
306 | else: | |||
|
307 | frmt = 'tar' | |||
|
308 | cmd = 'git archive --format=%s --prefix=%s/ %s' % (frmt, prefix, | |||
|
309 | self.raw_id) | |||
|
310 | if kind == 'tgz': | |||
|
311 | cmd += ' | gzip -9' | |||
|
312 | elif kind == 'tbz2': | |||
|
313 | cmd += ' | bzip2 -9' | |||
|
314 | ||||
|
315 | if stream is None: | |||
|
316 | raise VCSError('You need to pass in a valid stream for filling' | |||
|
317 | ' with archival data') | |||
|
318 | popen = Popen(cmd, stdout=PIPE, stderr=PIPE, shell=True, | |||
|
319 | cwd=self.repository.path) | |||
|
320 | ||||
|
321 | buffer_size = 1024 * 8 | |||
|
322 | chunk = popen.stdout.read(buffer_size) | |||
|
323 | while chunk: | |||
|
324 | stream.write(chunk) | |||
|
325 | chunk = popen.stdout.read(buffer_size) | |||
|
326 | # Make sure all descriptors would be read | |||
|
327 | popen.communicate() | |||
|
328 | ||||
|
329 | def get_nodes(self, path): | |||
|
330 | if self._get_kind(path) != NodeKind.DIR: | |||
|
331 | raise ChangesetError("Directory does not exist for revision %r at " | |||
|
332 | " %r" % (self.revision, path)) | |||
|
333 | path = self._fix_path(path) | |||
|
334 | id = self._get_id_for_path(path) | |||
|
335 | tree = self.repository._repo[id] | |||
|
336 | dirnodes = [] | |||
|
337 | filenodes = [] | |||
|
338 | for name, stat, id in tree.iteritems(): | |||
|
339 | obj = self.repository._repo.get_object(id) | |||
|
340 | if path != '': | |||
|
341 | obj_path = '/'.join((path, name)) | |||
|
342 | else: | |||
|
343 | obj_path = name | |||
|
344 | if obj_path not in self._stat_modes: | |||
|
345 | self._stat_modes[obj_path] = stat | |||
|
346 | if isinstance(obj, objects.Tree): | |||
|
347 | dirnodes.append(DirNode(obj_path, changeset=self)) | |||
|
348 | elif isinstance(obj, objects.Blob): | |||
|
349 | filenodes.append(FileNode(obj_path, changeset=self, mode=stat)) | |||
|
350 | else: | |||
|
351 | raise ChangesetError("Requested object should be Tree " | |||
|
352 | "or Blob, is %r" % type(obj)) | |||
|
353 | nodes = dirnodes + filenodes | |||
|
354 | for node in nodes: | |||
|
355 | if not node.path in self.nodes: | |||
|
356 | self.nodes[node.path] = node | |||
|
357 | nodes.sort() | |||
|
358 | return nodes | |||
|
359 | ||||
|
360 | def get_node(self, path): | |||
|
361 | if isinstance(path, unicode): | |||
|
362 | path = path.encode('utf-8') | |||
|
363 | path = self._fix_path(path) | |||
|
364 | if not path in self.nodes: | |||
|
365 | try: | |||
|
366 | id = self._get_id_for_path(path) | |||
|
367 | except ChangesetError: | |||
|
368 | raise NodeDoesNotExistError("Cannot find one of parents' " | |||
|
369 | "directories for a given path: %s" % path) | |||
|
370 | obj = self.repository._repo.get_object(id) | |||
|
371 | if isinstance(obj, objects.Tree): | |||
|
372 | if path == '': | |||
|
373 | node = RootNode(changeset=self) | |||
|
374 | else: | |||
|
375 | node = DirNode(path, changeset=self) | |||
|
376 | node._tree = obj | |||
|
377 | elif isinstance(obj, objects.Blob): | |||
|
378 | node = FileNode(path, changeset=self) | |||
|
379 | node._blob = obj | |||
|
380 | else: | |||
|
381 | raise NodeDoesNotExistError("There is no file nor directory " | |||
|
382 | "at the given path %r at revision %r" | |||
|
383 | % (path, self.short_id)) | |||
|
384 | # cache node | |||
|
385 | self.nodes[path] = node | |||
|
386 | return self.nodes[path] | |||
|
387 | ||||
|
388 | @LazyProperty | |||
|
389 | def affected_files(self): | |||
|
390 | """ | |||
|
391 | Get's a fast accessible file changes for given changeset | |||
|
392 | """ | |||
|
393 | ||||
|
394 | return self.added + self.changed | |||
|
395 | ||||
|
396 | @LazyProperty | |||
|
397 | def _diff_name_status(self): | |||
|
398 | output = [] | |||
|
399 | for parent in self.parents: | |||
|
400 | cmd = 'diff --name-status %s %s' % (parent.raw_id, self.raw_id) | |||
|
401 | so, se = self.repository.run_git_command(cmd) | |||
|
402 | output.append(so.strip()) | |||
|
403 | return '\n'.join(output) | |||
|
404 | ||||
|
405 | def _get_paths_for_status(self, status): | |||
|
406 | """ | |||
|
407 | Returns sorted list of paths for given ``status``. | |||
|
408 | ||||
|
409 | :param status: one of: *added*, *modified* or *deleted* | |||
|
410 | """ | |||
|
411 | paths = set() | |||
|
412 | char = status[0].upper() | |||
|
413 | for line in self._diff_name_status.splitlines(): | |||
|
414 | if not line: | |||
|
415 | continue | |||
|
416 | if line.startswith(char): | |||
|
417 | splitted = line.split(char,1) | |||
|
418 | if not len(splitted) == 2: | |||
|
419 | raise VCSError("Couldn't parse diff result:\n%s\n\n and " | |||
|
420 | "particularly that line: %s" % (self._diff_name_status, | |||
|
421 | line)) | |||
|
422 | paths.add(splitted[1].strip()) | |||
|
423 | return sorted(paths) | |||
|
424 | ||||
|
425 | @LazyProperty | |||
|
426 | def added(self): | |||
|
427 | """ | |||
|
428 | Returns list of added ``FileNode`` objects. | |||
|
429 | """ | |||
|
430 | if not self.parents: | |||
|
431 | return list(self._get_file_nodes()) | |||
|
432 | return [self.get_node(path) for path in self._get_paths_for_status('added')] | |||
|
433 | ||||
|
434 | @LazyProperty | |||
|
435 | def changed(self): | |||
|
436 | """ | |||
|
437 | Returns list of modified ``FileNode`` objects. | |||
|
438 | """ | |||
|
439 | if not self.parents: | |||
|
440 | return [] | |||
|
441 | return [self.get_node(path) for path in self._get_paths_for_status('modified')] | |||
|
442 | ||||
|
443 | @LazyProperty | |||
|
444 | def removed(self): | |||
|
445 | """ | |||
|
446 | Returns list of removed ``FileNode`` objects. | |||
|
447 | """ | |||
|
448 | if not self.parents: | |||
|
449 | return [] | |||
|
450 | return [RemovedFileNode(path) for path in self._get_paths_for_status('deleted')] |
@@ -0,0 +1,347 b'' | |||||
|
1 | # config.py - Reading and writing Git config files | |||
|
2 | # Copyright (C) 2011 Jelmer Vernooij <jelmer@samba.org> | |||
|
3 | # | |||
|
4 | # This program is free software; you can redistribute it and/or | |||
|
5 | # modify it under the terms of the GNU General Public License | |||
|
6 | # as published by the Free Software Foundation; version 2 | |||
|
7 | # of the License or (at your option) a later version. | |||
|
8 | # | |||
|
9 | # This program is distributed in the hope that it will be useful, | |||
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |||
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |||
|
12 | # GNU General Public License for more details. | |||
|
13 | # | |||
|
14 | # You should have received a copy of the GNU General Public License | |||
|
15 | # along with this program; if not, write to the Free Software | |||
|
16 | # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, | |||
|
17 | # MA 02110-1301, USA. | |||
|
18 | ||||
|
19 | """Reading and writing Git configuration files. | |||
|
20 | ||||
|
21 | TODO: | |||
|
22 | * preserve formatting when updating configuration files | |||
|
23 | * treat subsection names as case-insensitive for [branch.foo] style | |||
|
24 | subsections | |||
|
25 | """ | |||
|
26 | ||||
|
27 | # Taken from dulwich not yet released 0.8.3 version (until it is actually | |||
|
28 | # released) | |||
|
29 | ||||
|
30 | import errno | |||
|
31 | import os | |||
|
32 | import re | |||
|
33 | ||||
|
34 | from dulwich.file import GitFile | |||
|
35 | ||||
|
36 | ||||
|
37 | class Config(object): | |||
|
38 | """A Git configuration.""" | |||
|
39 | ||||
|
40 | def get(self, section, name): | |||
|
41 | """Retrieve the contents of a configuration setting. | |||
|
42 | ||||
|
43 | :param section: Tuple with section name and optional subsection namee | |||
|
44 | :param subsection: Subsection name | |||
|
45 | :return: Contents of the setting | |||
|
46 | :raise KeyError: if the value is not set | |||
|
47 | """ | |||
|
48 | raise NotImplementedError(self.get) | |||
|
49 | ||||
|
50 | def get_boolean(self, section, name, default=None): | |||
|
51 | """Retrieve a configuration setting as boolean. | |||
|
52 | ||||
|
53 | :param section: Tuple with section name and optional subsection namee | |||
|
54 | :param name: Name of the setting, including section and possible | |||
|
55 | subsection. | |||
|
56 | :return: Contents of the setting | |||
|
57 | :raise KeyError: if the value is not set | |||
|
58 | """ | |||
|
59 | try: | |||
|
60 | value = self.get(section, name) | |||
|
61 | except KeyError: | |||
|
62 | return default | |||
|
63 | if value.lower() == "true": | |||
|
64 | return True | |||
|
65 | elif value.lower() == "false": | |||
|
66 | return False | |||
|
67 | raise ValueError("not a valid boolean string: %r" % value) | |||
|
68 | ||||
|
69 | def set(self, section, name, value): | |||
|
70 | """Set a configuration value. | |||
|
71 | ||||
|
72 | :param name: Name of the configuration value, including section | |||
|
73 | and optional subsection | |||
|
74 | :param: Value of the setting | |||
|
75 | """ | |||
|
76 | raise NotImplementedError(self.set) | |||
|
77 | ||||
|
78 | ||||
|
79 | class ConfigDict(Config): | |||
|
80 | """Git configuration stored in a dictionary.""" | |||
|
81 | ||||
|
82 | def __init__(self, values=None): | |||
|
83 | """Create a new ConfigDict.""" | |||
|
84 | if values is None: | |||
|
85 | values = {} | |||
|
86 | self._values = values | |||
|
87 | ||||
|
88 | def __repr__(self): | |||
|
89 | return "%s(%r)" % (self.__class__.__name__, self._values) | |||
|
90 | ||||
|
91 | def __eq__(self, other): | |||
|
92 | return ( | |||
|
93 | isinstance(other, self.__class__) and | |||
|
94 | other._values == self._values) | |||
|
95 | ||||
|
96 | @classmethod | |||
|
97 | def _parse_setting(cls, name): | |||
|
98 | parts = name.split(".") | |||
|
99 | if len(parts) == 3: | |||
|
100 | return (parts[0], parts[1], parts[2]) | |||
|
101 | else: | |||
|
102 | return (parts[0], None, parts[1]) | |||
|
103 | ||||
|
104 | def get(self, section, name): | |||
|
105 | if isinstance(section, basestring): | |||
|
106 | section = (section, ) | |||
|
107 | if len(section) > 1: | |||
|
108 | try: | |||
|
109 | return self._values[section][name] | |||
|
110 | except KeyError: | |||
|
111 | pass | |||
|
112 | return self._values[(section[0],)][name] | |||
|
113 | ||||
|
114 | def set(self, section, name, value): | |||
|
115 | if isinstance(section, basestring): | |||
|
116 | section = (section, ) | |||
|
117 | self._values.setdefault(section, {})[name] = value | |||
|
118 | ||||
|
119 | ||||
|
120 | def _format_string(value): | |||
|
121 | if (value.startswith(" ") or | |||
|
122 | value.startswith("\t") or | |||
|
123 | value.endswith(" ") or | |||
|
124 | value.endswith("\t")): | |||
|
125 | return '"%s"' % _escape_value(value) | |||
|
126 | return _escape_value(value) | |||
|
127 | ||||
|
128 | ||||
|
129 | def _parse_string(value): | |||
|
130 | value = value.strip() | |||
|
131 | ret = [] | |||
|
132 | block = [] | |||
|
133 | in_quotes = False | |||
|
134 | for c in value: | |||
|
135 | if c == "\"": | |||
|
136 | in_quotes = (not in_quotes) | |||
|
137 | ret.append(_unescape_value("".join(block))) | |||
|
138 | block = [] | |||
|
139 | elif c in ("#", ";") and not in_quotes: | |||
|
140 | # the rest of the line is a comment | |||
|
141 | break | |||
|
142 | else: | |||
|
143 | block.append(c) | |||
|
144 | ||||
|
145 | if in_quotes: | |||
|
146 | raise ValueError("value starts with quote but lacks end quote") | |||
|
147 | ||||
|
148 | ret.append(_unescape_value("".join(block)).rstrip()) | |||
|
149 | ||||
|
150 | return "".join(ret) | |||
|
151 | ||||
|
152 | ||||
|
153 | def _unescape_value(value): | |||
|
154 | """Unescape a value.""" | |||
|
155 | def unescape(c): | |||
|
156 | return { | |||
|
157 | "\\\\": "\\", | |||
|
158 | "\\\"": "\"", | |||
|
159 | "\\n": "\n", | |||
|
160 | "\\t": "\t", | |||
|
161 | "\\b": "\b", | |||
|
162 | }[c.group(0)] | |||
|
163 | return re.sub(r"(\\.)", unescape, value) | |||
|
164 | ||||
|
165 | ||||
|
166 | def _escape_value(value): | |||
|
167 | """Escape a value.""" | |||
|
168 | return value.replace("\\", "\\\\").replace("\n", "\\n")\ | |||
|
169 | .replace("\t", "\\t").replace("\"", "\\\"") | |||
|
170 | ||||
|
171 | ||||
|
172 | def _check_variable_name(name): | |||
|
173 | for c in name: | |||
|
174 | if not c.isalnum() and c != '-': | |||
|
175 | return False | |||
|
176 | return True | |||
|
177 | ||||
|
178 | ||||
|
179 | def _check_section_name(name): | |||
|
180 | for c in name: | |||
|
181 | if not c.isalnum() and c not in ('-', '.'): | |||
|
182 | return False | |||
|
183 | return True | |||
|
184 | ||||
|
185 | ||||
|
186 | def _strip_comments(line): | |||
|
187 | line = line.split("#")[0] | |||
|
188 | line = line.split(";")[0] | |||
|
189 | return line | |||
|
190 | ||||
|
191 | ||||
|
192 | class ConfigFile(ConfigDict): | |||
|
193 | """A Git configuration file, like .git/config or ~/.gitconfig. | |||
|
194 | """ | |||
|
195 | ||||
|
196 | @classmethod | |||
|
197 | def from_file(cls, f): | |||
|
198 | """Read configuration from a file-like object.""" | |||
|
199 | ret = cls() | |||
|
200 | section = None | |||
|
201 | setting = None | |||
|
202 | for lineno, line in enumerate(f.readlines()): | |||
|
203 | line = line.lstrip() | |||
|
204 | if setting is None: | |||
|
205 | if _strip_comments(line).strip() == "": | |||
|
206 | continue | |||
|
207 | if line[0] == "[": | |||
|
208 | line = _strip_comments(line).rstrip() | |||
|
209 | if line[-1] != "]": | |||
|
210 | raise ValueError("expected trailing ]") | |||
|
211 | key = line.strip() | |||
|
212 | pts = key[1:-1].split(" ", 1) | |||
|
213 | pts[0] = pts[0].lower() | |||
|
214 | if len(pts) == 2: | |||
|
215 | if pts[1][0] != "\"" or pts[1][-1] != "\"": | |||
|
216 | raise ValueError( | |||
|
217 | "Invalid subsection " + pts[1]) | |||
|
218 | else: | |||
|
219 | pts[1] = pts[1][1:-1] | |||
|
220 | if not _check_section_name(pts[0]): | |||
|
221 | raise ValueError("invalid section name %s" % | |||
|
222 | pts[0]) | |||
|
223 | section = (pts[0], pts[1]) | |||
|
224 | else: | |||
|
225 | if not _check_section_name(pts[0]): | |||
|
226 | raise ValueError("invalid section name %s" % | |||
|
227 | pts[0]) | |||
|
228 | pts = pts[0].split(".", 1) | |||
|
229 | if len(pts) == 2: | |||
|
230 | section = (pts[0], pts[1]) | |||
|
231 | else: | |||
|
232 | section = (pts[0], ) | |||
|
233 | ret._values[section] = {} | |||
|
234 | else: | |||
|
235 | if section is None: | |||
|
236 | raise ValueError("setting %r without section" % line) | |||
|
237 | try: | |||
|
238 | setting, value = line.split("=", 1) | |||
|
239 | except ValueError: | |||
|
240 | setting = line | |||
|
241 | value = "true" | |||
|
242 | setting = setting.strip().lower() | |||
|
243 | if not _check_variable_name(setting): | |||
|
244 | raise ValueError("invalid variable name %s" % setting) | |||
|
245 | if value.endswith("\\\n"): | |||
|
246 | value = value[:-2] | |||
|
247 | continuation = True | |||
|
248 | else: | |||
|
249 | continuation = False | |||
|
250 | value = _parse_string(value) | |||
|
251 | ret._values[section][setting] = value | |||
|
252 | if not continuation: | |||
|
253 | setting = None | |||
|
254 | else: # continuation line | |||
|
255 | if line.endswith("\\\n"): | |||
|
256 | line = line[:-2] | |||
|
257 | continuation = True | |||
|
258 | else: | |||
|
259 | continuation = False | |||
|
260 | value = _parse_string(line) | |||
|
261 | ret._values[section][setting] += value | |||
|
262 | if not continuation: | |||
|
263 | setting = None | |||
|
264 | return ret | |||
|
265 | ||||
|
266 | @classmethod | |||
|
267 | def from_path(cls, path): | |||
|
268 | """Read configuration from a file on disk.""" | |||
|
269 | f = GitFile(path, 'rb') | |||
|
270 | try: | |||
|
271 | ret = cls.from_file(f) | |||
|
272 | ret.path = path | |||
|
273 | return ret | |||
|
274 | finally: | |||
|
275 | f.close() | |||
|
276 | ||||
|
277 | def write_to_path(self, path=None): | |||
|
278 | """Write configuration to a file on disk.""" | |||
|
279 | if path is None: | |||
|
280 | path = self.path | |||
|
281 | f = GitFile(path, 'wb') | |||
|
282 | try: | |||
|
283 | self.write_to_file(f) | |||
|
284 | finally: | |||
|
285 | f.close() | |||
|
286 | ||||
|
287 | def write_to_file(self, f): | |||
|
288 | """Write configuration to a file-like object.""" | |||
|
289 | for section, values in self._values.iteritems(): | |||
|
290 | try: | |||
|
291 | section_name, subsection_name = section | |||
|
292 | except ValueError: | |||
|
293 | (section_name, ) = section | |||
|
294 | subsection_name = None | |||
|
295 | if subsection_name is None: | |||
|
296 | f.write("[%s]\n" % section_name) | |||
|
297 | else: | |||
|
298 | f.write("[%s \"%s\"]\n" % (section_name, subsection_name)) | |||
|
299 | for key, value in values.iteritems(): | |||
|
300 | f.write("%s = %s\n" % (key, _escape_value(value))) | |||
|
301 | ||||
|
302 | ||||
|
303 | class StackedConfig(Config): | |||
|
304 | """Configuration which reads from multiple config files..""" | |||
|
305 | ||||
|
306 | def __init__(self, backends, writable=None): | |||
|
307 | self.backends = backends | |||
|
308 | self.writable = writable | |||
|
309 | ||||
|
310 | def __repr__(self): | |||
|
311 | return "<%s for %r>" % (self.__class__.__name__, self.backends) | |||
|
312 | ||||
|
313 | @classmethod | |||
|
314 | def default_backends(cls): | |||
|
315 | """Retrieve the default configuration. | |||
|
316 | ||||
|
317 | This will look in the repository configuration (if for_path is | |||
|
318 | specified), the users' home directory and the system | |||
|
319 | configuration. | |||
|
320 | """ | |||
|
321 | paths = [] | |||
|
322 | paths.append(os.path.expanduser("~/.gitconfig")) | |||
|
323 | paths.append("/etc/gitconfig") | |||
|
324 | backends = [] | |||
|
325 | for path in paths: | |||
|
326 | try: | |||
|
327 | cf = ConfigFile.from_path(path) | |||
|
328 | except (IOError, OSError), e: | |||
|
329 | if e.errno != errno.ENOENT: | |||
|
330 | raise | |||
|
331 | else: | |||
|
332 | continue | |||
|
333 | backends.append(cf) | |||
|
334 | return backends | |||
|
335 | ||||
|
336 | def get(self, section, name): | |||
|
337 | for backend in self.backends: | |||
|
338 | try: | |||
|
339 | return backend.get(section, name) | |||
|
340 | except KeyError: | |||
|
341 | pass | |||
|
342 | raise KeyError(name) | |||
|
343 | ||||
|
344 | def set(self, section, name, value): | |||
|
345 | if self.writable is None: | |||
|
346 | raise NotImplementedError(self.set) | |||
|
347 | return self.writable.set(section, name, value) |
@@ -0,0 +1,192 b'' | |||||
|
1 | import time | |||
|
2 | import datetime | |||
|
3 | import posixpath | |||
|
4 | from dulwich import objects | |||
|
5 | from dulwich.repo import Repo | |||
|
6 | from rhodecode.lib.vcs.backends.base import BaseInMemoryChangeset | |||
|
7 | from rhodecode.lib.vcs.exceptions import RepositoryError | |||
|
8 | ||||
|
9 | ||||
|
10 | class GitInMemoryChangeset(BaseInMemoryChangeset): | |||
|
11 | ||||
|
12 | def commit(self, message, author, parents=None, branch=None, date=None, | |||
|
13 | **kwargs): | |||
|
14 | """ | |||
|
15 | Performs in-memory commit (doesn't check workdir in any way) and | |||
|
16 | returns newly created ``Changeset``. Updates repository's | |||
|
17 | ``revisions``. | |||
|
18 | ||||
|
19 | :param message: message of the commit | |||
|
20 | :param author: full username, i.e. "Joe Doe <joe.doe@example.com>" | |||
|
21 | :param parents: single parent or sequence of parents from which commit | |||
|
22 | would be derieved | |||
|
23 | :param date: ``datetime.datetime`` instance. Defaults to | |||
|
24 | ``datetime.datetime.now()``. | |||
|
25 | :param branch: branch name, as string. If none given, default backend's | |||
|
26 | branch would be used. | |||
|
27 | ||||
|
28 | :raises ``CommitError``: if any error occurs while committing | |||
|
29 | """ | |||
|
30 | self.check_integrity(parents) | |||
|
31 | ||||
|
32 | from .repository import GitRepository | |||
|
33 | if branch is None: | |||
|
34 | branch = GitRepository.DEFAULT_BRANCH_NAME | |||
|
35 | ||||
|
36 | repo = self.repository._repo | |||
|
37 | object_store = repo.object_store | |||
|
38 | ||||
|
39 | ENCODING = "UTF-8" | |||
|
40 | DIRMOD = 040000 | |||
|
41 | ||||
|
42 | # Create tree and populates it with blobs | |||
|
43 | commit_tree = self.parents[0] and repo[self.parents[0]._commit.tree] or\ | |||
|
44 | objects.Tree() | |||
|
45 | for node in self.added + self.changed: | |||
|
46 | # Compute subdirs if needed | |||
|
47 | dirpath, nodename = posixpath.split(node.path) | |||
|
48 | dirnames = dirpath and dirpath.split('/') or [] | |||
|
49 | parent = commit_tree | |||
|
50 | ancestors = [('', parent)] | |||
|
51 | ||||
|
52 | # Tries to dig for the deepest existing tree | |||
|
53 | while dirnames: | |||
|
54 | curdir = dirnames.pop(0) | |||
|
55 | try: | |||
|
56 | dir_id = parent[curdir][1] | |||
|
57 | except KeyError: | |||
|
58 | # put curdir back into dirnames and stops | |||
|
59 | dirnames.insert(0, curdir) | |||
|
60 | break | |||
|
61 | else: | |||
|
62 | # If found, updates parent | |||
|
63 | parent = self.repository._repo[dir_id] | |||
|
64 | ancestors.append((curdir, parent)) | |||
|
65 | # Now parent is deepest exising tree and we need to create subtrees | |||
|
66 | # for dirnames (in reverse order) [this only applies for nodes from added] | |||
|
67 | new_trees = [] | |||
|
68 | blob = objects.Blob.from_string(node.content.encode(ENCODING)) | |||
|
69 | node_path = node.name.encode(ENCODING) | |||
|
70 | if dirnames: | |||
|
71 | # If there are trees which should be created we need to build | |||
|
72 | # them now (in reverse order) | |||
|
73 | reversed_dirnames = list(reversed(dirnames)) | |||
|
74 | curtree = objects.Tree() | |||
|
75 | curtree[node_path] = node.mode, blob.id | |||
|
76 | new_trees.append(curtree) | |||
|
77 | for dirname in reversed_dirnames[:-1]: | |||
|
78 | newtree = objects.Tree() | |||
|
79 | #newtree.add(DIRMOD, dirname, curtree.id) | |||
|
80 | newtree[dirname] = DIRMOD, curtree.id | |||
|
81 | new_trees.append(newtree) | |||
|
82 | curtree = newtree | |||
|
83 | parent[reversed_dirnames[-1]] = DIRMOD, curtree.id | |||
|
84 | else: | |||
|
85 | parent.add(node.mode, node_path, blob.id) | |||
|
86 | new_trees.append(parent) | |||
|
87 | # Update ancestors | |||
|
88 | for parent, tree, path in reversed([(a[1], b[1], b[0]) for a, b in | |||
|
89 | zip(ancestors, ancestors[1:])]): | |||
|
90 | parent[path] = DIRMOD, tree.id | |||
|
91 | object_store.add_object(tree) | |||
|
92 | ||||
|
93 | object_store.add_object(blob) | |||
|
94 | for tree in new_trees: | |||
|
95 | object_store.add_object(tree) | |||
|
96 | for node in self.removed: | |||
|
97 | paths = node.path.split('/') | |||
|
98 | tree = commit_tree | |||
|
99 | trees = [tree] | |||
|
100 | # Traverse deep into the forest... | |||
|
101 | for path in paths: | |||
|
102 | try: | |||
|
103 | obj = self.repository._repo[tree[path][1]] | |||
|
104 | if isinstance(obj, objects.Tree): | |||
|
105 | trees.append(obj) | |||
|
106 | tree = obj | |||
|
107 | except KeyError: | |||
|
108 | break | |||
|
109 | # Cut down the blob and all rotten trees on the way back... | |||
|
110 | for path, tree in reversed(zip(paths, trees)): | |||
|
111 | del tree[path] | |||
|
112 | if tree: | |||
|
113 | # This tree still has elements - don't remove it or any | |||
|
114 | # of it's parents | |||
|
115 | break | |||
|
116 | ||||
|
117 | object_store.add_object(commit_tree) | |||
|
118 | ||||
|
119 | # Create commit | |||
|
120 | commit = objects.Commit() | |||
|
121 | commit.tree = commit_tree.id | |||
|
122 | commit.parents = [p._commit.id for p in self.parents if p] | |||
|
123 | commit.author = commit.committer = author | |||
|
124 | commit.encoding = ENCODING | |||
|
125 | commit.message = message + ' ' | |||
|
126 | ||||
|
127 | # Compute date | |||
|
128 | if date is None: | |||
|
129 | date = time.time() | |||
|
130 | elif isinstance(date, datetime.datetime): | |||
|
131 | date = time.mktime(date.timetuple()) | |||
|
132 | ||||
|
133 | author_time = kwargs.pop('author_time', date) | |||
|
134 | commit.commit_time = int(date) | |||
|
135 | commit.author_time = int(author_time) | |||
|
136 | tz = time.timezone | |||
|
137 | author_tz = kwargs.pop('author_timezone', tz) | |||
|
138 | commit.commit_timezone = tz | |||
|
139 | commit.author_timezone = author_tz | |||
|
140 | ||||
|
141 | object_store.add_object(commit) | |||
|
142 | ||||
|
143 | ref = 'refs/heads/%s' % branch | |||
|
144 | repo.refs[ref] = commit.id | |||
|
145 | repo.refs.set_symbolic_ref('HEAD', ref) | |||
|
146 | ||||
|
147 | # Update vcs repository object & recreate dulwich repo | |||
|
148 | self.repository.revisions.append(commit.id) | |||
|
149 | self.repository._repo = Repo(self.repository.path) | |||
|
150 | tip = self.repository.get_changeset() | |||
|
151 | self.reset() | |||
|
152 | return tip | |||
|
153 | ||||
|
154 | def _get_missing_trees(self, path, root_tree): | |||
|
155 | """ | |||
|
156 | Creates missing ``Tree`` objects for the given path. | |||
|
157 | ||||
|
158 | :param path: path given as a string. It may be a path to a file node | |||
|
159 | (i.e. ``foo/bar/baz.txt``) or directory path - in that case it must | |||
|
160 | end with slash (i.e. ``foo/bar/``). | |||
|
161 | :param root_tree: ``dulwich.objects.Tree`` object from which we start | |||
|
162 | traversing (should be commit's root tree) | |||
|
163 | """ | |||
|
164 | dirpath = posixpath.split(path)[0] | |||
|
165 | dirs = dirpath.split('/') | |||
|
166 | if not dirs or dirs == ['']: | |||
|
167 | return [] | |||
|
168 | ||||
|
169 | def get_tree_for_dir(tree, dirname): | |||
|
170 | for name, mode, id in tree.iteritems(): | |||
|
171 | if name == dirname: | |||
|
172 | obj = self.repository._repo[id] | |||
|
173 | if isinstance(obj, objects.Tree): | |||
|
174 | return obj | |||
|
175 | else: | |||
|
176 | raise RepositoryError("Cannot create directory %s " | |||
|
177 | "at tree %s as path is occupied and is not a " | |||
|
178 | "Tree" % (dirname, tree)) | |||
|
179 | return None | |||
|
180 | ||||
|
181 | trees = [] | |||
|
182 | parent = root_tree | |||
|
183 | for dirname in dirs: | |||
|
184 | tree = get_tree_for_dir(parent, dirname) | |||
|
185 | if tree is None: | |||
|
186 | tree = objects.Tree() | |||
|
187 | dirmode = 040000 | |||
|
188 | parent.add(dirmode, dirname, tree.id) | |||
|
189 | parent = tree | |||
|
190 | # Always append tree | |||
|
191 | trees.append(tree) | |||
|
192 | return trees |
This diff has been collapsed as it changes many lines, (508 lines changed) Show them Hide them | |||||
@@ -0,0 +1,508 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | """ | |||
|
3 | vcs.backends.git | |||
|
4 | ~~~~~~~~~~~~~~~~ | |||
|
5 | ||||
|
6 | Git backend implementation. | |||
|
7 | ||||
|
8 | :created_on: Apr 8, 2010 | |||
|
9 | :copyright: (c) 2010-2011 by Marcin Kuzminski, Lukasz Balcerzak. | |||
|
10 | """ | |||
|
11 | ||||
|
12 | import os | |||
|
13 | import re | |||
|
14 | import time | |||
|
15 | import posixpath | |||
|
16 | from dulwich.repo import Repo, NotGitRepository | |||
|
17 | #from dulwich.config import ConfigFile | |||
|
18 | from string import Template | |||
|
19 | from subprocess import Popen, PIPE | |||
|
20 | from rhodecode.lib.vcs.backends.base import BaseRepository | |||
|
21 | from rhodecode.lib.vcs.exceptions import BranchDoesNotExistError | |||
|
22 | from rhodecode.lib.vcs.exceptions import ChangesetDoesNotExistError | |||
|
23 | from rhodecode.lib.vcs.exceptions import EmptyRepositoryError | |||
|
24 | from rhodecode.lib.vcs.exceptions import RepositoryError | |||
|
25 | from rhodecode.lib.vcs.exceptions import TagAlreadyExistError | |||
|
26 | from rhodecode.lib.vcs.exceptions import TagDoesNotExistError | |||
|
27 | from rhodecode.lib.vcs.utils import safe_unicode, makedate, date_fromtimestamp | |||
|
28 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |||
|
29 | from rhodecode.lib.vcs.utils.ordered_dict import OrderedDict | |||
|
30 | from rhodecode.lib.vcs.utils.paths import abspath | |||
|
31 | from rhodecode.lib.vcs.utils.paths import get_user_home | |||
|
32 | from .workdir import GitWorkdir | |||
|
33 | from .changeset import GitChangeset | |||
|
34 | from .inmemory import GitInMemoryChangeset | |||
|
35 | from .config import ConfigFile | |||
|
36 | ||||
|
37 | ||||
|
38 | class GitRepository(BaseRepository): | |||
|
39 | """ | |||
|
40 | Git repository backend. | |||
|
41 | """ | |||
|
42 | DEFAULT_BRANCH_NAME = 'master' | |||
|
43 | scm = 'git' | |||
|
44 | ||||
|
45 | def __init__(self, repo_path, create=False, src_url=None, | |||
|
46 | update_after_clone=False, bare=False): | |||
|
47 | ||||
|
48 | self.path = abspath(repo_path) | |||
|
49 | self._repo = self._get_repo(create, src_url, update_after_clone, bare) | |||
|
50 | try: | |||
|
51 | self.head = self._repo.head() | |||
|
52 | except KeyError: | |||
|
53 | self.head = None | |||
|
54 | ||||
|
55 | self._config_files = [ | |||
|
56 | bare and abspath(self.path, 'config') or abspath(self.path, '.git', | |||
|
57 | 'config'), | |||
|
58 | abspath(get_user_home(), '.gitconfig'), | |||
|
59 | ] | |||
|
60 | ||||
|
61 | @LazyProperty | |||
|
62 | def revisions(self): | |||
|
63 | """ | |||
|
64 | Returns list of revisions' ids, in ascending order. Being lazy | |||
|
65 | attribute allows external tools to inject shas from cache. | |||
|
66 | """ | |||
|
67 | return self._get_all_revisions() | |||
|
68 | ||||
|
69 | def run_git_command(self, cmd): | |||
|
70 | """ | |||
|
71 | Runs given ``cmd`` as git command and returns tuple | |||
|
72 | (returncode, stdout, stderr). | |||
|
73 | ||||
|
74 | .. note:: | |||
|
75 | This method exists only until log/blame functionality is implemented | |||
|
76 | at Dulwich (see https://bugs.launchpad.net/bugs/645142). Parsing | |||
|
77 | os command's output is road to hell... | |||
|
78 | ||||
|
79 | :param cmd: git command to be executed | |||
|
80 | """ | |||
|
81 | #cmd = '(cd %s && git %s)' % (self.path, cmd) | |||
|
82 | if isinstance(cmd, basestring): | |||
|
83 | cmd = 'git %s' % cmd | |||
|
84 | else: | |||
|
85 | cmd = ['git'] + cmd | |||
|
86 | try: | |||
|
87 | opts = dict( | |||
|
88 | shell=isinstance(cmd, basestring), | |||
|
89 | stdout=PIPE, | |||
|
90 | stderr=PIPE) | |||
|
91 | if os.path.isdir(self.path): | |||
|
92 | opts['cwd'] = self.path | |||
|
93 | p = Popen(cmd, **opts) | |||
|
94 | except OSError, err: | |||
|
95 | raise RepositoryError("Couldn't run git command (%s).\n" | |||
|
96 | "Original error was:%s" % (cmd, err)) | |||
|
97 | so, se = p.communicate() | |||
|
98 | if not se.startswith("fatal: bad default revision 'HEAD'") and \ | |||
|
99 | p.returncode != 0: | |||
|
100 | raise RepositoryError("Couldn't run git command (%s).\n" | |||
|
101 | "stderr:\n%s" % (cmd, se)) | |||
|
102 | return so, se | |||
|
103 | ||||
|
104 | def _check_url(self, url): | |||
|
105 | """ | |||
|
106 | Functon will check given url and try to verify if it's a valid | |||
|
107 | link. Sometimes it may happened that mercurial will issue basic | |||
|
108 | auth request that can cause whole API to hang when used from python | |||
|
109 | or other external calls. | |||
|
110 | ||||
|
111 | On failures it'll raise urllib2.HTTPError | |||
|
112 | """ | |||
|
113 | ||||
|
114 | #TODO: implement this | |||
|
115 | pass | |||
|
116 | ||||
|
117 | def _get_repo(self, create, src_url=None, update_after_clone=False, | |||
|
118 | bare=False): | |||
|
119 | if create and os.path.exists(self.path): | |||
|
120 | raise RepositoryError("Location already exist") | |||
|
121 | if src_url and not create: | |||
|
122 | raise RepositoryError("Create should be set to True if src_url is " | |||
|
123 | "given (clone operation creates repository)") | |||
|
124 | try: | |||
|
125 | if create and src_url: | |||
|
126 | self._check_url(src_url) | |||
|
127 | self.clone(src_url, update_after_clone, bare) | |||
|
128 | return Repo(self.path) | |||
|
129 | elif create: | |||
|
130 | os.mkdir(self.path) | |||
|
131 | if bare: | |||
|
132 | return Repo.init_bare(self.path) | |||
|
133 | else: | |||
|
134 | return Repo.init(self.path) | |||
|
135 | else: | |||
|
136 | return Repo(self.path) | |||
|
137 | except (NotGitRepository, OSError), err: | |||
|
138 | raise RepositoryError(err) | |||
|
139 | ||||
|
140 | def _get_all_revisions(self): | |||
|
141 | cmd = 'rev-list --all --date-order' | |||
|
142 | try: | |||
|
143 | so, se = self.run_git_command(cmd) | |||
|
144 | except RepositoryError: | |||
|
145 | # Can be raised for empty repositories | |||
|
146 | return [] | |||
|
147 | revisions = so.splitlines() | |||
|
148 | revisions.reverse() | |||
|
149 | return revisions | |||
|
150 | ||||
|
151 | def _get_revision(self, revision): | |||
|
152 | """ | |||
|
153 | For git backend we always return integer here. This way we ensure | |||
|
154 | that changset's revision attribute would become integer. | |||
|
155 | """ | |||
|
156 | pattern = re.compile(r'^[[0-9a-fA-F]{12}|[0-9a-fA-F]{40}]$') | |||
|
157 | is_bstr = lambda o: isinstance(o, (str, unicode)) | |||
|
158 | is_null = lambda o: len(o) == revision.count('0') | |||
|
159 | ||||
|
160 | if len(self.revisions) == 0: | |||
|
161 | raise EmptyRepositoryError("There are no changesets yet") | |||
|
162 | ||||
|
163 | if revision in (None, '', 'tip', 'HEAD', 'head', -1): | |||
|
164 | revision = self.revisions[-1] | |||
|
165 | ||||
|
166 | if ((is_bstr(revision) and revision.isdigit() and len(revision) < 12) | |||
|
167 | or isinstance(revision, int) or is_null(revision)): | |||
|
168 | try: | |||
|
169 | revision = self.revisions[int(revision)] | |||
|
170 | except: | |||
|
171 | raise ChangesetDoesNotExistError("Revision %r does not exist " | |||
|
172 | "for this repository %s" % (revision, self)) | |||
|
173 | ||||
|
174 | elif is_bstr(revision): | |||
|
175 | if not pattern.match(revision) or revision not in self.revisions: | |||
|
176 | raise ChangesetDoesNotExistError("Revision %r does not exist " | |||
|
177 | "for this repository %s" % (revision, self)) | |||
|
178 | ||||
|
179 | # Ensure we return full id | |||
|
180 | if not pattern.match(str(revision)): | |||
|
181 | raise ChangesetDoesNotExistError("Given revision %r not recognized" | |||
|
182 | % revision) | |||
|
183 | return revision | |||
|
184 | ||||
|
185 | def _get_archives(self, archive_name='tip'): | |||
|
186 | ||||
|
187 | for i in [('zip', '.zip'), ('gz', '.tar.gz'), ('bz2', '.tar.bz2')]: | |||
|
188 | yield {"type": i[0], "extension": i[1], "node": archive_name} | |||
|
189 | ||||
|
190 | def _get_url(self, url): | |||
|
191 | """ | |||
|
192 | Returns normalized url. If schema is not given, would fall to | |||
|
193 | filesystem (``file:///``) schema. | |||
|
194 | """ | |||
|
195 | url = str(url) | |||
|
196 | if url != 'default' and not '://' in url: | |||
|
197 | url = ':///'.join(('file', url)) | |||
|
198 | return url | |||
|
199 | ||||
|
200 | @LazyProperty | |||
|
201 | def name(self): | |||
|
202 | return os.path.basename(self.path) | |||
|
203 | ||||
|
204 | @LazyProperty | |||
|
205 | def last_change(self): | |||
|
206 | """ | |||
|
207 | Returns last change made on this repository as datetime object | |||
|
208 | """ | |||
|
209 | return date_fromtimestamp(self._get_mtime(), makedate()[1]) | |||
|
210 | ||||
|
211 | def _get_mtime(self): | |||
|
212 | try: | |||
|
213 | return time.mktime(self.get_changeset().date.timetuple()) | |||
|
214 | except RepositoryError: | |||
|
215 | # fallback to filesystem | |||
|
216 | in_path = os.path.join(self.path, '.git', "index") | |||
|
217 | he_path = os.path.join(self.path, '.git', "HEAD") | |||
|
218 | if os.path.exists(in_path): | |||
|
219 | return os.stat(in_path).st_mtime | |||
|
220 | else: | |||
|
221 | return os.stat(he_path).st_mtime | |||
|
222 | ||||
|
223 | @LazyProperty | |||
|
224 | def description(self): | |||
|
225 | undefined_description = u'unknown' | |||
|
226 | description_path = os.path.join(self.path, '.git', 'description') | |||
|
227 | if os.path.isfile(description_path): | |||
|
228 | return safe_unicode(open(description_path).read()) | |||
|
229 | else: | |||
|
230 | return undefined_description | |||
|
231 | ||||
|
232 | @LazyProperty | |||
|
233 | def contact(self): | |||
|
234 | undefined_contact = u'Unknown' | |||
|
235 | return undefined_contact | |||
|
236 | ||||
|
237 | @property | |||
|
238 | def branches(self): | |||
|
239 | if not self.revisions: | |||
|
240 | return {} | |||
|
241 | refs = self._repo.refs.as_dict() | |||
|
242 | sortkey = lambda ctx: ctx[0] | |||
|
243 | _branches = [('/'.join(ref.split('/')[2:]), head) | |||
|
244 | for ref, head in refs.items() | |||
|
245 | if ref.startswith('refs/heads/') or | |||
|
246 | ref.startswith('refs/remotes/') and not ref.endswith('/HEAD')] | |||
|
247 | return OrderedDict(sorted(_branches, key=sortkey, reverse=False)) | |||
|
248 | ||||
|
249 | def _get_tags(self): | |||
|
250 | if not self.revisions: | |||
|
251 | return {} | |||
|
252 | sortkey = lambda ctx: ctx[0] | |||
|
253 | _tags = [('/'.join(ref.split('/')[2:]), head) for ref, head in | |||
|
254 | self._repo.get_refs().items() if ref.startswith('refs/tags/')] | |||
|
255 | return OrderedDict(sorted(_tags, key=sortkey, reverse=True)) | |||
|
256 | ||||
|
257 | @LazyProperty | |||
|
258 | def tags(self): | |||
|
259 | return self._get_tags() | |||
|
260 | ||||
|
261 | def tag(self, name, user, revision=None, message=None, date=None, | |||
|
262 | **kwargs): | |||
|
263 | """ | |||
|
264 | Creates and returns a tag for the given ``revision``. | |||
|
265 | ||||
|
266 | :param name: name for new tag | |||
|
267 | :param user: full username, i.e.: "Joe Doe <joe.doe@example.com>" | |||
|
268 | :param revision: changeset id for which new tag would be created | |||
|
269 | :param message: message of the tag's commit | |||
|
270 | :param date: date of tag's commit | |||
|
271 | ||||
|
272 | :raises TagAlreadyExistError: if tag with same name already exists | |||
|
273 | """ | |||
|
274 | if name in self.tags: | |||
|
275 | raise TagAlreadyExistError("Tag %s already exists" % name) | |||
|
276 | changeset = self.get_changeset(revision) | |||
|
277 | message = message or "Added tag %s for commit %s" % (name, | |||
|
278 | changeset.raw_id) | |||
|
279 | self._repo.refs["refs/tags/%s" % name] = changeset._commit.id | |||
|
280 | ||||
|
281 | self.tags = self._get_tags() | |||
|
282 | return changeset | |||
|
283 | ||||
|
284 | def remove_tag(self, name, user, message=None, date=None): | |||
|
285 | """ | |||
|
286 | Removes tag with the given ``name``. | |||
|
287 | ||||
|
288 | :param name: name of the tag to be removed | |||
|
289 | :param user: full username, i.e.: "Joe Doe <joe.doe@example.com>" | |||
|
290 | :param message: message of the tag's removal commit | |||
|
291 | :param date: date of tag's removal commit | |||
|
292 | ||||
|
293 | :raises TagDoesNotExistError: if tag with given name does not exists | |||
|
294 | """ | |||
|
295 | if name not in self.tags: | |||
|
296 | raise TagDoesNotExistError("Tag %s does not exist" % name) | |||
|
297 | tagpath = posixpath.join(self._repo.refs.path, 'refs', 'tags', name) | |||
|
298 | try: | |||
|
299 | os.remove(tagpath) | |||
|
300 | self.tags = self._get_tags() | |||
|
301 | except OSError, e: | |||
|
302 | raise RepositoryError(e.strerror) | |||
|
303 | ||||
|
304 | def get_changeset(self, revision=None): | |||
|
305 | """ | |||
|
306 | Returns ``GitChangeset`` object representing commit from git repository | |||
|
307 | at the given revision or head (most recent commit) if None given. | |||
|
308 | """ | |||
|
309 | if isinstance(revision, GitChangeset): | |||
|
310 | return revision | |||
|
311 | revision = self._get_revision(revision) | |||
|
312 | changeset = GitChangeset(repository=self, revision=revision) | |||
|
313 | return changeset | |||
|
314 | ||||
|
315 | def get_changesets(self, start=None, end=None, start_date=None, | |||
|
316 | end_date=None, branch_name=None, reverse=False): | |||
|
317 | """ | |||
|
318 | Returns iterator of ``GitChangeset`` objects from start to end (both | |||
|
319 | are inclusive), in ascending date order (unless ``reverse`` is set). | |||
|
320 | ||||
|
321 | :param start: changeset ID, as str; first returned changeset | |||
|
322 | :param end: changeset ID, as str; last returned changeset | |||
|
323 | :param start_date: if specified, changesets with commit date less than | |||
|
324 | ``start_date`` would be filtered out from returned set | |||
|
325 | :param end_date: if specified, changesets with commit date greater than | |||
|
326 | ``end_date`` would be filtered out from returned set | |||
|
327 | :param branch_name: if specified, changesets not reachable from given | |||
|
328 | branch would be filtered out from returned set | |||
|
329 | :param reverse: if ``True``, returned generator would be reversed | |||
|
330 | (meaning that returned changesets would have descending date order) | |||
|
331 | ||||
|
332 | :raise BranchDoesNotExistError: If given ``branch_name`` does not | |||
|
333 | exist. | |||
|
334 | :raise ChangesetDoesNotExistError: If changeset for given ``start`` or | |||
|
335 | ``end`` could not be found. | |||
|
336 | ||||
|
337 | """ | |||
|
338 | if branch_name and branch_name not in self.branches: | |||
|
339 | raise BranchDoesNotExistError("Branch '%s' not found" \ | |||
|
340 | % branch_name) | |||
|
341 | # %H at format means (full) commit hash, initial hashes are retrieved | |||
|
342 | # in ascending date order | |||
|
343 | cmd_template = 'log --date-order --reverse --pretty=format:"%H"' | |||
|
344 | cmd_params = {} | |||
|
345 | if start_date: | |||
|
346 | cmd_template += ' --since "$since"' | |||
|
347 | cmd_params['since'] = start_date.strftime('%m/%d/%y %H:%M:%S') | |||
|
348 | if end_date: | |||
|
349 | cmd_template += ' --until "$until"' | |||
|
350 | cmd_params['until'] = end_date.strftime('%m/%d/%y %H:%M:%S') | |||
|
351 | if branch_name: | |||
|
352 | cmd_template += ' $branch_name' | |||
|
353 | cmd_params['branch_name'] = branch_name | |||
|
354 | else: | |||
|
355 | cmd_template += ' --all' | |||
|
356 | ||||
|
357 | cmd = Template(cmd_template).safe_substitute(**cmd_params) | |||
|
358 | revs = self.run_git_command(cmd)[0].splitlines() | |||
|
359 | start_pos = 0 | |||
|
360 | end_pos = len(revs) | |||
|
361 | if start: | |||
|
362 | _start = self._get_revision(start) | |||
|
363 | try: | |||
|
364 | start_pos = revs.index(_start) | |||
|
365 | except ValueError: | |||
|
366 | pass | |||
|
367 | ||||
|
368 | if end is not None: | |||
|
369 | _end = self._get_revision(end) | |||
|
370 | try: | |||
|
371 | end_pos = revs.index(_end) | |||
|
372 | except ValueError: | |||
|
373 | pass | |||
|
374 | ||||
|
375 | if None not in [start, end] and start_pos > end_pos: | |||
|
376 | raise RepositoryError('start cannot be after end') | |||
|
377 | ||||
|
378 | if end_pos is not None: | |||
|
379 | end_pos += 1 | |||
|
380 | ||||
|
381 | revs = revs[start_pos:end_pos] | |||
|
382 | if reverse: | |||
|
383 | revs = reversed(revs) | |||
|
384 | for rev in revs: | |||
|
385 | yield self.get_changeset(rev) | |||
|
386 | ||||
|
387 | def get_diff(self, rev1, rev2, path=None, ignore_whitespace=False, | |||
|
388 | context=3): | |||
|
389 | """ | |||
|
390 | Returns (git like) *diff*, as plain text. Shows changes introduced by | |||
|
391 | ``rev2`` since ``rev1``. | |||
|
392 | ||||
|
393 | :param rev1: Entry point from which diff is shown. Can be | |||
|
394 | ``self.EMPTY_CHANGESET`` - in this case, patch showing all | |||
|
395 | the changes since empty state of the repository until ``rev2`` | |||
|
396 | :param rev2: Until which revision changes should be shown. | |||
|
397 | :param ignore_whitespace: If set to ``True``, would not show whitespace | |||
|
398 | changes. Defaults to ``False``. | |||
|
399 | :param context: How many lines before/after changed lines should be | |||
|
400 | shown. Defaults to ``3``. | |||
|
401 | """ | |||
|
402 | flags = ['-U%s' % context] | |||
|
403 | if ignore_whitespace: | |||
|
404 | flags.append('-w') | |||
|
405 | ||||
|
406 | if rev1 == self.EMPTY_CHANGESET: | |||
|
407 | rev2 = self.get_changeset(rev2).raw_id | |||
|
408 | cmd = ' '.join(['show'] + flags + [rev2]) | |||
|
409 | else: | |||
|
410 | rev1 = self.get_changeset(rev1).raw_id | |||
|
411 | rev2 = self.get_changeset(rev2).raw_id | |||
|
412 | cmd = ' '.join(['diff'] + flags + [rev1, rev2]) | |||
|
413 | ||||
|
414 | if path: | |||
|
415 | cmd += ' -- "%s"' % path | |||
|
416 | stdout, stderr = self.run_git_command(cmd) | |||
|
417 | # If we used 'show' command, strip first few lines (until actual diff | |||
|
418 | # starts) | |||
|
419 | if rev1 == self.EMPTY_CHANGESET: | |||
|
420 | lines = stdout.splitlines() | |||
|
421 | x = 0 | |||
|
422 | for line in lines: | |||
|
423 | if line.startswith('diff'): | |||
|
424 | break | |||
|
425 | x += 1 | |||
|
426 | # Append new line just like 'diff' command do | |||
|
427 | stdout = '\n'.join(lines[x:]) + '\n' | |||
|
428 | return stdout | |||
|
429 | ||||
|
430 | @LazyProperty | |||
|
431 | def in_memory_changeset(self): | |||
|
432 | """ | |||
|
433 | Returns ``GitInMemoryChangeset`` object for this repository. | |||
|
434 | """ | |||
|
435 | return GitInMemoryChangeset(self) | |||
|
436 | ||||
|
437 | def clone(self, url, update_after_clone=True, bare=False): | |||
|
438 | """ | |||
|
439 | Tries to clone changes from external location. | |||
|
440 | ||||
|
441 | :param update_after_clone: If set to ``False``, git won't checkout | |||
|
442 | working directory | |||
|
443 | :param bare: If set to ``True``, repository would be cloned into | |||
|
444 | *bare* git repository (no working directory at all). | |||
|
445 | """ | |||
|
446 | url = self._get_url(url) | |||
|
447 | cmd = ['clone'] | |||
|
448 | if bare: | |||
|
449 | cmd.append('--bare') | |||
|
450 | elif not update_after_clone: | |||
|
451 | cmd.append('--no-checkout') | |||
|
452 | cmd += ['--', '"%s"' % url, '"%s"' % self.path] | |||
|
453 | cmd = ' '.join(cmd) | |||
|
454 | # If error occurs run_git_command raises RepositoryError already | |||
|
455 | self.run_git_command(cmd) | |||
|
456 | ||||
|
457 | @LazyProperty | |||
|
458 | def workdir(self): | |||
|
459 | """ | |||
|
460 | Returns ``Workdir`` instance for this repository. | |||
|
461 | """ | |||
|
462 | return GitWorkdir(self) | |||
|
463 | ||||
|
464 | def get_config_value(self, section, name, config_file=None): | |||
|
465 | """ | |||
|
466 | Returns configuration value for a given [``section``] and ``name``. | |||
|
467 | ||||
|
468 | :param section: Section we want to retrieve value from | |||
|
469 | :param name: Name of configuration we want to retrieve | |||
|
470 | :param config_file: A path to file which should be used to retrieve | |||
|
471 | configuration from (might also be a list of file paths) | |||
|
472 | """ | |||
|
473 | if config_file is None: | |||
|
474 | config_file = [] | |||
|
475 | elif isinstance(config_file, basestring): | |||
|
476 | config_file = [config_file] | |||
|
477 | ||||
|
478 | def gen_configs(): | |||
|
479 | for path in config_file + self._config_files: | |||
|
480 | try: | |||
|
481 | yield ConfigFile.from_path(path) | |||
|
482 | except (IOError, OSError, ValueError): | |||
|
483 | continue | |||
|
484 | ||||
|
485 | for config in gen_configs(): | |||
|
486 | try: | |||
|
487 | return config.get(section, name) | |||
|
488 | except KeyError: | |||
|
489 | continue | |||
|
490 | return None | |||
|
491 | ||||
|
492 | def get_user_name(self, config_file=None): | |||
|
493 | """ | |||
|
494 | Returns user's name from global configuration file. | |||
|
495 | ||||
|
496 | :param config_file: A path to file which should be used to retrieve | |||
|
497 | configuration from (might also be a list of file paths) | |||
|
498 | """ | |||
|
499 | return self.get_config_value('user', 'name', config_file) | |||
|
500 | ||||
|
501 | def get_user_email(self, config_file=None): | |||
|
502 | """ | |||
|
503 | Returns user's email from global configuration file. | |||
|
504 | ||||
|
505 | :param config_file: A path to file which should be used to retrieve | |||
|
506 | configuration from (might also be a list of file paths) | |||
|
507 | """ | |||
|
508 | return self.get_config_value('user', 'email', config_file) |
@@ -0,0 +1,31 b'' | |||||
|
1 | import re | |||
|
2 | from rhodecode.lib.vcs.backends.base import BaseWorkdir | |||
|
3 | from rhodecode.lib.vcs.exceptions import RepositoryError | |||
|
4 | from rhodecode.lib.vcs.exceptions import BranchDoesNotExistError | |||
|
5 | ||||
|
6 | ||||
|
7 | class GitWorkdir(BaseWorkdir): | |||
|
8 | ||||
|
9 | def get_branch(self): | |||
|
10 | headpath = self.repository._repo.refs.refpath('HEAD') | |||
|
11 | try: | |||
|
12 | content = open(headpath).read() | |||
|
13 | match = re.match(r'^ref: refs/heads/(?P<branch>.+)\n$', content) | |||
|
14 | if match: | |||
|
15 | return match.groupdict()['branch'] | |||
|
16 | else: | |||
|
17 | raise RepositoryError("Couldn't compute workdir's branch") | |||
|
18 | except IOError: | |||
|
19 | # Try naive way... | |||
|
20 | raise RepositoryError("Couldn't compute workdir's branch") | |||
|
21 | ||||
|
22 | def get_changeset(self): | |||
|
23 | return self.repository.get_changeset( | |||
|
24 | self.repository._repo.refs.as_dict().get('HEAD')) | |||
|
25 | ||||
|
26 | def checkout_branch(self, branch=None): | |||
|
27 | if branch is None: | |||
|
28 | branch = self.repository.DEFAULT_BRANCH_NAME | |||
|
29 | if branch not in self.repository.branches: | |||
|
30 | raise BranchDoesNotExistError | |||
|
31 | self.repository.run_git_command(['checkout', branch]) |
@@ -0,0 +1,21 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | """ | |||
|
3 | vcs.backends.hg | |||
|
4 | ~~~~~~~~~~~~~~~~ | |||
|
5 | ||||
|
6 | Mercurial backend implementation. | |||
|
7 | ||||
|
8 | :created_on: Apr 8, 2010 | |||
|
9 | :copyright: (c) 2010-2011 by Marcin Kuzminski, Lukasz Balcerzak. | |||
|
10 | """ | |||
|
11 | ||||
|
12 | from .repository import MercurialRepository | |||
|
13 | from .changeset import MercurialChangeset | |||
|
14 | from .inmemory import MercurialInMemoryChangeset | |||
|
15 | from .workdir import MercurialWorkdir | |||
|
16 | ||||
|
17 | ||||
|
18 | __all__ = [ | |||
|
19 | 'MercurialRepository', 'MercurialChangeset', | |||
|
20 | 'MercurialInMemoryChangeset', 'MercurialWorkdir', | |||
|
21 | ] |
@@ -0,0 +1,338 b'' | |||||
|
1 | import os | |||
|
2 | import posixpath | |||
|
3 | ||||
|
4 | from rhodecode.lib.vcs.backends.base import BaseChangeset | |||
|
5 | from rhodecode.lib.vcs.conf import settings | |||
|
6 | from rhodecode.lib.vcs.exceptions import ChangesetDoesNotExistError, \ | |||
|
7 | ChangesetError, ImproperArchiveTypeError, NodeDoesNotExistError, VCSError | |||
|
8 | from rhodecode.lib.vcs.nodes import AddedFileNodesGenerator, ChangedFileNodesGenerator, \ | |||
|
9 | DirNode, FileNode, NodeKind, RemovedFileNodesGenerator, RootNode | |||
|
10 | ||||
|
11 | from rhodecode.lib.vcs.utils import safe_str, safe_unicode, date_fromtimestamp | |||
|
12 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |||
|
13 | from rhodecode.lib.vcs.utils.paths import get_dirs_for_path | |||
|
14 | ||||
|
15 | from ...utils.hgcompat import archival, hex | |||
|
16 | ||||
|
17 | ||||
|
18 | class MercurialChangeset(BaseChangeset): | |||
|
19 | """ | |||
|
20 | Represents state of the repository at the single revision. | |||
|
21 | """ | |||
|
22 | ||||
|
23 | def __init__(self, repository, revision): | |||
|
24 | self.repository = repository | |||
|
25 | self.raw_id = revision | |||
|
26 | self._ctx = repository._repo[revision] | |||
|
27 | self.revision = self._ctx._rev | |||
|
28 | self.nodes = {} | |||
|
29 | ||||
|
30 | @LazyProperty | |||
|
31 | def tags(self): | |||
|
32 | return map(safe_unicode, self._ctx.tags()) | |||
|
33 | ||||
|
34 | @LazyProperty | |||
|
35 | def branch(self): | |||
|
36 | return safe_unicode(self._ctx.branch()) | |||
|
37 | ||||
|
38 | @LazyProperty | |||
|
39 | def message(self): | |||
|
40 | return safe_unicode(self._ctx.description()) | |||
|
41 | ||||
|
42 | @LazyProperty | |||
|
43 | def author(self): | |||
|
44 | return safe_unicode(self._ctx.user()) | |||
|
45 | ||||
|
46 | @LazyProperty | |||
|
47 | def date(self): | |||
|
48 | return date_fromtimestamp(*self._ctx.date()) | |||
|
49 | ||||
|
50 | @LazyProperty | |||
|
51 | def status(self): | |||
|
52 | """ | |||
|
53 | Returns modified, added, removed, deleted files for current changeset | |||
|
54 | """ | |||
|
55 | return self.repository._repo.status(self._ctx.p1().node(), | |||
|
56 | self._ctx.node()) | |||
|
57 | ||||
|
58 | @LazyProperty | |||
|
59 | def _file_paths(self): | |||
|
60 | return list(self._ctx) | |||
|
61 | ||||
|
62 | @LazyProperty | |||
|
63 | def _dir_paths(self): | |||
|
64 | p = list(set(get_dirs_for_path(*self._file_paths))) | |||
|
65 | p.insert(0, '') | |||
|
66 | return p | |||
|
67 | ||||
|
68 | @LazyProperty | |||
|
69 | def _paths(self): | |||
|
70 | return self._dir_paths + self._file_paths | |||
|
71 | ||||
|
72 | @LazyProperty | |||
|
73 | def id(self): | |||
|
74 | if self.last: | |||
|
75 | return u'tip' | |||
|
76 | return self.short_id | |||
|
77 | ||||
|
78 | @LazyProperty | |||
|
79 | def short_id(self): | |||
|
80 | return self.raw_id[:12] | |||
|
81 | ||||
|
82 | @LazyProperty | |||
|
83 | def parents(self): | |||
|
84 | """ | |||
|
85 | Returns list of parents changesets. | |||
|
86 | """ | |||
|
87 | return [self.repository.get_changeset(parent.rev()) | |||
|
88 | for parent in self._ctx.parents() if parent.rev() >= 0] | |||
|
89 | ||||
|
90 | def next(self, branch=None): | |||
|
91 | ||||
|
92 | if branch and self.branch != branch: | |||
|
93 | raise VCSError('Branch option used on changeset not belonging ' | |||
|
94 | 'to that branch') | |||
|
95 | ||||
|
96 | def _next(changeset, branch): | |||
|
97 | try: | |||
|
98 | next_ = changeset.revision + 1 | |||
|
99 | next_rev = changeset.repository.revisions[next_] | |||
|
100 | except IndexError: | |||
|
101 | raise ChangesetDoesNotExistError | |||
|
102 | cs = changeset.repository.get_changeset(next_rev) | |||
|
103 | ||||
|
104 | if branch and branch != cs.branch: | |||
|
105 | return _next(cs, branch) | |||
|
106 | ||||
|
107 | return cs | |||
|
108 | ||||
|
109 | return _next(self, branch) | |||
|
110 | ||||
|
111 | def prev(self, branch=None): | |||
|
112 | if branch and self.branch != branch: | |||
|
113 | raise VCSError('Branch option used on changeset not belonging ' | |||
|
114 | 'to that branch') | |||
|
115 | ||||
|
116 | def _prev(changeset, branch): | |||
|
117 | try: | |||
|
118 | prev_ = changeset.revision - 1 | |||
|
119 | if prev_ < 0: | |||
|
120 | raise IndexError | |||
|
121 | prev_rev = changeset.repository.revisions[prev_] | |||
|
122 | except IndexError: | |||
|
123 | raise ChangesetDoesNotExistError | |||
|
124 | ||||
|
125 | cs = changeset.repository.get_changeset(prev_rev) | |||
|
126 | ||||
|
127 | if branch and branch != cs.branch: | |||
|
128 | return _prev(cs, branch) | |||
|
129 | ||||
|
130 | return cs | |||
|
131 | ||||
|
132 | return _prev(self, branch) | |||
|
133 | ||||
|
134 | def _fix_path(self, path): | |||
|
135 | """ | |||
|
136 | Paths are stored without trailing slash so we need to get rid off it if | |||
|
137 | needed. Also mercurial keeps filenodes as str so we need to decode | |||
|
138 | from unicode to str | |||
|
139 | """ | |||
|
140 | if path.endswith('/'): | |||
|
141 | path = path.rstrip('/') | |||
|
142 | ||||
|
143 | return safe_str(path) | |||
|
144 | ||||
|
145 | def _get_kind(self, path): | |||
|
146 | path = self._fix_path(path) | |||
|
147 | if path in self._file_paths: | |||
|
148 | return NodeKind.FILE | |||
|
149 | elif path in self._dir_paths: | |||
|
150 | return NodeKind.DIR | |||
|
151 | else: | |||
|
152 | raise ChangesetError("Node does not exist at the given path %r" | |||
|
153 | % (path)) | |||
|
154 | ||||
|
155 | def _get_filectx(self, path): | |||
|
156 | path = self._fix_path(path) | |||
|
157 | if self._get_kind(path) != NodeKind.FILE: | |||
|
158 | raise ChangesetError("File does not exist for revision %r at " | |||
|
159 | " %r" % (self.revision, path)) | |||
|
160 | return self._ctx.filectx(path) | |||
|
161 | ||||
|
162 | def get_file_mode(self, path): | |||
|
163 | """ | |||
|
164 | Returns stat mode of the file at the given ``path``. | |||
|
165 | """ | |||
|
166 | fctx = self._get_filectx(path) | |||
|
167 | if 'x' in fctx.flags(): | |||
|
168 | return 0100755 | |||
|
169 | else: | |||
|
170 | return 0100644 | |||
|
171 | ||||
|
172 | def get_file_content(self, path): | |||
|
173 | """ | |||
|
174 | Returns content of the file at given ``path``. | |||
|
175 | """ | |||
|
176 | fctx = self._get_filectx(path) | |||
|
177 | return fctx.data() | |||
|
178 | ||||
|
179 | def get_file_size(self, path): | |||
|
180 | """ | |||
|
181 | Returns size of the file at given ``path``. | |||
|
182 | """ | |||
|
183 | fctx = self._get_filectx(path) | |||
|
184 | return fctx.size() | |||
|
185 | ||||
|
186 | def get_file_changeset(self, path): | |||
|
187 | """ | |||
|
188 | Returns last commit of the file at the given ``path``. | |||
|
189 | """ | |||
|
190 | fctx = self._get_filectx(path) | |||
|
191 | changeset = self.repository.get_changeset(fctx.linkrev()) | |||
|
192 | return changeset | |||
|
193 | ||||
|
194 | def get_file_history(self, path): | |||
|
195 | """ | |||
|
196 | Returns history of file as reversed list of ``Changeset`` objects for | |||
|
197 | which file at given ``path`` has been modified. | |||
|
198 | """ | |||
|
199 | fctx = self._get_filectx(path) | |||
|
200 | nodes = [fctx.filectx(x).node() for x in fctx.filelog()] | |||
|
201 | changesets = [self.repository.get_changeset(hex(node)) | |||
|
202 | for node in reversed(nodes)] | |||
|
203 | return changesets | |||
|
204 | ||||
|
205 | def get_file_annotate(self, path): | |||
|
206 | """ | |||
|
207 | Returns a list of three element tuples with lineno,changeset and line | |||
|
208 | """ | |||
|
209 | fctx = self._get_filectx(path) | |||
|
210 | annotate = [] | |||
|
211 | for i, annotate_data in enumerate(fctx.annotate()): | |||
|
212 | ln_no = i + 1 | |||
|
213 | annotate.append((ln_no, self.repository\ | |||
|
214 | .get_changeset(hex(annotate_data[0].node())), | |||
|
215 | annotate_data[1],)) | |||
|
216 | ||||
|
217 | return annotate | |||
|
218 | ||||
|
219 | def fill_archive(self, stream=None, kind='tgz', prefix=None, | |||
|
220 | subrepos=False): | |||
|
221 | """ | |||
|
222 | Fills up given stream. | |||
|
223 | ||||
|
224 | :param stream: file like object. | |||
|
225 | :param kind: one of following: ``zip``, ``tgz`` or ``tbz2``. | |||
|
226 | Default: ``tgz``. | |||
|
227 | :param prefix: name of root directory in archive. | |||
|
228 | Default is repository name and changeset's raw_id joined with dash | |||
|
229 | (``repo-tip.<KIND>``). | |||
|
230 | :param subrepos: include subrepos in this archive. | |||
|
231 | ||||
|
232 | :raise ImproperArchiveTypeError: If given kind is wrong. | |||
|
233 | :raise VcsError: If given stream is None | |||
|
234 | """ | |||
|
235 | ||||
|
236 | allowed_kinds = settings.ARCHIVE_SPECS.keys() | |||
|
237 | if kind not in allowed_kinds: | |||
|
238 | raise ImproperArchiveTypeError('Archive kind not supported use one' | |||
|
239 | 'of %s', allowed_kinds) | |||
|
240 | ||||
|
241 | if stream is None: | |||
|
242 | raise VCSError('You need to pass in a valid stream for filling' | |||
|
243 | ' with archival data') | |||
|
244 | ||||
|
245 | if prefix is None: | |||
|
246 | prefix = '%s-%s' % (self.repository.name, self.short_id) | |||
|
247 | elif prefix.startswith('/'): | |||
|
248 | raise VCSError("Prefix cannot start with leading slash") | |||
|
249 | elif prefix.strip() == '': | |||
|
250 | raise VCSError("Prefix cannot be empty") | |||
|
251 | ||||
|
252 | archival.archive(self.repository._repo, stream, self.raw_id, | |||
|
253 | kind, prefix=prefix, subrepos=subrepos) | |||
|
254 | ||||
|
255 | #stream.close() | |||
|
256 | ||||
|
257 | if stream.closed and hasattr(stream, 'name'): | |||
|
258 | stream = open(stream.name, 'rb') | |||
|
259 | elif hasattr(stream, 'mode') and 'r' not in stream.mode: | |||
|
260 | stream = open(stream.name, 'rb') | |||
|
261 | else: | |||
|
262 | stream.seek(0) | |||
|
263 | ||||
|
264 | def get_nodes(self, path): | |||
|
265 | """ | |||
|
266 | Returns combined ``DirNode`` and ``FileNode`` objects list representing | |||
|
267 | state of changeset at the given ``path``. If node at the given ``path`` | |||
|
268 | is not instance of ``DirNode``, ChangesetError would be raised. | |||
|
269 | """ | |||
|
270 | ||||
|
271 | if self._get_kind(path) != NodeKind.DIR: | |||
|
272 | raise ChangesetError("Directory does not exist for revision %r at " | |||
|
273 | " %r" % (self.revision, path)) | |||
|
274 | path = self._fix_path(path) | |||
|
275 | filenodes = [FileNode(f, changeset=self) for f in self._file_paths | |||
|
276 | if os.path.dirname(f) == path] | |||
|
277 | dirs = path == '' and '' or [d for d in self._dir_paths | |||
|
278 | if d and posixpath.dirname(d) == path] | |||
|
279 | dirnodes = [DirNode(d, changeset=self) for d in dirs | |||
|
280 | if os.path.dirname(d) == path] | |||
|
281 | nodes = dirnodes + filenodes | |||
|
282 | # cache nodes | |||
|
283 | for node in nodes: | |||
|
284 | self.nodes[node.path] = node | |||
|
285 | nodes.sort() | |||
|
286 | return nodes | |||
|
287 | ||||
|
288 | def get_node(self, path): | |||
|
289 | """ | |||
|
290 | Returns ``Node`` object from the given ``path``. If there is no node at | |||
|
291 | the given ``path``, ``ChangesetError`` would be raised. | |||
|
292 | """ | |||
|
293 | ||||
|
294 | path = self._fix_path(path) | |||
|
295 | ||||
|
296 | if not path in self.nodes: | |||
|
297 | if path in self._file_paths: | |||
|
298 | node = FileNode(path, changeset=self) | |||
|
299 | elif path in self._dir_paths or path in self._dir_paths: | |||
|
300 | if path == '': | |||
|
301 | node = RootNode(changeset=self) | |||
|
302 | else: | |||
|
303 | node = DirNode(path, changeset=self) | |||
|
304 | else: | |||
|
305 | raise NodeDoesNotExistError("There is no file nor directory " | |||
|
306 | "at the given path: %r at revision %r" | |||
|
307 | % (path, self.short_id)) | |||
|
308 | # cache node | |||
|
309 | self.nodes[path] = node | |||
|
310 | return self.nodes[path] | |||
|
311 | ||||
|
312 | @LazyProperty | |||
|
313 | def affected_files(self): | |||
|
314 | """ | |||
|
315 | Get's a fast accessible file changes for given changeset | |||
|
316 | """ | |||
|
317 | return self._ctx.files() | |||
|
318 | ||||
|
319 | @property | |||
|
320 | def added(self): | |||
|
321 | """ | |||
|
322 | Returns list of added ``FileNode`` objects. | |||
|
323 | """ | |||
|
324 | return AddedFileNodesGenerator([n for n in self.status[1]], self) | |||
|
325 | ||||
|
326 | @property | |||
|
327 | def changed(self): | |||
|
328 | """ | |||
|
329 | Returns list of modified ``FileNode`` objects. | |||
|
330 | """ | |||
|
331 | return ChangedFileNodesGenerator([n for n in self.status[0]], self) | |||
|
332 | ||||
|
333 | @property | |||
|
334 | def removed(self): | |||
|
335 | """ | |||
|
336 | Returns list of removed ``FileNode`` objects. | |||
|
337 | """ | |||
|
338 | return RemovedFileNodesGenerator([n for n in self.status[2]], self) |
@@ -0,0 +1,110 b'' | |||||
|
1 | import datetime | |||
|
2 | import errno | |||
|
3 | ||||
|
4 | from rhodecode.lib.vcs.backends.base import BaseInMemoryChangeset | |||
|
5 | from rhodecode.lib.vcs.exceptions import RepositoryError | |||
|
6 | ||||
|
7 | from ...utils.hgcompat import memfilectx, memctx, hex | |||
|
8 | ||||
|
9 | ||||
|
10 | class MercurialInMemoryChangeset(BaseInMemoryChangeset): | |||
|
11 | ||||
|
12 | def commit(self, message, author, parents=None, branch=None, date=None, | |||
|
13 | **kwargs): | |||
|
14 | """ | |||
|
15 | Performs in-memory commit (doesn't check workdir in any way) and | |||
|
16 | returns newly created ``Changeset``. Updates repository's | |||
|
17 | ``revisions``. | |||
|
18 | ||||
|
19 | :param message: message of the commit | |||
|
20 | :param author: full username, i.e. "Joe Doe <joe.doe@example.com>" | |||
|
21 | :param parents: single parent or sequence of parents from which commit | |||
|
22 | would be derieved | |||
|
23 | :param date: ``datetime.datetime`` instance. Defaults to | |||
|
24 | ``datetime.datetime.now()``. | |||
|
25 | :param branch: branch name, as string. If none given, default backend's | |||
|
26 | branch would be used. | |||
|
27 | ||||
|
28 | :raises ``CommitError``: if any error occurs while committing | |||
|
29 | """ | |||
|
30 | self.check_integrity(parents) | |||
|
31 | ||||
|
32 | from .repository import MercurialRepository | |||
|
33 | if not isinstance(message, str) or not isinstance(author, str): | |||
|
34 | raise RepositoryError('Given message and author needs to be ' | |||
|
35 | 'an <str> instance') | |||
|
36 | ||||
|
37 | if branch is None: | |||
|
38 | branch = MercurialRepository.DEFAULT_BRANCH_NAME | |||
|
39 | kwargs['branch'] = branch | |||
|
40 | ||||
|
41 | def filectxfn(_repo, memctx, path): | |||
|
42 | """ | |||
|
43 | Marks given path as added/changed/removed in a given _repo. This is | |||
|
44 | for internal mercurial commit function. | |||
|
45 | """ | |||
|
46 | ||||
|
47 | # check if this path is removed | |||
|
48 | if path in (node.path for node in self.removed): | |||
|
49 | # Raising exception is a way to mark node for removal | |||
|
50 | raise IOError(errno.ENOENT, '%s is deleted' % path) | |||
|
51 | ||||
|
52 | # check if this path is added | |||
|
53 | for node in self.added: | |||
|
54 | if node.path == path: | |||
|
55 | return memfilectx(path=node.path, | |||
|
56 | data=(node.content.encode('utf8') | |||
|
57 | if not node.is_binary else node.content), | |||
|
58 | islink=False, | |||
|
59 | isexec=node.is_executable, | |||
|
60 | copied=False) | |||
|
61 | ||||
|
62 | # or changed | |||
|
63 | for node in self.changed: | |||
|
64 | if node.path == path: | |||
|
65 | return memfilectx(path=node.path, | |||
|
66 | data=(node.content.encode('utf8') | |||
|
67 | if not node.is_binary else node.content), | |||
|
68 | islink=False, | |||
|
69 | isexec=node.is_executable, | |||
|
70 | copied=False) | |||
|
71 | ||||
|
72 | raise RepositoryError("Given path haven't been marked as added," | |||
|
73 | "changed or removed (%s)" % path) | |||
|
74 | ||||
|
75 | parents = [None, None] | |||
|
76 | for i, parent in enumerate(self.parents): | |||
|
77 | if parent is not None: | |||
|
78 | parents[i] = parent._ctx.node() | |||
|
79 | ||||
|
80 | if date and isinstance(date, datetime.datetime): | |||
|
81 | date = date.ctime() | |||
|
82 | ||||
|
83 | commit_ctx = memctx(repo=self.repository._repo, | |||
|
84 | parents=parents, | |||
|
85 | text='', | |||
|
86 | files=self.get_paths(), | |||
|
87 | filectxfn=filectxfn, | |||
|
88 | user=author, | |||
|
89 | date=date, | |||
|
90 | extra=kwargs) | |||
|
91 | ||||
|
92 | # injecting given _repo params | |||
|
93 | commit_ctx._text = message | |||
|
94 | commit_ctx._user = author | |||
|
95 | commit_ctx._date = date | |||
|
96 | ||||
|
97 | # TODO: Catch exceptions! | |||
|
98 | n = self.repository._repo.commitctx(commit_ctx) | |||
|
99 | # Returns mercurial node | |||
|
100 | self._commit_ctx = commit_ctx # For reference | |||
|
101 | # Update vcs repository object & recreate mercurial _repo | |||
|
102 | # new_ctx = self.repository._repo[node] | |||
|
103 | # new_tip = self.repository.get_changeset(new_ctx.hex()) | |||
|
104 | new_id = hex(n) | |||
|
105 | self.repository.revisions.append(new_id) | |||
|
106 | self._repo = self.repository._get_repo(create=False) | |||
|
107 | self.repository.branches = self.repository._get_branches() | |||
|
108 | tip = self.repository.get_changeset() | |||
|
109 | self.reset() | |||
|
110 | return tip |
This diff has been collapsed as it changes many lines, (521 lines changed) Show them Hide them | |||||
@@ -0,0 +1,521 b'' | |||||
|
1 | import os | |||
|
2 | import time | |||
|
3 | import datetime | |||
|
4 | import urllib | |||
|
5 | import urllib2 | |||
|
6 | ||||
|
7 | from rhodecode.lib.vcs.backends.base import BaseRepository | |||
|
8 | from .workdir import MercurialWorkdir | |||
|
9 | from .changeset import MercurialChangeset | |||
|
10 | from .inmemory import MercurialInMemoryChangeset | |||
|
11 | ||||
|
12 | from rhodecode.lib.vcs.exceptions import BranchDoesNotExistError, \ | |||
|
13 | ChangesetDoesNotExistError, EmptyRepositoryError, RepositoryError, \ | |||
|
14 | VCSError, TagAlreadyExistError, TagDoesNotExistError | |||
|
15 | from rhodecode.lib.vcs.utils import author_email, author_name, date_fromtimestamp, \ | |||
|
16 | makedate, safe_unicode | |||
|
17 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |||
|
18 | from rhodecode.lib.vcs.utils.ordered_dict import OrderedDict | |||
|
19 | from rhodecode.lib.vcs.utils.paths import abspath | |||
|
20 | ||||
|
21 | from ...utils.hgcompat import ui, nullid, match, patch, diffopts, clone, \ | |||
|
22 | get_contact, pull, localrepository, RepoLookupError, Abort, RepoError, hex | |||
|
23 | ||||
|
24 | ||||
|
25 | class MercurialRepository(BaseRepository): | |||
|
26 | """ | |||
|
27 | Mercurial repository backend | |||
|
28 | """ | |||
|
29 | DEFAULT_BRANCH_NAME = 'default' | |||
|
30 | scm = 'hg' | |||
|
31 | ||||
|
32 | def __init__(self, repo_path, create=False, baseui=None, src_url=None, | |||
|
33 | update_after_clone=False): | |||
|
34 | """ | |||
|
35 | Raises RepositoryError if repository could not be find at the given | |||
|
36 | ``repo_path``. | |||
|
37 | ||||
|
38 | :param repo_path: local path of the repository | |||
|
39 | :param create=False: if set to True, would try to create repository if | |||
|
40 | it does not exist rather than raising exception | |||
|
41 | :param baseui=None: user data | |||
|
42 | :param src_url=None: would try to clone repository from given location | |||
|
43 | :param update_after_clone=False: sets update of working copy after | |||
|
44 | making a clone | |||
|
45 | """ | |||
|
46 | ||||
|
47 | if not isinstance(repo_path, str): | |||
|
48 | raise VCSError('Mercurial backend requires repository path to ' | |||
|
49 | 'be instance of <str> got %s instead' % | |||
|
50 | type(repo_path)) | |||
|
51 | ||||
|
52 | self.path = abspath(repo_path) | |||
|
53 | self.baseui = baseui or ui.ui() | |||
|
54 | # We've set path and ui, now we can set _repo itself | |||
|
55 | self._repo = self._get_repo(create, src_url, update_after_clone) | |||
|
56 | ||||
|
57 | @property | |||
|
58 | def _empty(self): | |||
|
59 | """ | |||
|
60 | Checks if repository is empty without any changesets | |||
|
61 | """ | |||
|
62 | # TODO: Following raises errors when using InMemoryChangeset... | |||
|
63 | # return len(self._repo.changelog) == 0 | |||
|
64 | return len(self.revisions) == 0 | |||
|
65 | ||||
|
66 | @LazyProperty | |||
|
67 | def revisions(self): | |||
|
68 | """ | |||
|
69 | Returns list of revisions' ids, in ascending order. Being lazy | |||
|
70 | attribute allows external tools to inject shas from cache. | |||
|
71 | """ | |||
|
72 | return self._get_all_revisions() | |||
|
73 | ||||
|
74 | @LazyProperty | |||
|
75 | def name(self): | |||
|
76 | return os.path.basename(self.path) | |||
|
77 | ||||
|
78 | @LazyProperty | |||
|
79 | def branches(self): | |||
|
80 | return self._get_branches() | |||
|
81 | ||||
|
82 | def _get_branches(self, closed=False): | |||
|
83 | """ | |||
|
84 | Get's branches for this repository | |||
|
85 | Returns only not closed branches by default | |||
|
86 | ||||
|
87 | :param closed: return also closed branches for mercurial | |||
|
88 | """ | |||
|
89 | ||||
|
90 | if self._empty: | |||
|
91 | return {} | |||
|
92 | ||||
|
93 | def _branchtags(localrepo): | |||
|
94 | """ | |||
|
95 | Patched version of mercurial branchtags to not return the closed | |||
|
96 | branches | |||
|
97 | ||||
|
98 | :param localrepo: locarepository instance | |||
|
99 | """ | |||
|
100 | ||||
|
101 | bt = {} | |||
|
102 | bt_closed = {} | |||
|
103 | for bn, heads in localrepo.branchmap().iteritems(): | |||
|
104 | tip = heads[-1] | |||
|
105 | if 'close' in localrepo.changelog.read(tip)[5]: | |||
|
106 | bt_closed[bn] = tip | |||
|
107 | else: | |||
|
108 | bt[bn] = tip | |||
|
109 | ||||
|
110 | if closed: | |||
|
111 | bt.update(bt_closed) | |||
|
112 | return bt | |||
|
113 | ||||
|
114 | sortkey = lambda ctx: ctx[0] # sort by name | |||
|
115 | _branches = [(safe_unicode(n), hex(h),) for n, h in | |||
|
116 | _branchtags(self._repo).items()] | |||
|
117 | ||||
|
118 | return OrderedDict(sorted(_branches, key=sortkey, reverse=False)) | |||
|
119 | ||||
|
120 | @LazyProperty | |||
|
121 | def tags(self): | |||
|
122 | """ | |||
|
123 | Get's tags for this repository | |||
|
124 | """ | |||
|
125 | return self._get_tags() | |||
|
126 | ||||
|
127 | def _get_tags(self): | |||
|
128 | if self._empty: | |||
|
129 | return {} | |||
|
130 | ||||
|
131 | sortkey = lambda ctx: ctx[0] # sort by name | |||
|
132 | _tags = [(safe_unicode(n), hex(h),) for n, h in | |||
|
133 | self._repo.tags().items()] | |||
|
134 | ||||
|
135 | return OrderedDict(sorted(_tags, key=sortkey, reverse=True)) | |||
|
136 | ||||
|
137 | def tag(self, name, user, revision=None, message=None, date=None, | |||
|
138 | **kwargs): | |||
|
139 | """ | |||
|
140 | Creates and returns a tag for the given ``revision``. | |||
|
141 | ||||
|
142 | :param name: name for new tag | |||
|
143 | :param user: full username, i.e.: "Joe Doe <joe.doe@example.com>" | |||
|
144 | :param revision: changeset id for which new tag would be created | |||
|
145 | :param message: message of the tag's commit | |||
|
146 | :param date: date of tag's commit | |||
|
147 | ||||
|
148 | :raises TagAlreadyExistError: if tag with same name already exists | |||
|
149 | """ | |||
|
150 | if name in self.tags: | |||
|
151 | raise TagAlreadyExistError("Tag %s already exists" % name) | |||
|
152 | changeset = self.get_changeset(revision) | |||
|
153 | local = kwargs.setdefault('local', False) | |||
|
154 | ||||
|
155 | if message is None: | |||
|
156 | message = "Added tag %s for changeset %s" % (name, | |||
|
157 | changeset.short_id) | |||
|
158 | ||||
|
159 | if date is None: | |||
|
160 | date = datetime.datetime.now().ctime() | |||
|
161 | ||||
|
162 | try: | |||
|
163 | self._repo.tag(name, changeset._ctx.node(), message, local, user, | |||
|
164 | date) | |||
|
165 | except Abort, e: | |||
|
166 | raise RepositoryError(e.message) | |||
|
167 | ||||
|
168 | # Reinitialize tags | |||
|
169 | self.tags = self._get_tags() | |||
|
170 | tag_id = self.tags[name] | |||
|
171 | ||||
|
172 | return self.get_changeset(revision=tag_id) | |||
|
173 | ||||
|
174 | def remove_tag(self, name, user, message=None, date=None): | |||
|
175 | """ | |||
|
176 | Removes tag with the given ``name``. | |||
|
177 | ||||
|
178 | :param name: name of the tag to be removed | |||
|
179 | :param user: full username, i.e.: "Joe Doe <joe.doe@example.com>" | |||
|
180 | :param message: message of the tag's removal commit | |||
|
181 | :param date: date of tag's removal commit | |||
|
182 | ||||
|
183 | :raises TagDoesNotExistError: if tag with given name does not exists | |||
|
184 | """ | |||
|
185 | if name not in self.tags: | |||
|
186 | raise TagDoesNotExistError("Tag %s does not exist" % name) | |||
|
187 | if message is None: | |||
|
188 | message = "Removed tag %s" % name | |||
|
189 | if date is None: | |||
|
190 | date = datetime.datetime.now().ctime() | |||
|
191 | local = False | |||
|
192 | ||||
|
193 | try: | |||
|
194 | self._repo.tag(name, nullid, message, local, user, date) | |||
|
195 | self.tags = self._get_tags() | |||
|
196 | except Abort, e: | |||
|
197 | raise RepositoryError(e.message) | |||
|
198 | ||||
|
199 | @LazyProperty | |||
|
200 | def bookmarks(self): | |||
|
201 | """ | |||
|
202 | Get's bookmarks for this repository | |||
|
203 | """ | |||
|
204 | return self._get_bookmarks() | |||
|
205 | ||||
|
206 | def _get_bookmarks(self): | |||
|
207 | if self._empty: | |||
|
208 | return {} | |||
|
209 | ||||
|
210 | sortkey = lambda ctx: ctx[0] # sort by name | |||
|
211 | _bookmarks = [(safe_unicode(n), hex(h),) for n, h in | |||
|
212 | self._repo._bookmarks.items()] | |||
|
213 | return OrderedDict(sorted(_bookmarks, key=sortkey, reverse=True)) | |||
|
214 | ||||
|
215 | def _get_all_revisions(self): | |||
|
216 | ||||
|
217 | return map(lambda x: hex(x[7]), self._repo.changelog.index)[:-1] | |||
|
218 | ||||
|
219 | def get_diff(self, rev1, rev2, path='', ignore_whitespace=False, | |||
|
220 | context=3): | |||
|
221 | """ | |||
|
222 | Returns (git like) *diff*, as plain text. Shows changes introduced by | |||
|
223 | ``rev2`` since ``rev1``. | |||
|
224 | ||||
|
225 | :param rev1: Entry point from which diff is shown. Can be | |||
|
226 | ``self.EMPTY_CHANGESET`` - in this case, patch showing all | |||
|
227 | the changes since empty state of the repository until ``rev2`` | |||
|
228 | :param rev2: Until which revision changes should be shown. | |||
|
229 | :param ignore_whitespace: If set to ``True``, would not show whitespace | |||
|
230 | changes. Defaults to ``False``. | |||
|
231 | :param context: How many lines before/after changed lines should be | |||
|
232 | shown. Defaults to ``3``. | |||
|
233 | """ | |||
|
234 | # Check if given revisions are present at repository (may raise | |||
|
235 | # ChangesetDoesNotExistError) | |||
|
236 | if rev1 != self.EMPTY_CHANGESET: | |||
|
237 | self.get_changeset(rev1) | |||
|
238 | self.get_changeset(rev2) | |||
|
239 | ||||
|
240 | file_filter = match(self.path, '', [path]) | |||
|
241 | return ''.join(patch.diff(self._repo, rev1, rev2, match=file_filter, | |||
|
242 | opts=diffopts(git=True, | |||
|
243 | ignorews=ignore_whitespace, | |||
|
244 | context=context))) | |||
|
245 | ||||
|
246 | def _check_url(self, url): | |||
|
247 | """ | |||
|
248 | Function will check given url and try to verify if it's a valid | |||
|
249 | link. Sometimes it may happened that mercurial will issue basic | |||
|
250 | auth request that can cause whole API to hang when used from python | |||
|
251 | or other external calls. | |||
|
252 | ||||
|
253 | On failures it'll raise urllib2.HTTPError, return code 200 if url | |||
|
254 | is valid or True if it's a local path | |||
|
255 | """ | |||
|
256 | ||||
|
257 | from mercurial.util import url as Url | |||
|
258 | ||||
|
259 | # those authnadlers are patched for python 2.6.5 bug an | |||
|
260 | # infinit looping when given invalid resources | |||
|
261 | from mercurial.url import httpbasicauthhandler, httpdigestauthhandler | |||
|
262 | ||||
|
263 | # check first if it's not an local url | |||
|
264 | if os.path.isdir(url) or url.startswith('file:'): | |||
|
265 | return True | |||
|
266 | ||||
|
267 | handlers = [] | |||
|
268 | test_uri, authinfo = Url(url).authinfo() | |||
|
269 | ||||
|
270 | if authinfo: | |||
|
271 | #create a password manager | |||
|
272 | passmgr = urllib2.HTTPPasswordMgrWithDefaultRealm() | |||
|
273 | passmgr.add_password(*authinfo) | |||
|
274 | ||||
|
275 | handlers.extend((httpbasicauthhandler(passmgr), | |||
|
276 | httpdigestauthhandler(passmgr))) | |||
|
277 | ||||
|
278 | o = urllib2.build_opener(*handlers) | |||
|
279 | o.addheaders = [('Content-Type', 'application/mercurial-0.1'), | |||
|
280 | ('Accept', 'application/mercurial-0.1')] | |||
|
281 | ||||
|
282 | q = {"cmd": 'between'} | |||
|
283 | q.update({'pairs': "%s-%s" % ('0' * 40, '0' * 40)}) | |||
|
284 | qs = '?%s' % urllib.urlencode(q) | |||
|
285 | cu = "%s%s" % (test_uri, qs) | |||
|
286 | req = urllib2.Request(cu, None, {}) | |||
|
287 | ||||
|
288 | try: | |||
|
289 | resp = o.open(req) | |||
|
290 | return resp.code == 200 | |||
|
291 | except Exception, e: | |||
|
292 | # means it cannot be cloned | |||
|
293 | raise urllib2.URLError(e) | |||
|
294 | ||||
|
295 | def _get_repo(self, create, src_url=None, update_after_clone=False): | |||
|
296 | """ | |||
|
297 | Function will check for mercurial repository in given path and return | |||
|
298 | a localrepo object. If there is no repository in that path it will | |||
|
299 | raise an exception unless ``create`` parameter is set to True - in | |||
|
300 | that case repository would be created and returned. | |||
|
301 | If ``src_url`` is given, would try to clone repository from the | |||
|
302 | location at given clone_point. Additionally it'll make update to | |||
|
303 | working copy accordingly to ``update_after_clone`` flag | |||
|
304 | """ | |||
|
305 | try: | |||
|
306 | if src_url: | |||
|
307 | url = str(self._get_url(src_url)) | |||
|
308 | opts = {} | |||
|
309 | if not update_after_clone: | |||
|
310 | opts.update({'noupdate': True}) | |||
|
311 | try: | |||
|
312 | self._check_url(url) | |||
|
313 | clone(self.baseui, url, self.path, **opts) | |||
|
314 | # except urllib2.URLError: | |||
|
315 | # raise Abort("Got HTTP 404 error") | |||
|
316 | except Exception: | |||
|
317 | raise | |||
|
318 | # Don't try to create if we've already cloned repo | |||
|
319 | create = False | |||
|
320 | return localrepository(self.baseui, self.path, create=create) | |||
|
321 | except (Abort, RepoError), err: | |||
|
322 | if create: | |||
|
323 | msg = "Cannot create repository at %s. Original error was %s"\ | |||
|
324 | % (self.path, err) | |||
|
325 | else: | |||
|
326 | msg = "Not valid repository at %s. Original error was %s"\ | |||
|
327 | % (self.path, err) | |||
|
328 | raise RepositoryError(msg) | |||
|
329 | ||||
|
330 | @LazyProperty | |||
|
331 | def in_memory_changeset(self): | |||
|
332 | return MercurialInMemoryChangeset(self) | |||
|
333 | ||||
|
334 | @LazyProperty | |||
|
335 | def description(self): | |||
|
336 | undefined_description = u'unknown' | |||
|
337 | return safe_unicode(self._repo.ui.config('web', 'description', | |||
|
338 | undefined_description, untrusted=True)) | |||
|
339 | ||||
|
340 | @LazyProperty | |||
|
341 | def contact(self): | |||
|
342 | undefined_contact = u'Unknown' | |||
|
343 | return safe_unicode(get_contact(self._repo.ui.config) | |||
|
344 | or undefined_contact) | |||
|
345 | ||||
|
346 | @LazyProperty | |||
|
347 | def last_change(self): | |||
|
348 | """ | |||
|
349 | Returns last change made on this repository as datetime object | |||
|
350 | """ | |||
|
351 | return date_fromtimestamp(self._get_mtime(), makedate()[1]) | |||
|
352 | ||||
|
353 | def _get_mtime(self): | |||
|
354 | try: | |||
|
355 | return time.mktime(self.get_changeset().date.timetuple()) | |||
|
356 | except RepositoryError: | |||
|
357 | #fallback to filesystem | |||
|
358 | cl_path = os.path.join(self.path, '.hg', "00changelog.i") | |||
|
359 | st_path = os.path.join(self.path, '.hg', "store") | |||
|
360 | if os.path.exists(cl_path): | |||
|
361 | return os.stat(cl_path).st_mtime | |||
|
362 | else: | |||
|
363 | return os.stat(st_path).st_mtime | |||
|
364 | ||||
|
365 | def _get_hidden(self): | |||
|
366 | return self._repo.ui.configbool("web", "hidden", untrusted=True) | |||
|
367 | ||||
|
368 | def _get_revision(self, revision): | |||
|
369 | """ | |||
|
370 | Get's an ID revision given as str. This will always return a fill | |||
|
371 | 40 char revision number | |||
|
372 | ||||
|
373 | :param revision: str or int or None | |||
|
374 | """ | |||
|
375 | ||||
|
376 | if self._empty: | |||
|
377 | raise EmptyRepositoryError("There are no changesets yet") | |||
|
378 | ||||
|
379 | if revision in [-1, 'tip', None]: | |||
|
380 | revision = 'tip' | |||
|
381 | ||||
|
382 | try: | |||
|
383 | revision = hex(self._repo.lookup(revision)) | |||
|
384 | except (IndexError, ValueError, RepoLookupError, TypeError): | |||
|
385 | raise ChangesetDoesNotExistError("Revision %r does not " | |||
|
386 | "exist for this repository %s" \ | |||
|
387 | % (revision, self)) | |||
|
388 | return revision | |||
|
389 | ||||
|
390 | def _get_archives(self, archive_name='tip'): | |||
|
391 | allowed = self.baseui.configlist("web", "allow_archive", | |||
|
392 | untrusted=True) | |||
|
393 | for i in [('zip', '.zip'), ('gz', '.tar.gz'), ('bz2', '.tar.bz2')]: | |||
|
394 | if i[0] in allowed or self._repo.ui.configbool("web", | |||
|
395 | "allow" + i[0], | |||
|
396 | untrusted=True): | |||
|
397 | yield {"type": i[0], "extension": i[1], "node": archive_name} | |||
|
398 | ||||
|
399 | def _get_url(self, url): | |||
|
400 | """ | |||
|
401 | Returns normalized url. If schema is not given, would fall | |||
|
402 | to filesystem | |||
|
403 | (``file:///``) schema. | |||
|
404 | """ | |||
|
405 | url = str(url) | |||
|
406 | if url != 'default' and not '://' in url: | |||
|
407 | url = "file:" + urllib.pathname2url(url) | |||
|
408 | return url | |||
|
409 | ||||
|
410 | def get_changeset(self, revision=None): | |||
|
411 | """ | |||
|
412 | Returns ``MercurialChangeset`` object representing repository's | |||
|
413 | changeset at the given ``revision``. | |||
|
414 | """ | |||
|
415 | revision = self._get_revision(revision) | |||
|
416 | changeset = MercurialChangeset(repository=self, revision=revision) | |||
|
417 | return changeset | |||
|
418 | ||||
|
419 | def get_changesets(self, start=None, end=None, start_date=None, | |||
|
420 | end_date=None, branch_name=None, reverse=False): | |||
|
421 | """ | |||
|
422 | Returns iterator of ``MercurialChangeset`` objects from start to end | |||
|
423 | (both are inclusive) | |||
|
424 | ||||
|
425 | :param start: None, str, int or mercurial lookup format | |||
|
426 | :param end: None, str, int or mercurial lookup format | |||
|
427 | :param start_date: | |||
|
428 | :param end_date: | |||
|
429 | :param branch_name: | |||
|
430 | :param reversed: return changesets in reversed order | |||
|
431 | """ | |||
|
432 | ||||
|
433 | start_raw_id = self._get_revision(start) | |||
|
434 | start_pos = self.revisions.index(start_raw_id) if start else None | |||
|
435 | end_raw_id = self._get_revision(end) | |||
|
436 | end_pos = self.revisions.index(end_raw_id) if end else None | |||
|
437 | ||||
|
438 | if None not in [start, end] and start_pos > end_pos: | |||
|
439 | raise RepositoryError("start revision '%s' cannot be " | |||
|
440 | "after end revision '%s'" % (start, end)) | |||
|
441 | ||||
|
442 | if branch_name and branch_name not in self.branches.keys(): | |||
|
443 | raise BranchDoesNotExistError('Such branch %s does not exists for' | |||
|
444 | ' this repository' % branch_name) | |||
|
445 | if end_pos is not None: | |||
|
446 | end_pos += 1 | |||
|
447 | ||||
|
448 | slice_ = reversed(self.revisions[start_pos:end_pos]) if reverse else \ | |||
|
449 | self.revisions[start_pos:end_pos] | |||
|
450 | ||||
|
451 | for id_ in slice_: | |||
|
452 | cs = self.get_changeset(id_) | |||
|
453 | if branch_name and cs.branch != branch_name: | |||
|
454 | continue | |||
|
455 | if start_date and cs.date < start_date: | |||
|
456 | continue | |||
|
457 | if end_date and cs.date > end_date: | |||
|
458 | continue | |||
|
459 | ||||
|
460 | yield cs | |||
|
461 | ||||
|
462 | def pull(self, url): | |||
|
463 | """ | |||
|
464 | Tries to pull changes from external location. | |||
|
465 | """ | |||
|
466 | url = self._get_url(url) | |||
|
467 | try: | |||
|
468 | pull(self.baseui, self._repo, url) | |||
|
469 | except Abort, err: | |||
|
470 | # Propagate error but with vcs's type | |||
|
471 | raise RepositoryError(str(err)) | |||
|
472 | ||||
|
473 | @LazyProperty | |||
|
474 | def workdir(self): | |||
|
475 | """ | |||
|
476 | Returns ``Workdir`` instance for this repository. | |||
|
477 | """ | |||
|
478 | return MercurialWorkdir(self) | |||
|
479 | ||||
|
480 | def get_config_value(self, section, name, config_file=None): | |||
|
481 | """ | |||
|
482 | Returns configuration value for a given [``section``] and ``name``. | |||
|
483 | ||||
|
484 | :param section: Section we want to retrieve value from | |||
|
485 | :param name: Name of configuration we want to retrieve | |||
|
486 | :param config_file: A path to file which should be used to retrieve | |||
|
487 | configuration from (might also be a list of file paths) | |||
|
488 | """ | |||
|
489 | if config_file is None: | |||
|
490 | config_file = [] | |||
|
491 | elif isinstance(config_file, basestring): | |||
|
492 | config_file = [config_file] | |||
|
493 | ||||
|
494 | config = self._repo.ui | |||
|
495 | for path in config_file: | |||
|
496 | config.readconfig(path) | |||
|
497 | return config.config(section, name) | |||
|
498 | ||||
|
499 | def get_user_name(self, config_file=None): | |||
|
500 | """ | |||
|
501 | Returns user's name from global configuration file. | |||
|
502 | ||||
|
503 | :param config_file: A path to file which should be used to retrieve | |||
|
504 | configuration from (might also be a list of file paths) | |||
|
505 | """ | |||
|
506 | username = self.get_config_value('ui', 'username') | |||
|
507 | if username: | |||
|
508 | return author_name(username) | |||
|
509 | return None | |||
|
510 | ||||
|
511 | def get_user_email(self, config_file=None): | |||
|
512 | """ | |||
|
513 | Returns user's email from global configuration file. | |||
|
514 | ||||
|
515 | :param config_file: A path to file which should be used to retrieve | |||
|
516 | configuration from (might also be a list of file paths) | |||
|
517 | """ | |||
|
518 | username = self.get_config_value('ui', 'username') | |||
|
519 | if username: | |||
|
520 | return author_email(username) | |||
|
521 | return None |
@@ -0,0 +1,21 b'' | |||||
|
1 | from rhodecode.lib.vcs.backends.base import BaseWorkdir | |||
|
2 | from rhodecode.lib.vcs.exceptions import BranchDoesNotExistError | |||
|
3 | ||||
|
4 | from ...utils.hgcompat import hg_merge | |||
|
5 | ||||
|
6 | ||||
|
7 | class MercurialWorkdir(BaseWorkdir): | |||
|
8 | ||||
|
9 | def get_branch(self): | |||
|
10 | return self.repository._repo.dirstate.branch() | |||
|
11 | ||||
|
12 | def get_changeset(self): | |||
|
13 | return self.repository.get_changeset() | |||
|
14 | ||||
|
15 | def checkout_branch(self, branch=None): | |||
|
16 | if branch is None: | |||
|
17 | branch = self.repository.DEFAULT_BRANCH_NAME | |||
|
18 | if branch not in self.repository.branches: | |||
|
19 | raise BranchDoesNotExistError | |||
|
20 | ||||
|
21 | hg_merge.update(self.repository._repo, branch, False, False, None) |
1 | NO CONTENT: new file 100644 |
|
NO CONTENT: new file 100644 |
@@ -0,0 +1,33 b'' | |||||
|
1 | import os | |||
|
2 | import tempfile | |||
|
3 | from rhodecode.lib.vcs.utils.paths import get_user_home | |||
|
4 | ||||
|
5 | abspath = lambda * p: os.path.abspath(os.path.join(*p)) | |||
|
6 | ||||
|
7 | VCSRC_PATH = os.environ.get('VCSRC_PATH') | |||
|
8 | ||||
|
9 | if not VCSRC_PATH: | |||
|
10 | HOME_ = get_user_home() | |||
|
11 | if not HOME_: | |||
|
12 | HOME_ = tempfile.gettempdir() | |||
|
13 | ||||
|
14 | VCSRC_PATH = VCSRC_PATH or abspath(HOME_, '.vcsrc') | |||
|
15 | if os.path.isdir(VCSRC_PATH): | |||
|
16 | VCSRC_PATH = os.path.join(VCSRC_PATH, '__init__.py') | |||
|
17 | ||||
|
18 | BACKENDS = { | |||
|
19 | 'hg': 'vcs.backends.hg.MercurialRepository', | |||
|
20 | 'git': 'vcs.backends.git.GitRepository', | |||
|
21 | } | |||
|
22 | ||||
|
23 | ARCHIVE_SPECS = { | |||
|
24 | 'tar': ('application/x-tar', '.tar'), | |||
|
25 | 'tbz2': ('application/x-bzip2', '.tar.bz2'), | |||
|
26 | 'tgz': ('application/x-gzip', '.tar.gz'), | |||
|
27 | 'zip': ('application/zip', '.zip'), | |||
|
28 | } | |||
|
29 | ||||
|
30 | BACKENDS = { | |||
|
31 | 'hg': 'rhodecode.lib.vcs.backends.hg.MercurialRepository', | |||
|
32 | 'git': 'rhodecode.lib.vcs.backends.git.GitRepository', | |||
|
33 | } |
@@ -0,0 +1,93 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | """ | |||
|
3 | vcs.exceptions | |||
|
4 | ~~~~~~~~~~~~~~ | |||
|
5 | ||||
|
6 | Custom exceptions module | |||
|
7 | ||||
|
8 | :created_on: Apr 8, 2010 | |||
|
9 | :copyright: (c) 2010-2011 by Marcin Kuzminski, Lukasz Balcerzak. | |||
|
10 | """ | |||
|
11 | ||||
|
12 | ||||
|
13 | class VCSError(Exception): | |||
|
14 | pass | |||
|
15 | ||||
|
16 | ||||
|
17 | class RepositoryError(VCSError): | |||
|
18 | pass | |||
|
19 | ||||
|
20 | ||||
|
21 | class EmptyRepositoryError(RepositoryError): | |||
|
22 | pass | |||
|
23 | ||||
|
24 | ||||
|
25 | class TagAlreadyExistError(RepositoryError): | |||
|
26 | pass | |||
|
27 | ||||
|
28 | ||||
|
29 | class TagDoesNotExistError(RepositoryError): | |||
|
30 | pass | |||
|
31 | ||||
|
32 | ||||
|
33 | class BranchAlreadyExistError(RepositoryError): | |||
|
34 | pass | |||
|
35 | ||||
|
36 | ||||
|
37 | class BranchDoesNotExistError(RepositoryError): | |||
|
38 | pass | |||
|
39 | ||||
|
40 | ||||
|
41 | class ChangesetError(RepositoryError): | |||
|
42 | pass | |||
|
43 | ||||
|
44 | ||||
|
45 | class ChangesetDoesNotExistError(ChangesetError): | |||
|
46 | pass | |||
|
47 | ||||
|
48 | ||||
|
49 | class CommitError(RepositoryError): | |||
|
50 | pass | |||
|
51 | ||||
|
52 | ||||
|
53 | class NothingChangedError(CommitError): | |||
|
54 | pass | |||
|
55 | ||||
|
56 | ||||
|
57 | class NodeError(VCSError): | |||
|
58 | pass | |||
|
59 | ||||
|
60 | ||||
|
61 | class RemovedFileNodeError(NodeError): | |||
|
62 | pass | |||
|
63 | ||||
|
64 | ||||
|
65 | class NodeAlreadyExistsError(CommitError): | |||
|
66 | pass | |||
|
67 | ||||
|
68 | ||||
|
69 | class NodeAlreadyChangedError(CommitError): | |||
|
70 | pass | |||
|
71 | ||||
|
72 | ||||
|
73 | class NodeDoesNotExistError(CommitError): | |||
|
74 | pass | |||
|
75 | ||||
|
76 | ||||
|
77 | class NodeNotChangedError(CommitError): | |||
|
78 | pass | |||
|
79 | ||||
|
80 | ||||
|
81 | class NodeAlreadyAddedError(CommitError): | |||
|
82 | pass | |||
|
83 | ||||
|
84 | ||||
|
85 | class NodeAlreadyRemovedError(CommitError): | |||
|
86 | pass | |||
|
87 | ||||
|
88 | ||||
|
89 | class ImproperArchiveTypeError(VCSError): | |||
|
90 | pass | |||
|
91 | ||||
|
92 | class CommandError(VCSError): | |||
|
93 | pass |
This diff has been collapsed as it changes many lines, (551 lines changed) Show them Hide them | |||||
@@ -0,0 +1,551 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | """ | |||
|
3 | vcs.nodes | |||
|
4 | ~~~~~~~~~ | |||
|
5 | ||||
|
6 | Module holding everything related to vcs nodes. | |||
|
7 | ||||
|
8 | :created_on: Apr 8, 2010 | |||
|
9 | :copyright: (c) 2010-2011 by Marcin Kuzminski, Lukasz Balcerzak. | |||
|
10 | """ | |||
|
11 | import stat | |||
|
12 | import posixpath | |||
|
13 | import mimetypes | |||
|
14 | ||||
|
15 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |||
|
16 | from rhodecode.lib.vcs.utils import safe_unicode | |||
|
17 | from rhodecode.lib.vcs.exceptions import NodeError | |||
|
18 | from rhodecode.lib.vcs.exceptions import RemovedFileNodeError | |||
|
19 | ||||
|
20 | from pygments import lexers | |||
|
21 | ||||
|
22 | ||||
|
23 | class NodeKind: | |||
|
24 | DIR = 1 | |||
|
25 | FILE = 2 | |||
|
26 | ||||
|
27 | ||||
|
28 | class NodeState: | |||
|
29 | ADDED = u'added' | |||
|
30 | CHANGED = u'changed' | |||
|
31 | NOT_CHANGED = u'not changed' | |||
|
32 | REMOVED = u'removed' | |||
|
33 | ||||
|
34 | ||||
|
35 | class NodeGeneratorBase(object): | |||
|
36 | """ | |||
|
37 | Base class for removed added and changed filenodes, it's a lazy generator | |||
|
38 | class that will create filenodes only on iteration or call | |||
|
39 | ||||
|
40 | The len method doesn't need to create filenodes at all | |||
|
41 | """ | |||
|
42 | ||||
|
43 | def __init__(self, current_paths, cs): | |||
|
44 | self.cs = cs | |||
|
45 | self.current_paths = current_paths | |||
|
46 | ||||
|
47 | def __call__(self): | |||
|
48 | return [n for n in self] | |||
|
49 | ||||
|
50 | def __getslice__(self, i, j): | |||
|
51 | for p in self.current_paths[i:j]: | |||
|
52 | yield self.cs.get_node(p) | |||
|
53 | ||||
|
54 | def __len__(self): | |||
|
55 | return len(self.current_paths) | |||
|
56 | ||||
|
57 | def __iter__(self): | |||
|
58 | for p in self.current_paths: | |||
|
59 | yield self.cs.get_node(p) | |||
|
60 | ||||
|
61 | ||||
|
62 | class AddedFileNodesGenerator(NodeGeneratorBase): | |||
|
63 | """ | |||
|
64 | Class holding Added files for current changeset | |||
|
65 | """ | |||
|
66 | pass | |||
|
67 | ||||
|
68 | ||||
|
69 | class ChangedFileNodesGenerator(NodeGeneratorBase): | |||
|
70 | """ | |||
|
71 | Class holding Changed files for current changeset | |||
|
72 | """ | |||
|
73 | pass | |||
|
74 | ||||
|
75 | ||||
|
76 | class RemovedFileNodesGenerator(NodeGeneratorBase): | |||
|
77 | """ | |||
|
78 | Class holding removed files for current changeset | |||
|
79 | """ | |||
|
80 | def __iter__(self): | |||
|
81 | for p in self.current_paths: | |||
|
82 | yield RemovedFileNode(path=p) | |||
|
83 | ||||
|
84 | def __getslice__(self, i, j): | |||
|
85 | for p in self.current_paths[i:j]: | |||
|
86 | yield RemovedFileNode(path=p) | |||
|
87 | ||||
|
88 | ||||
|
89 | class Node(object): | |||
|
90 | """ | |||
|
91 | Simplest class representing file or directory on repository. SCM backends | |||
|
92 | should use ``FileNode`` and ``DirNode`` subclasses rather than ``Node`` | |||
|
93 | directly. | |||
|
94 | ||||
|
95 | Node's ``path`` cannot start with slash as we operate on *relative* paths | |||
|
96 | only. Moreover, every single node is identified by the ``path`` attribute, | |||
|
97 | so it cannot end with slash, too. Otherwise, path could lead to mistakes. | |||
|
98 | """ | |||
|
99 | ||||
|
100 | def __init__(self, path, kind): | |||
|
101 | if path.startswith('/'): | |||
|
102 | raise NodeError("Cannot initialize Node objects with slash at " | |||
|
103 | "the beginning as only relative paths are supported") | |||
|
104 | self.path = path.rstrip('/') | |||
|
105 | if path == '' and kind != NodeKind.DIR: | |||
|
106 | raise NodeError("Only DirNode and its subclasses may be " | |||
|
107 | "initialized with empty path") | |||
|
108 | self.kind = kind | |||
|
109 | #self.dirs, self.files = [], [] | |||
|
110 | if self.is_root() and not self.is_dir(): | |||
|
111 | raise NodeError("Root node cannot be FILE kind") | |||
|
112 | ||||
|
113 | @LazyProperty | |||
|
114 | def parent(self): | |||
|
115 | parent_path = self.get_parent_path() | |||
|
116 | if parent_path: | |||
|
117 | if self.changeset: | |||
|
118 | return self.changeset.get_node(parent_path) | |||
|
119 | return DirNode(parent_path) | |||
|
120 | return None | |||
|
121 | ||||
|
122 | @LazyProperty | |||
|
123 | def name(self): | |||
|
124 | """ | |||
|
125 | Returns name of the node so if its path | |||
|
126 | then only last part is returned. | |||
|
127 | """ | |||
|
128 | return safe_unicode(self.path.rstrip('/').split('/')[-1]) | |||
|
129 | ||||
|
130 | def _get_kind(self): | |||
|
131 | return self._kind | |||
|
132 | ||||
|
133 | def _set_kind(self, kind): | |||
|
134 | if hasattr(self, '_kind'): | |||
|
135 | raise NodeError("Cannot change node's kind") | |||
|
136 | else: | |||
|
137 | self._kind = kind | |||
|
138 | # Post setter check (path's trailing slash) | |||
|
139 | if self.path.endswith('/'): | |||
|
140 | raise NodeError("Node's path cannot end with slash") | |||
|
141 | ||||
|
142 | kind = property(_get_kind, _set_kind) | |||
|
143 | ||||
|
144 | def __cmp__(self, other): | |||
|
145 | """ | |||
|
146 | Comparator using name of the node, needed for quick list sorting. | |||
|
147 | """ | |||
|
148 | kind_cmp = cmp(self.kind, other.kind) | |||
|
149 | if kind_cmp: | |||
|
150 | return kind_cmp | |||
|
151 | return cmp(self.name, other.name) | |||
|
152 | ||||
|
153 | def __eq__(self, other): | |||
|
154 | for attr in ['name', 'path', 'kind']: | |||
|
155 | if getattr(self, attr) != getattr(other, attr): | |||
|
156 | return False | |||
|
157 | if self.is_file(): | |||
|
158 | if self.content != other.content: | |||
|
159 | return False | |||
|
160 | else: | |||
|
161 | # For DirNode's check without entering each dir | |||
|
162 | self_nodes_paths = list(sorted(n.path for n in self.nodes)) | |||
|
163 | other_nodes_paths = list(sorted(n.path for n in self.nodes)) | |||
|
164 | if self_nodes_paths != other_nodes_paths: | |||
|
165 | return False | |||
|
166 | return True | |||
|
167 | ||||
|
168 | def __nq__(self, other): | |||
|
169 | return not self.__eq__(other) | |||
|
170 | ||||
|
171 | def __repr__(self): | |||
|
172 | return '<%s %r>' % (self.__class__.__name__, self.path) | |||
|
173 | ||||
|
174 | def __str__(self): | |||
|
175 | return self.__repr__() | |||
|
176 | ||||
|
177 | def __unicode__(self): | |||
|
178 | return self.name | |||
|
179 | ||||
|
180 | def get_parent_path(self): | |||
|
181 | """ | |||
|
182 | Returns node's parent path or empty string if node is root. | |||
|
183 | """ | |||
|
184 | if self.is_root(): | |||
|
185 | return '' | |||
|
186 | return posixpath.dirname(self.path.rstrip('/')) + '/' | |||
|
187 | ||||
|
188 | def is_file(self): | |||
|
189 | """ | |||
|
190 | Returns ``True`` if node's kind is ``NodeKind.FILE``, ``False`` | |||
|
191 | otherwise. | |||
|
192 | """ | |||
|
193 | return self.kind == NodeKind.FILE | |||
|
194 | ||||
|
195 | def is_dir(self): | |||
|
196 | """ | |||
|
197 | Returns ``True`` if node's kind is ``NodeKind.DIR``, ``False`` | |||
|
198 | otherwise. | |||
|
199 | """ | |||
|
200 | return self.kind == NodeKind.DIR | |||
|
201 | ||||
|
202 | def is_root(self): | |||
|
203 | """ | |||
|
204 | Returns ``True`` if node is a root node and ``False`` otherwise. | |||
|
205 | """ | |||
|
206 | return self.kind == NodeKind.DIR and self.path == '' | |||
|
207 | ||||
|
208 | @LazyProperty | |||
|
209 | def added(self): | |||
|
210 | return self.state is NodeState.ADDED | |||
|
211 | ||||
|
212 | @LazyProperty | |||
|
213 | def changed(self): | |||
|
214 | return self.state is NodeState.CHANGED | |||
|
215 | ||||
|
216 | @LazyProperty | |||
|
217 | def not_changed(self): | |||
|
218 | return self.state is NodeState.NOT_CHANGED | |||
|
219 | ||||
|
220 | @LazyProperty | |||
|
221 | def removed(self): | |||
|
222 | return self.state is NodeState.REMOVED | |||
|
223 | ||||
|
224 | ||||
|
225 | class FileNode(Node): | |||
|
226 | """ | |||
|
227 | Class representing file nodes. | |||
|
228 | ||||
|
229 | :attribute: path: path to the node, relative to repostiory's root | |||
|
230 | :attribute: content: if given arbitrary sets content of the file | |||
|
231 | :attribute: changeset: if given, first time content is accessed, callback | |||
|
232 | :attribute: mode: octal stat mode for a node. Default is 0100644. | |||
|
233 | """ | |||
|
234 | ||||
|
235 | def __init__(self, path, content=None, changeset=None, mode=None): | |||
|
236 | """ | |||
|
237 | Only one of ``content`` and ``changeset`` may be given. Passing both | |||
|
238 | would raise ``NodeError`` exception. | |||
|
239 | ||||
|
240 | :param path: relative path to the node | |||
|
241 | :param content: content may be passed to constructor | |||
|
242 | :param changeset: if given, will use it to lazily fetch content | |||
|
243 | :param mode: octal representation of ST_MODE (i.e. 0100644) | |||
|
244 | """ | |||
|
245 | ||||
|
246 | if content and changeset: | |||
|
247 | raise NodeError("Cannot use both content and changeset") | |||
|
248 | super(FileNode, self).__init__(path, kind=NodeKind.FILE) | |||
|
249 | self.changeset = changeset | |||
|
250 | self._content = content | |||
|
251 | self._mode = mode or 0100644 | |||
|
252 | ||||
|
253 | @LazyProperty | |||
|
254 | def mode(self): | |||
|
255 | """ | |||
|
256 | Returns lazily mode of the FileNode. If ``changeset`` is not set, would | |||
|
257 | use value given at initialization or 0100644 (default). | |||
|
258 | """ | |||
|
259 | if self.changeset: | |||
|
260 | mode = self.changeset.get_file_mode(self.path) | |||
|
261 | else: | |||
|
262 | mode = self._mode | |||
|
263 | return mode | |||
|
264 | ||||
|
265 | @property | |||
|
266 | def content(self): | |||
|
267 | """ | |||
|
268 | Returns lazily content of the FileNode. If possible, would try to | |||
|
269 | decode content from UTF-8. | |||
|
270 | """ | |||
|
271 | if self.changeset: | |||
|
272 | content = self.changeset.get_file_content(self.path) | |||
|
273 | else: | |||
|
274 | content = self._content | |||
|
275 | ||||
|
276 | if bool(content and '\0' in content): | |||
|
277 | return content | |||
|
278 | return safe_unicode(content) | |||
|
279 | ||||
|
280 | @LazyProperty | |||
|
281 | def size(self): | |||
|
282 | if self.changeset: | |||
|
283 | return self.changeset.get_file_size(self.path) | |||
|
284 | raise NodeError("Cannot retrieve size of the file without related " | |||
|
285 | "changeset attribute") | |||
|
286 | ||||
|
287 | @LazyProperty | |||
|
288 | def message(self): | |||
|
289 | if self.changeset: | |||
|
290 | return self.last_changeset.message | |||
|
291 | raise NodeError("Cannot retrieve message of the file without related " | |||
|
292 | "changeset attribute") | |||
|
293 | ||||
|
294 | @LazyProperty | |||
|
295 | def last_changeset(self): | |||
|
296 | if self.changeset: | |||
|
297 | return self.changeset.get_file_changeset(self.path) | |||
|
298 | raise NodeError("Cannot retrieve last changeset of the file without " | |||
|
299 | "related changeset attribute") | |||
|
300 | ||||
|
301 | def get_mimetype(self): | |||
|
302 | """ | |||
|
303 | Mimetype is calculated based on the file's content. If ``_mimetype`` | |||
|
304 | attribute is available, it will be returned (backends which store | |||
|
305 | mimetypes or can easily recognize them, should set this private | |||
|
306 | attribute to indicate that type should *NOT* be calculated). | |||
|
307 | """ | |||
|
308 | if hasattr(self, '_mimetype'): | |||
|
309 | if (isinstance(self._mimetype,(tuple,list,)) and | |||
|
310 | len(self._mimetype) == 2): | |||
|
311 | return self._mimetype | |||
|
312 | else: | |||
|
313 | raise NodeError('given _mimetype attribute must be an 2 ' | |||
|
314 | 'element list or tuple') | |||
|
315 | ||||
|
316 | mtype,encoding = mimetypes.guess_type(self.name) | |||
|
317 | ||||
|
318 | if mtype is None: | |||
|
319 | if self.is_binary: | |||
|
320 | mtype = 'application/octet-stream' | |||
|
321 | encoding = None | |||
|
322 | else: | |||
|
323 | mtype = 'text/plain' | |||
|
324 | encoding = None | |||
|
325 | return mtype,encoding | |||
|
326 | ||||
|
327 | @LazyProperty | |||
|
328 | def mimetype(self): | |||
|
329 | """ | |||
|
330 | Wrapper around full mimetype info. It returns only type of fetched | |||
|
331 | mimetype without the encoding part. use get_mimetype function to fetch | |||
|
332 | full set of (type,encoding) | |||
|
333 | """ | |||
|
334 | return self.get_mimetype()[0] | |||
|
335 | ||||
|
336 | @LazyProperty | |||
|
337 | def mimetype_main(self): | |||
|
338 | return self.mimetype.split('/')[0] | |||
|
339 | ||||
|
340 | @LazyProperty | |||
|
341 | def lexer(self): | |||
|
342 | """ | |||
|
343 | Returns pygment's lexer class. Would try to guess lexer taking file's | |||
|
344 | content, name and mimetype. | |||
|
345 | """ | |||
|
346 | try: | |||
|
347 | lexer = lexers.guess_lexer_for_filename(self.name, self.content) | |||
|
348 | except lexers.ClassNotFound: | |||
|
349 | lexer = lexers.TextLexer() | |||
|
350 | # returns first alias | |||
|
351 | return lexer | |||
|
352 | ||||
|
353 | @LazyProperty | |||
|
354 | def lexer_alias(self): | |||
|
355 | """ | |||
|
356 | Returns first alias of the lexer guessed for this file. | |||
|
357 | """ | |||
|
358 | return self.lexer.aliases[0] | |||
|
359 | ||||
|
360 | @LazyProperty | |||
|
361 | def history(self): | |||
|
362 | """ | |||
|
363 | Returns a list of changeset for this file in which the file was changed | |||
|
364 | """ | |||
|
365 | if self.changeset is None: | |||
|
366 | raise NodeError('Unable to get changeset for this FileNode') | |||
|
367 | return self.changeset.get_file_history(self.path) | |||
|
368 | ||||
|
369 | @LazyProperty | |||
|
370 | def annotate(self): | |||
|
371 | """ | |||
|
372 | Returns a list of three element tuples with lineno,changeset and line | |||
|
373 | """ | |||
|
374 | if self.changeset is None: | |||
|
375 | raise NodeError('Unable to get changeset for this FileNode') | |||
|
376 | return self.changeset.get_file_annotate(self.path) | |||
|
377 | ||||
|
378 | @LazyProperty | |||
|
379 | def state(self): | |||
|
380 | if not self.changeset: | |||
|
381 | raise NodeError("Cannot check state of the node if it's not " | |||
|
382 | "linked with changeset") | |||
|
383 | elif self.path in (node.path for node in self.changeset.added): | |||
|
384 | return NodeState.ADDED | |||
|
385 | elif self.path in (node.path for node in self.changeset.changed): | |||
|
386 | return NodeState.CHANGED | |||
|
387 | else: | |||
|
388 | return NodeState.NOT_CHANGED | |||
|
389 | ||||
|
390 | @property | |||
|
391 | def is_binary(self): | |||
|
392 | """ | |||
|
393 | Returns True if file has binary content. | |||
|
394 | """ | |||
|
395 | bin = '\0' in self.content | |||
|
396 | return bin | |||
|
397 | ||||
|
398 | @LazyProperty | |||
|
399 | def extension(self): | |||
|
400 | """Returns filenode extension""" | |||
|
401 | return self.name.split('.')[-1] | |||
|
402 | ||||
|
403 | def is_executable(self): | |||
|
404 | """ | |||
|
405 | Returns ``True`` if file has executable flag turned on. | |||
|
406 | """ | |||
|
407 | return bool(self.mode & stat.S_IXUSR) | |||
|
408 | ||||
|
409 | ||||
|
410 | class RemovedFileNode(FileNode): | |||
|
411 | """ | |||
|
412 | Dummy FileNode class - trying to access any public attribute except path, | |||
|
413 | name, kind or state (or methods/attributes checking those two) would raise | |||
|
414 | RemovedFileNodeError. | |||
|
415 | """ | |||
|
416 | ALLOWED_ATTRIBUTES = ['name', 'path', 'state', 'is_root', 'is_file', | |||
|
417 | 'is_dir', 'kind', 'added', 'changed', 'not_changed', 'removed'] | |||
|
418 | ||||
|
419 | def __init__(self, path): | |||
|
420 | """ | |||
|
421 | :param path: relative path to the node | |||
|
422 | """ | |||
|
423 | super(RemovedFileNode, self).__init__(path=path) | |||
|
424 | ||||
|
425 | def __getattribute__(self, attr): | |||
|
426 | if attr.startswith('_') or attr in RemovedFileNode.ALLOWED_ATTRIBUTES: | |||
|
427 | return super(RemovedFileNode, self).__getattribute__(attr) | |||
|
428 | raise RemovedFileNodeError("Cannot access attribute %s on " | |||
|
429 | "RemovedFileNode" % attr) | |||
|
430 | ||||
|
431 | @LazyProperty | |||
|
432 | def state(self): | |||
|
433 | return NodeState.REMOVED | |||
|
434 | ||||
|
435 | ||||
|
436 | class DirNode(Node): | |||
|
437 | """ | |||
|
438 | DirNode stores list of files and directories within this node. | |||
|
439 | Nodes may be used standalone but within repository context they | |||
|
440 | lazily fetch data within same repositorty's changeset. | |||
|
441 | """ | |||
|
442 | ||||
|
443 | def __init__(self, path, nodes=(), changeset=None): | |||
|
444 | """ | |||
|
445 | Only one of ``nodes`` and ``changeset`` may be given. Passing both | |||
|
446 | would raise ``NodeError`` exception. | |||
|
447 | ||||
|
448 | :param path: relative path to the node | |||
|
449 | :param nodes: content may be passed to constructor | |||
|
450 | :param changeset: if given, will use it to lazily fetch content | |||
|
451 | :param size: always 0 for ``DirNode`` | |||
|
452 | """ | |||
|
453 | if nodes and changeset: | |||
|
454 | raise NodeError("Cannot use both nodes and changeset") | |||
|
455 | super(DirNode, self).__init__(path, NodeKind.DIR) | |||
|
456 | self.changeset = changeset | |||
|
457 | self._nodes = nodes | |||
|
458 | ||||
|
459 | @LazyProperty | |||
|
460 | def content(self): | |||
|
461 | raise NodeError("%s represents a dir and has no ``content`` attribute" | |||
|
462 | % self) | |||
|
463 | ||||
|
464 | @LazyProperty | |||
|
465 | def nodes(self): | |||
|
466 | if self.changeset: | |||
|
467 | nodes = self.changeset.get_nodes(self.path) | |||
|
468 | else: | |||
|
469 | nodes = self._nodes | |||
|
470 | self._nodes_dict = dict((node.path, node) for node in nodes) | |||
|
471 | return sorted(nodes) | |||
|
472 | ||||
|
473 | @LazyProperty | |||
|
474 | def files(self): | |||
|
475 | return sorted((node for node in self.nodes if node.is_file())) | |||
|
476 | ||||
|
477 | @LazyProperty | |||
|
478 | def dirs(self): | |||
|
479 | return sorted((node for node in self.nodes if node.is_dir())) | |||
|
480 | ||||
|
481 | def __iter__(self): | |||
|
482 | for node in self.nodes: | |||
|
483 | yield node | |||
|
484 | ||||
|
485 | def get_node(self, path): | |||
|
486 | """ | |||
|
487 | Returns node from within this particular ``DirNode``, so it is now | |||
|
488 | allowed to fetch, i.e. node located at 'docs/api/index.rst' from node | |||
|
489 | 'docs'. In order to access deeper nodes one must fetch nodes between | |||
|
490 | them first - this would work:: | |||
|
491 | ||||
|
492 | docs = root.get_node('docs') | |||
|
493 | docs.get_node('api').get_node('index.rst') | |||
|
494 | ||||
|
495 | :param: path - relative to the current node | |||
|
496 | ||||
|
497 | .. note:: | |||
|
498 | To access lazily (as in example above) node have to be initialized | |||
|
499 | with related changeset object - without it node is out of | |||
|
500 | context and may know nothing about anything else than nearest | |||
|
501 | (located at same level) nodes. | |||
|
502 | """ | |||
|
503 | try: | |||
|
504 | path = path.rstrip('/') | |||
|
505 | if path == '': | |||
|
506 | raise NodeError("Cannot retrieve node without path") | |||
|
507 | self.nodes # access nodes first in order to set _nodes_dict | |||
|
508 | paths = path.split('/') | |||
|
509 | if len(paths) == 1: | |||
|
510 | if not self.is_root(): | |||
|
511 | path = '/'.join((self.path, paths[0])) | |||
|
512 | else: | |||
|
513 | path = paths[0] | |||
|
514 | return self._nodes_dict[path] | |||
|
515 | elif len(paths) > 1: | |||
|
516 | if self.changeset is None: | |||
|
517 | raise NodeError("Cannot access deeper " | |||
|
518 | "nodes without changeset") | |||
|
519 | else: | |||
|
520 | path1, path2 = paths[0], '/'.join(paths[1:]) | |||
|
521 | return self.get_node(path1).get_node(path2) | |||
|
522 | else: | |||
|
523 | raise KeyError | |||
|
524 | except KeyError: | |||
|
525 | raise NodeError("Node does not exist at %s" % path) | |||
|
526 | ||||
|
527 | @LazyProperty | |||
|
528 | def state(self): | |||
|
529 | raise NodeError("Cannot access state of DirNode") | |||
|
530 | ||||
|
531 | @LazyProperty | |||
|
532 | def size(self): | |||
|
533 | size = 0 | |||
|
534 | for root, dirs, files in self.changeset.walk(self.path): | |||
|
535 | for f in files: | |||
|
536 | size += f.size | |||
|
537 | ||||
|
538 | return size | |||
|
539 | ||||
|
540 | ||||
|
541 | class RootNode(DirNode): | |||
|
542 | """ | |||
|
543 | DirNode being the root node of the repository. | |||
|
544 | """ | |||
|
545 | ||||
|
546 | def __init__(self, nodes=(), changeset=None): | |||
|
547 | super(RootNode, self).__init__(path='', nodes=nodes, | |||
|
548 | changeset=changeset) | |||
|
549 | ||||
|
550 | def __repr__(self): | |||
|
551 | return '<%s>' % self.__class__.__name__ |
@@ -0,0 +1,133 b'' | |||||
|
1 | """ | |||
|
2 | This module provides some useful tools for ``vcs`` like annotate/diff html | |||
|
3 | output. It also includes some internal helpers. | |||
|
4 | """ | |||
|
5 | import sys | |||
|
6 | import time | |||
|
7 | import datetime | |||
|
8 | ||||
|
9 | ||||
|
10 | def makedate(): | |||
|
11 | lt = time.localtime() | |||
|
12 | if lt[8] == 1 and time.daylight: | |||
|
13 | tz = time.altzone | |||
|
14 | else: | |||
|
15 | tz = time.timezone | |||
|
16 | return time.mktime(lt), tz | |||
|
17 | ||||
|
18 | ||||
|
19 | def date_fromtimestamp(unixts, tzoffset=0): | |||
|
20 | """ | |||
|
21 | Makes a local datetime object out of unix timestamp | |||
|
22 | ||||
|
23 | :param unixts: | |||
|
24 | :param tzoffset: | |||
|
25 | """ | |||
|
26 | ||||
|
27 | return datetime.datetime.fromtimestamp(float(unixts)) | |||
|
28 | ||||
|
29 | ||||
|
30 | def safe_unicode(str_, from_encoding='utf8'): | |||
|
31 | """ | |||
|
32 | safe unicode function. Does few trick to turn str_ into unicode | |||
|
33 | ||||
|
34 | In case of UnicodeDecode error we try to return it with encoding detected | |||
|
35 | by chardet library if it fails fallback to unicode with errors replaced | |||
|
36 | ||||
|
37 | :param str_: string to decode | |||
|
38 | :rtype: unicode | |||
|
39 | :returns: unicode object | |||
|
40 | """ | |||
|
41 | if isinstance(str_, unicode): | |||
|
42 | return str_ | |||
|
43 | ||||
|
44 | try: | |||
|
45 | return unicode(str_) | |||
|
46 | except UnicodeDecodeError: | |||
|
47 | pass | |||
|
48 | ||||
|
49 | try: | |||
|
50 | return unicode(str_, from_encoding) | |||
|
51 | except UnicodeDecodeError: | |||
|
52 | pass | |||
|
53 | ||||
|
54 | try: | |||
|
55 | import chardet | |||
|
56 | encoding = chardet.detect(str_)['encoding'] | |||
|
57 | if encoding is None: | |||
|
58 | raise Exception() | |||
|
59 | return str_.decode(encoding) | |||
|
60 | except (ImportError, UnicodeDecodeError, Exception): | |||
|
61 | return unicode(str_, from_encoding, 'replace') | |||
|
62 | ||||
|
63 | ||||
|
64 | def safe_str(unicode_, to_encoding='utf8'): | |||
|
65 | """ | |||
|
66 | safe str function. Does few trick to turn unicode_ into string | |||
|
67 | ||||
|
68 | In case of UnicodeEncodeError we try to return it with encoding detected | |||
|
69 | by chardet library if it fails fallback to string with errors replaced | |||
|
70 | ||||
|
71 | :param unicode_: unicode to encode | |||
|
72 | :rtype: str | |||
|
73 | :returns: str object | |||
|
74 | """ | |||
|
75 | ||||
|
76 | if isinstance(unicode_, str): | |||
|
77 | return unicode_ | |||
|
78 | ||||
|
79 | try: | |||
|
80 | return unicode_.encode(to_encoding) | |||
|
81 | except UnicodeEncodeError: | |||
|
82 | pass | |||
|
83 | ||||
|
84 | try: | |||
|
85 | import chardet | |||
|
86 | encoding = chardet.detect(unicode_)['encoding'] | |||
|
87 | print encoding | |||
|
88 | if encoding is None: | |||
|
89 | raise UnicodeEncodeError() | |||
|
90 | ||||
|
91 | return unicode_.encode(encoding) | |||
|
92 | except (ImportError, UnicodeEncodeError): | |||
|
93 | return unicode_.encode(to_encoding, 'replace') | |||
|
94 | ||||
|
95 | return safe_str | |||
|
96 | ||||
|
97 | ||||
|
98 | def author_email(author): | |||
|
99 | """ | |||
|
100 | returns email address of given author. | |||
|
101 | If any of <,> sign are found, it fallbacks to regex findall() | |||
|
102 | and returns first found result or empty string | |||
|
103 | ||||
|
104 | Regex taken from http://www.regular-expressions.info/email.html | |||
|
105 | """ | |||
|
106 | import re | |||
|
107 | r = author.find('>') | |||
|
108 | l = author.find('<') | |||
|
109 | ||||
|
110 | if l == -1 or r == -1: | |||
|
111 | # fallback to regex match of email out of a string | |||
|
112 | email_re = re.compile(r"""[a-z0-9!#$%&'*+/=?^_`{|}~-]+(?:\.[a-z0-9!""" | |||
|
113 | r"""#$%&'*+/=?^_`{|}~-]+)*@(?:[a-z0-9](?:[a-z""" | |||
|
114 | r"""0-9-]*[a-z0-9])?\.)+[a-z0-9](?:[a-z0-9-]""" | |||
|
115 | r"""*[a-z0-9])?""", re.IGNORECASE) | |||
|
116 | m = re.findall(email_re, author) | |||
|
117 | return m[0] if m else '' | |||
|
118 | ||||
|
119 | return author[l + 1:r].strip() | |||
|
120 | ||||
|
121 | ||||
|
122 | def author_name(author): | |||
|
123 | """ | |||
|
124 | get name of author, or else username. | |||
|
125 | It'll try to find an email in the author string and just cut it off | |||
|
126 | to get the username | |||
|
127 | """ | |||
|
128 | ||||
|
129 | if not '@' in author: | |||
|
130 | return author | |||
|
131 | else: | |||
|
132 | return author.replace(author_email(author), '').replace('<', '')\ | |||
|
133 | .replace('>', '').strip() |
@@ -0,0 +1,177 b'' | |||||
|
1 | from rhodecode.lib.vcs.exceptions import VCSError | |||
|
2 | from rhodecode.lib.vcs.nodes import FileNode | |||
|
3 | from pygments.formatters import HtmlFormatter | |||
|
4 | from pygments import highlight | |||
|
5 | ||||
|
6 | import StringIO | |||
|
7 | ||||
|
8 | ||||
|
9 | def annotate_highlight(filenode, annotate_from_changeset_func=None, | |||
|
10 | order=None, headers=None, **options): | |||
|
11 | """ | |||
|
12 | Returns html portion containing annotated table with 3 columns: line | |||
|
13 | numbers, changeset information and pygmentized line of code. | |||
|
14 | ||||
|
15 | :param filenode: FileNode object | |||
|
16 | :param annotate_from_changeset_func: function taking changeset and | |||
|
17 | returning single annotate cell; needs break line at the end | |||
|
18 | :param order: ordered sequence of ``ls`` (line numbers column), | |||
|
19 | ``annotate`` (annotate column), ``code`` (code column); Default is | |||
|
20 | ``['ls', 'annotate', 'code']`` | |||
|
21 | :param headers: dictionary with headers (keys are whats in ``order`` | |||
|
22 | parameter) | |||
|
23 | """ | |||
|
24 | options['linenos'] = True | |||
|
25 | formatter = AnnotateHtmlFormatter(filenode=filenode, order=order, | |||
|
26 | headers=headers, | |||
|
27 | annotate_from_changeset_func=annotate_from_changeset_func, **options) | |||
|
28 | lexer = filenode.lexer | |||
|
29 | highlighted = highlight(filenode.content, lexer, formatter) | |||
|
30 | return highlighted | |||
|
31 | ||||
|
32 | ||||
|
33 | class AnnotateHtmlFormatter(HtmlFormatter): | |||
|
34 | ||||
|
35 | def __init__(self, filenode, annotate_from_changeset_func=None, | |||
|
36 | order=None, **options): | |||
|
37 | """ | |||
|
38 | If ``annotate_from_changeset_func`` is passed it should be a function | |||
|
39 | which returns string from the given changeset. For example, we may pass | |||
|
40 | following function as ``annotate_from_changeset_func``:: | |||
|
41 | ||||
|
42 | def changeset_to_anchor(changeset): | |||
|
43 | return '<a href="/changesets/%s/">%s</a>\n' %\ | |||
|
44 | (changeset.id, changeset.id) | |||
|
45 | ||||
|
46 | :param annotate_from_changeset_func: see above | |||
|
47 | :param order: (default: ``['ls', 'annotate', 'code']``); order of | |||
|
48 | columns; | |||
|
49 | :param options: standard pygment's HtmlFormatter options, there is | |||
|
50 | extra option tough, ``headers``. For instance we can pass:: | |||
|
51 | ||||
|
52 | formatter = AnnotateHtmlFormatter(filenode, headers={ | |||
|
53 | 'ls': '#', | |||
|
54 | 'annotate': 'Annotate', | |||
|
55 | 'code': 'Code', | |||
|
56 | }) | |||
|
57 | ||||
|
58 | """ | |||
|
59 | super(AnnotateHtmlFormatter, self).__init__(**options) | |||
|
60 | self.annotate_from_changeset_func = annotate_from_changeset_func | |||
|
61 | self.order = order or ('ls', 'annotate', 'code') | |||
|
62 | headers = options.pop('headers', None) | |||
|
63 | if headers and not ('ls' in headers and 'annotate' in headers and | |||
|
64 | 'code' in headers): | |||
|
65 | raise ValueError("If headers option dict is specified it must " | |||
|
66 | "all 'ls', 'annotate' and 'code' keys") | |||
|
67 | self.headers = headers | |||
|
68 | if isinstance(filenode, FileNode): | |||
|
69 | self.filenode = filenode | |||
|
70 | else: | |||
|
71 | raise VCSError("This formatter expect FileNode parameter, not %r" | |||
|
72 | % type(filenode)) | |||
|
73 | ||||
|
74 | def annotate_from_changeset(self, changeset): | |||
|
75 | """ | |||
|
76 | Returns full html line for single changeset per annotated line. | |||
|
77 | """ | |||
|
78 | if self.annotate_from_changeset_func: | |||
|
79 | return self.annotate_from_changeset_func(changeset) | |||
|
80 | else: | |||
|
81 | return ''.join((changeset.id, '\n')) | |||
|
82 | ||||
|
83 | def _wrap_tablelinenos(self, inner): | |||
|
84 | dummyoutfile = StringIO.StringIO() | |||
|
85 | lncount = 0 | |||
|
86 | for t, line in inner: | |||
|
87 | if t: | |||
|
88 | lncount += 1 | |||
|
89 | dummyoutfile.write(line) | |||
|
90 | ||||
|
91 | fl = self.linenostart | |||
|
92 | mw = len(str(lncount + fl - 1)) | |||
|
93 | sp = self.linenospecial | |||
|
94 | st = self.linenostep | |||
|
95 | la = self.lineanchors | |||
|
96 | aln = self.anchorlinenos | |||
|
97 | if sp: | |||
|
98 | lines = [] | |||
|
99 | ||||
|
100 | for i in range(fl, fl + lncount): | |||
|
101 | if i % st == 0: | |||
|
102 | if i % sp == 0: | |||
|
103 | if aln: | |||
|
104 | lines.append('<a href="#%s-%d" class="special">' | |||
|
105 | '%*d</a>' % | |||
|
106 | (la, i, mw, i)) | |||
|
107 | else: | |||
|
108 | lines.append('<span class="special">' | |||
|
109 | '%*d</span>' % (mw, i)) | |||
|
110 | else: | |||
|
111 | if aln: | |||
|
112 | lines.append('<a href="#%s-%d">' | |||
|
113 | '%*d</a>' % (la, i, mw, i)) | |||
|
114 | else: | |||
|
115 | lines.append('%*d' % (mw, i)) | |||
|
116 | else: | |||
|
117 | lines.append('') | |||
|
118 | ls = '\n'.join(lines) | |||
|
119 | else: | |||
|
120 | lines = [] | |||
|
121 | for i in range(fl, fl + lncount): | |||
|
122 | if i % st == 0: | |||
|
123 | if aln: | |||
|
124 | lines.append('<a href="#%s-%d">%*d</a>' \ | |||
|
125 | % (la, i, mw, i)) | |||
|
126 | else: | |||
|
127 | lines.append('%*d' % (mw, i)) | |||
|
128 | else: | |||
|
129 | lines.append('') | |||
|
130 | ls = '\n'.join(lines) | |||
|
131 | ||||
|
132 | annotate_changesets = [tup[1] for tup in self.filenode.annotate] | |||
|
133 | # If pygments cropped last lines break we need do that too | |||
|
134 | ln_cs = len(annotate_changesets) | |||
|
135 | ln_ = len(ls.splitlines()) | |||
|
136 | if ln_cs > ln_: | |||
|
137 | annotate_changesets = annotate_changesets[:ln_ - ln_cs] | |||
|
138 | annotate = ''.join((self.annotate_from_changeset(changeset) | |||
|
139 | for changeset in annotate_changesets)) | |||
|
140 | # in case you wonder about the seemingly redundant <div> here: | |||
|
141 | # since the content in the other cell also is wrapped in a div, | |||
|
142 | # some browsers in some configurations seem to mess up the formatting. | |||
|
143 | ''' | |||
|
144 | yield 0, ('<table class="%stable">' % self.cssclass + | |||
|
145 | '<tr><td class="linenos"><div class="linenodiv"><pre>' + | |||
|
146 | ls + '</pre></div></td>' + | |||
|
147 | '<td class="code">') | |||
|
148 | yield 0, dummyoutfile.getvalue() | |||
|
149 | yield 0, '</td></tr></table>' | |||
|
150 | ||||
|
151 | ''' | |||
|
152 | headers_row = [] | |||
|
153 | if self.headers: | |||
|
154 | headers_row = ['<tr class="annotate-header">'] | |||
|
155 | for key in self.order: | |||
|
156 | td = ''.join(('<td>', self.headers[key], '</td>')) | |||
|
157 | headers_row.append(td) | |||
|
158 | headers_row.append('</tr>') | |||
|
159 | ||||
|
160 | body_row_start = ['<tr>'] | |||
|
161 | for key in self.order: | |||
|
162 | if key == 'ls': | |||
|
163 | body_row_start.append( | |||
|
164 | '<td class="linenos"><div class="linenodiv"><pre>' + | |||
|
165 | ls + '</pre></div></td>') | |||
|
166 | elif key == 'annotate': | |||
|
167 | body_row_start.append( | |||
|
168 | '<td class="annotate"><div class="annotatediv"><pre>' + | |||
|
169 | annotate + '</pre></div></td>') | |||
|
170 | elif key == 'code': | |||
|
171 | body_row_start.append('<td class="code">') | |||
|
172 | yield 0, ('<table class="%stable">' % self.cssclass + | |||
|
173 | ''.join(headers_row) + | |||
|
174 | ''.join(body_row_start) | |||
|
175 | ) | |||
|
176 | yield 0, dummyoutfile.getvalue() | |||
|
177 | yield 0, '</td></tr></table>' |
@@ -0,0 +1,67 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | """ | |||
|
3 | vcs.utils.archivers | |||
|
4 | ~~~~~~~~~~~~~~~~~~~ | |||
|
5 | ||||
|
6 | set of archiver functions for creating archives from repository content | |||
|
7 | ||||
|
8 | :created_on: Jan 21, 2011 | |||
|
9 | :copyright: (c) 2010-2011 by Marcin Kuzminski, Lukasz Balcerzak. | |||
|
10 | """ | |||
|
11 | ||||
|
12 | ||||
|
13 | class BaseArchiver(object): | |||
|
14 | ||||
|
15 | def __init__(self): | |||
|
16 | self.archive_file = self._get_archive_file() | |||
|
17 | ||||
|
18 | def addfile(self): | |||
|
19 | """ | |||
|
20 | Adds a file to archive container | |||
|
21 | """ | |||
|
22 | pass | |||
|
23 | ||||
|
24 | def close(self): | |||
|
25 | """ | |||
|
26 | Closes and finalizes operation of archive container object | |||
|
27 | """ | |||
|
28 | self.archive_file.close() | |||
|
29 | ||||
|
30 | def _get_archive_file(self): | |||
|
31 | """ | |||
|
32 | Returns container for specific archive | |||
|
33 | """ | |||
|
34 | raise NotImplementedError() | |||
|
35 | ||||
|
36 | ||||
|
37 | class TarArchiver(BaseArchiver): | |||
|
38 | pass | |||
|
39 | ||||
|
40 | ||||
|
41 | class Tbz2Archiver(BaseArchiver): | |||
|
42 | pass | |||
|
43 | ||||
|
44 | ||||
|
45 | class TgzArchiver(BaseArchiver): | |||
|
46 | pass | |||
|
47 | ||||
|
48 | ||||
|
49 | class ZipArchiver(BaseArchiver): | |||
|
50 | pass | |||
|
51 | ||||
|
52 | ||||
|
53 | def get_archiver(self, kind): | |||
|
54 | """ | |||
|
55 | Returns instance of archiver class specific to given kind | |||
|
56 | ||||
|
57 | :param kind: archive kind | |||
|
58 | """ | |||
|
59 | ||||
|
60 | archivers = { | |||
|
61 | 'tar': TarArchiver, | |||
|
62 | 'tbz2': Tbz2Archiver, | |||
|
63 | 'tgz': TgzArchiver, | |||
|
64 | 'zip': ZipArchiver, | |||
|
65 | } | |||
|
66 | ||||
|
67 | return archivers[kind]() |
@@ -0,0 +1,47 b'' | |||||
|
1 | from mercurial import ui, config | |||
|
2 | ||||
|
3 | ||||
|
4 | def make_ui(self, path='hgwebdir.config'): | |||
|
5 | """ | |||
|
6 | A funcion that will read python rc files and make an ui from read options | |||
|
7 | ||||
|
8 | :param path: path to mercurial config file | |||
|
9 | """ | |||
|
10 | #propagated from mercurial documentation | |||
|
11 | sections = [ | |||
|
12 | 'alias', | |||
|
13 | 'auth', | |||
|
14 | 'decode/encode', | |||
|
15 | 'defaults', | |||
|
16 | 'diff', | |||
|
17 | 'email', | |||
|
18 | 'extensions', | |||
|
19 | 'format', | |||
|
20 | 'merge-patterns', | |||
|
21 | 'merge-tools', | |||
|
22 | 'hooks', | |||
|
23 | 'http_proxy', | |||
|
24 | 'smtp', | |||
|
25 | 'patch', | |||
|
26 | 'paths', | |||
|
27 | 'profiling', | |||
|
28 | 'server', | |||
|
29 | 'trusted', | |||
|
30 | 'ui', | |||
|
31 | 'web', | |||
|
32 | ] | |||
|
33 | ||||
|
34 | repos = path | |||
|
35 | baseui = ui.ui() | |||
|
36 | cfg = config.config() | |||
|
37 | cfg.read(repos) | |||
|
38 | self.paths = cfg.items('paths') | |||
|
39 | self.base_path = self.paths[0][1].replace('*', '') | |||
|
40 | self.check_repo_dir(self.paths) | |||
|
41 | self.set_statics(cfg) | |||
|
42 | ||||
|
43 | for section in sections: | |||
|
44 | for k, v in cfg.items(section): | |||
|
45 | baseui.setconfig(section, k, v) | |||
|
46 | ||||
|
47 | return baseui |
@@ -0,0 +1,13 b'' | |||||
|
1 | """ | |||
|
2 | Various utilities to work with Python < 2.7. | |||
|
3 | ||||
|
4 | Those utilities may be deleted once ``vcs`` stops support for older Python | |||
|
5 | versions. | |||
|
6 | """ | |||
|
7 | import sys | |||
|
8 | ||||
|
9 | ||||
|
10 | if sys.version_info >= (2, 7): | |||
|
11 | unittest = __import__('unittest') | |||
|
12 | else: | |||
|
13 | unittest = __import__('unittest2') |
@@ -0,0 +1,460 b'' | |||||
|
1 | # -*- coding: utf-8 -*- | |||
|
2 | # original copyright: 2007-2008 by Armin Ronacher | |||
|
3 | # licensed under the BSD license. | |||
|
4 | ||||
|
5 | import re | |||
|
6 | import difflib | |||
|
7 | import logging | |||
|
8 | ||||
|
9 | from difflib import unified_diff | |||
|
10 | from itertools import tee, imap | |||
|
11 | ||||
|
12 | from mercurial.match import match | |||
|
13 | ||||
|
14 | from rhodecode.lib.vcs.exceptions import VCSError | |||
|
15 | from rhodecode.lib.vcs.nodes import FileNode, NodeError | |||
|
16 | ||||
|
17 | ||||
|
18 | def get_udiff(filenode_old, filenode_new,show_whitespace=True): | |||
|
19 | """ | |||
|
20 | Returns unified diff between given ``filenode_old`` and ``filenode_new``. | |||
|
21 | """ | |||
|
22 | try: | |||
|
23 | filenode_old_date = filenode_old.last_changeset.date | |||
|
24 | except NodeError: | |||
|
25 | filenode_old_date = None | |||
|
26 | ||||
|
27 | try: | |||
|
28 | filenode_new_date = filenode_new.last_changeset.date | |||
|
29 | except NodeError: | |||
|
30 | filenode_new_date = None | |||
|
31 | ||||
|
32 | for filenode in (filenode_old, filenode_new): | |||
|
33 | if not isinstance(filenode, FileNode): | |||
|
34 | raise VCSError("Given object should be FileNode object, not %s" | |||
|
35 | % filenode.__class__) | |||
|
36 | ||||
|
37 | if filenode_old_date and filenode_new_date: | |||
|
38 | if not filenode_old_date < filenode_new_date: | |||
|
39 | logging.debug("Generating udiff for filenodes with not increasing " | |||
|
40 | "dates") | |||
|
41 | ||||
|
42 | vcs_udiff = unified_diff(filenode_old.content.splitlines(True), | |||
|
43 | filenode_new.content.splitlines(True), | |||
|
44 | filenode_old.name, | |||
|
45 | filenode_new.name, | |||
|
46 | filenode_old_date, | |||
|
47 | filenode_old_date) | |||
|
48 | return vcs_udiff | |||
|
49 | ||||
|
50 | ||||
|
51 | def get_gitdiff(filenode_old, filenode_new, ignore_whitespace=True): | |||
|
52 | """ | |||
|
53 | Returns git style diff between given ``filenode_old`` and ``filenode_new``. | |||
|
54 | ||||
|
55 | :param ignore_whitespace: ignore whitespaces in diff | |||
|
56 | """ | |||
|
57 | ||||
|
58 | for filenode in (filenode_old, filenode_new): | |||
|
59 | if not isinstance(filenode, FileNode): | |||
|
60 | raise VCSError("Given object should be FileNode object, not %s" | |||
|
61 | % filenode.__class__) | |||
|
62 | ||||
|
63 | old_raw_id = getattr(filenode_old.changeset, 'raw_id', '0' * 40) | |||
|
64 | new_raw_id = getattr(filenode_new.changeset, 'raw_id', '0' * 40) | |||
|
65 | ||||
|
66 | repo = filenode_new.changeset.repository | |||
|
67 | vcs_gitdiff = repo._get_diff(old_raw_id, new_raw_id, filenode_new.path, | |||
|
68 | ignore_whitespace) | |||
|
69 | ||||
|
70 | return vcs_gitdiff | |||
|
71 | ||||
|
72 | ||||
|
73 | class DiffProcessor(object): | |||
|
74 | """ | |||
|
75 | Give it a unified diff and it returns a list of the files that were | |||
|
76 | mentioned in the diff together with a dict of meta information that | |||
|
77 | can be used to render it in a HTML template. | |||
|
78 | """ | |||
|
79 | _chunk_re = re.compile(r'@@ -(\d+)(?:,(\d+))? \+(\d+)(?:,(\d+))? @@(.*)') | |||
|
80 | ||||
|
81 | def __init__(self, diff, differ='diff', format='udiff'): | |||
|
82 | """ | |||
|
83 | :param diff: a text in diff format or generator | |||
|
84 | :param format: format of diff passed, `udiff` or `gitdiff` | |||
|
85 | """ | |||
|
86 | if isinstance(diff, basestring): | |||
|
87 | diff = [diff] | |||
|
88 | ||||
|
89 | self.__udiff = diff | |||
|
90 | self.__format = format | |||
|
91 | self.adds = 0 | |||
|
92 | self.removes = 0 | |||
|
93 | ||||
|
94 | if isinstance(self.__udiff, basestring): | |||
|
95 | self.lines = iter(self.__udiff.splitlines(1)) | |||
|
96 | ||||
|
97 | elif self.__format == 'gitdiff': | |||
|
98 | udiff_copy = self.copy_iterator() | |||
|
99 | self.lines = imap(self.escaper, self._parse_gitdiff(udiff_copy)) | |||
|
100 | else: | |||
|
101 | udiff_copy = self.copy_iterator() | |||
|
102 | self.lines = imap(self.escaper, udiff_copy) | |||
|
103 | ||||
|
104 | # Select a differ. | |||
|
105 | if differ == 'difflib': | |||
|
106 | self.differ = self._highlight_line_difflib | |||
|
107 | else: | |||
|
108 | self.differ = self._highlight_line_udiff | |||
|
109 | ||||
|
110 | def escaper(self, string): | |||
|
111 | return string.replace('<', '<').replace('>', '>') | |||
|
112 | ||||
|
113 | def copy_iterator(self): | |||
|
114 | """ | |||
|
115 | make a fresh copy of generator, we should not iterate thru | |||
|
116 | an original as it's needed for repeating operations on | |||
|
117 | this instance of DiffProcessor | |||
|
118 | """ | |||
|
119 | self.__udiff, iterator_copy = tee(self.__udiff) | |||
|
120 | return iterator_copy | |||
|
121 | ||||
|
122 | def _extract_rev(self, line1, line2): | |||
|
123 | """ | |||
|
124 | Extract the filename and revision hint from a line. | |||
|
125 | """ | |||
|
126 | ||||
|
127 | try: | |||
|
128 | if line1.startswith('--- ') and line2.startswith('+++ '): | |||
|
129 | l1 = line1[4:].split(None, 1) | |||
|
130 | old_filename = l1[0].lstrip('a/') if len(l1) >= 1 else None | |||
|
131 | old_rev = l1[1] if len(l1) == 2 else 'old' | |||
|
132 | ||||
|
133 | l2 = line2[4:].split(None, 1) | |||
|
134 | new_filename = l2[0].lstrip('b/') if len(l1) >= 1 else None | |||
|
135 | new_rev = l2[1] if len(l2) == 2 else 'new' | |||
|
136 | ||||
|
137 | filename = old_filename if (old_filename != | |||
|
138 | 'dev/null') else new_filename | |||
|
139 | ||||
|
140 | return filename, new_rev, old_rev | |||
|
141 | except (ValueError, IndexError): | |||
|
142 | pass | |||
|
143 | ||||
|
144 | return None, None, None | |||
|
145 | ||||
|
146 | def _parse_gitdiff(self, diffiterator): | |||
|
147 | def line_decoder(l): | |||
|
148 | if l.startswith('+') and not l.startswith('+++'): | |||
|
149 | self.adds += 1 | |||
|
150 | elif l.startswith('-') and not l.startswith('---'): | |||
|
151 | self.removes += 1 | |||
|
152 | return l.decode('utf8', 'replace') | |||
|
153 | ||||
|
154 | output = list(diffiterator) | |||
|
155 | size = len(output) | |||
|
156 | ||||
|
157 | if size == 2: | |||
|
158 | l = [] | |||
|
159 | l.extend([output[0]]) | |||
|
160 | l.extend(output[1].splitlines(1)) | |||
|
161 | return map(line_decoder, l) | |||
|
162 | elif size == 1: | |||
|
163 | return map(line_decoder, output[0].splitlines(1)) | |||
|
164 | elif size == 0: | |||
|
165 | return [] | |||
|
166 | ||||
|
167 | raise Exception('wrong size of diff %s' % size) | |||
|
168 | ||||
|
169 | def _highlight_line_difflib(self, line, next): | |||
|
170 | """ | |||
|
171 | Highlight inline changes in both lines. | |||
|
172 | """ | |||
|
173 | ||||
|
174 | if line['action'] == 'del': | |||
|
175 | old, new = line, next | |||
|
176 | else: | |||
|
177 | old, new = next, line | |||
|
178 | ||||
|
179 | oldwords = re.split(r'(\W)', old['line']) | |||
|
180 | newwords = re.split(r'(\W)', new['line']) | |||
|
181 | ||||
|
182 | sequence = difflib.SequenceMatcher(None, oldwords, newwords) | |||
|
183 | ||||
|
184 | oldfragments, newfragments = [], [] | |||
|
185 | for tag, i1, i2, j1, j2 in sequence.get_opcodes(): | |||
|
186 | oldfrag = ''.join(oldwords[i1:i2]) | |||
|
187 | newfrag = ''.join(newwords[j1:j2]) | |||
|
188 | if tag != 'equal': | |||
|
189 | if oldfrag: | |||
|
190 | oldfrag = '<del>%s</del>' % oldfrag | |||
|
191 | if newfrag: | |||
|
192 | newfrag = '<ins>%s</ins>' % newfrag | |||
|
193 | oldfragments.append(oldfrag) | |||
|
194 | newfragments.append(newfrag) | |||
|
195 | ||||
|
196 | old['line'] = "".join(oldfragments) | |||
|
197 | new['line'] = "".join(newfragments) | |||
|
198 | ||||
|
199 | def _highlight_line_udiff(self, line, next): | |||
|
200 | """ | |||
|
201 | Highlight inline changes in both lines. | |||
|
202 | """ | |||
|
203 | start = 0 | |||
|
204 | limit = min(len(line['line']), len(next['line'])) | |||
|
205 | while start < limit and line['line'][start] == next['line'][start]: | |||
|
206 | start += 1 | |||
|
207 | end = -1 | |||
|
208 | limit -= start | |||
|
209 | while -end <= limit and line['line'][end] == next['line'][end]: | |||
|
210 | end -= 1 | |||
|
211 | end += 1 | |||
|
212 | if start or end: | |||
|
213 | def do(l): | |||
|
214 | last = end + len(l['line']) | |||
|
215 | if l['action'] == 'add': | |||
|
216 | tag = 'ins' | |||
|
217 | else: | |||
|
218 | tag = 'del' | |||
|
219 | l['line'] = '%s<%s>%s</%s>%s' % ( | |||
|
220 | l['line'][:start], | |||
|
221 | tag, | |||
|
222 | l['line'][start:last], | |||
|
223 | tag, | |||
|
224 | l['line'][last:] | |||
|
225 | ) | |||
|
226 | do(line) | |||
|
227 | do(next) | |||
|
228 | ||||
|
229 | def _parse_udiff(self): | |||
|
230 | """ | |||
|
231 | Parse the diff an return data for the template. | |||
|
232 | """ | |||
|
233 | lineiter = self.lines | |||
|
234 | files = [] | |||
|
235 | try: | |||
|
236 | line = lineiter.next() | |||
|
237 | # skip first context | |||
|
238 | skipfirst = True | |||
|
239 | while 1: | |||
|
240 | # continue until we found the old file | |||
|
241 | if not line.startswith('--- '): | |||
|
242 | line = lineiter.next() | |||
|
243 | continue | |||
|
244 | ||||
|
245 | chunks = [] | |||
|
246 | filename, old_rev, new_rev = \ | |||
|
247 | self._extract_rev(line, lineiter.next()) | |||
|
248 | files.append({ | |||
|
249 | 'filename': filename, | |||
|
250 | 'old_revision': old_rev, | |||
|
251 | 'new_revision': new_rev, | |||
|
252 | 'chunks': chunks | |||
|
253 | }) | |||
|
254 | ||||
|
255 | line = lineiter.next() | |||
|
256 | while line: | |||
|
257 | match = self._chunk_re.match(line) | |||
|
258 | if not match: | |||
|
259 | break | |||
|
260 | ||||
|
261 | lines = [] | |||
|
262 | chunks.append(lines) | |||
|
263 | ||||
|
264 | old_line, old_end, new_line, new_end = \ | |||
|
265 | [int(x or 1) for x in match.groups()[:-1]] | |||
|
266 | old_line -= 1 | |||
|
267 | new_line -= 1 | |||
|
268 | context = len(match.groups()) == 5 | |||
|
269 | old_end += old_line | |||
|
270 | new_end += new_line | |||
|
271 | ||||
|
272 | if context: | |||
|
273 | if not skipfirst: | |||
|
274 | lines.append({ | |||
|
275 | 'old_lineno': '...', | |||
|
276 | 'new_lineno': '...', | |||
|
277 | 'action': 'context', | |||
|
278 | 'line': line, | |||
|
279 | }) | |||
|
280 | else: | |||
|
281 | skipfirst = False | |||
|
282 | ||||
|
283 | line = lineiter.next() | |||
|
284 | while old_line < old_end or new_line < new_end: | |||
|
285 | if line: | |||
|
286 | command, line = line[0], line[1:] | |||
|
287 | else: | |||
|
288 | command = ' ' | |||
|
289 | affects_old = affects_new = False | |||
|
290 | ||||
|
291 | # ignore those if we don't expect them | |||
|
292 | if command in '#@': | |||
|
293 | continue | |||
|
294 | elif command == '+': | |||
|
295 | affects_new = True | |||
|
296 | action = 'add' | |||
|
297 | elif command == '-': | |||
|
298 | affects_old = True | |||
|
299 | action = 'del' | |||
|
300 | else: | |||
|
301 | affects_old = affects_new = True | |||
|
302 | action = 'unmod' | |||
|
303 | ||||
|
304 | old_line += affects_old | |||
|
305 | new_line += affects_new | |||
|
306 | lines.append({ | |||
|
307 | 'old_lineno': affects_old and old_line or '', | |||
|
308 | 'new_lineno': affects_new and new_line or '', | |||
|
309 | 'action': action, | |||
|
310 | 'line': line | |||
|
311 | }) | |||
|
312 | line = lineiter.next() | |||
|
313 | ||||
|
314 | except StopIteration: | |||
|
315 | pass | |||
|
316 | ||||
|
317 | # highlight inline changes | |||
|
318 | for file in files: | |||
|
319 | for chunk in chunks: | |||
|
320 | lineiter = iter(chunk) | |||
|
321 | #first = True | |||
|
322 | try: | |||
|
323 | while 1: | |||
|
324 | line = lineiter.next() | |||
|
325 | if line['action'] != 'unmod': | |||
|
326 | nextline = lineiter.next() | |||
|
327 | if nextline['action'] == 'unmod' or \ | |||
|
328 | nextline['action'] == line['action']: | |||
|
329 | continue | |||
|
330 | self.differ(line, nextline) | |||
|
331 | except StopIteration: | |||
|
332 | pass | |||
|
333 | ||||
|
334 | return files | |||
|
335 | ||||
|
336 | def prepare(self): | |||
|
337 | """ | |||
|
338 | Prepare the passed udiff for HTML rendering. It'l return a list | |||
|
339 | of dicts | |||
|
340 | """ | |||
|
341 | return self._parse_udiff() | |||
|
342 | ||||
|
343 | def _safe_id(self, idstring): | |||
|
344 | """Make a string safe for including in an id attribute. | |||
|
345 | ||||
|
346 | The HTML spec says that id attributes 'must begin with | |||
|
347 | a letter ([A-Za-z]) and may be followed by any number | |||
|
348 | of letters, digits ([0-9]), hyphens ("-"), underscores | |||
|
349 | ("_"), colons (":"), and periods (".")'. These regexps | |||
|
350 | are slightly over-zealous, in that they remove colons | |||
|
351 | and periods unnecessarily. | |||
|
352 | ||||
|
353 | Whitespace is transformed into underscores, and then | |||
|
354 | anything which is not a hyphen or a character that | |||
|
355 | matches \w (alphanumerics and underscore) is removed. | |||
|
356 | ||||
|
357 | """ | |||
|
358 | # Transform all whitespace to underscore | |||
|
359 | idstring = re.sub(r'\s', "_", '%s' % idstring) | |||
|
360 | # Remove everything that is not a hyphen or a member of \w | |||
|
361 | idstring = re.sub(r'(?!-)\W', "", idstring).lower() | |||
|
362 | return idstring | |||
|
363 | ||||
|
364 | def raw_diff(self): | |||
|
365 | """ | |||
|
366 | Returns raw string as udiff | |||
|
367 | """ | |||
|
368 | udiff_copy = self.copy_iterator() | |||
|
369 | if self.__format == 'gitdiff': | |||
|
370 | udiff_copy = self._parse_gitdiff(udiff_copy) | |||
|
371 | return u''.join(udiff_copy) | |||
|
372 | ||||
|
373 | def as_html(self, table_class='code-difftable', line_class='line', | |||
|
374 | new_lineno_class='lineno old', old_lineno_class='lineno new', | |||
|
375 | code_class='code'): | |||
|
376 | """ | |||
|
377 | Return udiff as html table with customized css classes | |||
|
378 | """ | |||
|
379 | def _link_to_if(condition, label, url): | |||
|
380 | """ | |||
|
381 | Generates a link if condition is meet or just the label if not. | |||
|
382 | """ | |||
|
383 | ||||
|
384 | if condition: | |||
|
385 | return '''<a href="%(url)s">%(label)s</a>''' % {'url': url, | |||
|
386 | 'label': label} | |||
|
387 | else: | |||
|
388 | return label | |||
|
389 | diff_lines = self.prepare() | |||
|
390 | _html_empty = True | |||
|
391 | _html = [] | |||
|
392 | _html.append('''<table class="%(table_class)s">\n''' \ | |||
|
393 | % {'table_class': table_class}) | |||
|
394 | for diff in diff_lines: | |||
|
395 | for line in diff['chunks']: | |||
|
396 | _html_empty = False | |||
|
397 | for change in line: | |||
|
398 | _html.append('''<tr class="%(line_class)s %(action)s">\n''' \ | |||
|
399 | % {'line_class': line_class, | |||
|
400 | 'action': change['action']}) | |||
|
401 | anchor_old_id = '' | |||
|
402 | anchor_new_id = '' | |||
|
403 | anchor_old = "%(filename)s_o%(oldline_no)s" % \ | |||
|
404 | {'filename': self._safe_id(diff['filename']), | |||
|
405 | 'oldline_no': change['old_lineno']} | |||
|
406 | anchor_new = "%(filename)s_n%(oldline_no)s" % \ | |||
|
407 | {'filename': self._safe_id(diff['filename']), | |||
|
408 | 'oldline_no': change['new_lineno']} | |||
|
409 | cond_old = change['old_lineno'] != '...' and \ | |||
|
410 | change['old_lineno'] | |||
|
411 | cond_new = change['new_lineno'] != '...' and \ | |||
|
412 | change['new_lineno'] | |||
|
413 | if cond_old: | |||
|
414 | anchor_old_id = 'id="%s"' % anchor_old | |||
|
415 | if cond_new: | |||
|
416 | anchor_new_id = 'id="%s"' % anchor_new | |||
|
417 | ########################################################### | |||
|
418 | # OLD LINE NUMBER | |||
|
419 | ########################################################### | |||
|
420 | _html.append('''\t<td %(a_id)s class="%(old_lineno_cls)s">''' \ | |||
|
421 | % {'a_id': anchor_old_id, | |||
|
422 | 'old_lineno_cls': old_lineno_class}) | |||
|
423 | ||||
|
424 | _html.append('''<pre>%(link)s</pre>''' \ | |||
|
425 | % {'link': | |||
|
426 | _link_to_if(cond_old, change['old_lineno'], '#%s' \ | |||
|
427 | % anchor_old)}) | |||
|
428 | _html.append('''</td>\n''') | |||
|
429 | ########################################################### | |||
|
430 | # NEW LINE NUMBER | |||
|
431 | ########################################################### | |||
|
432 | ||||
|
433 | _html.append('''\t<td %(a_id)s class="%(new_lineno_cls)s">''' \ | |||
|
434 | % {'a_id': anchor_new_id, | |||
|
435 | 'new_lineno_cls': new_lineno_class}) | |||
|
436 | ||||
|
437 | _html.append('''<pre>%(link)s</pre>''' \ | |||
|
438 | % {'link': | |||
|
439 | _link_to_if(cond_new, change['new_lineno'], '#%s' \ | |||
|
440 | % anchor_new)}) | |||
|
441 | _html.append('''</td>\n''') | |||
|
442 | ########################################################### | |||
|
443 | # CODE | |||
|
444 | ########################################################### | |||
|
445 | _html.append('''\t<td class="%(code_class)s">''' \ | |||
|
446 | % {'code_class': code_class}) | |||
|
447 | _html.append('''\n\t\t<pre>%(code)s</pre>\n''' \ | |||
|
448 | % {'code': change['line']}) | |||
|
449 | _html.append('''\t</td>''') | |||
|
450 | _html.append('''\n</tr>\n''') | |||
|
451 | _html.append('''</table>''') | |||
|
452 | if _html_empty: | |||
|
453 | return None | |||
|
454 | return ''.join(_html) | |||
|
455 | ||||
|
456 | def stat(self): | |||
|
457 | """ | |||
|
458 | Returns tuple of adde,and removed lines for this instance | |||
|
459 | """ | |||
|
460 | return self.adds, self.removes |
@@ -0,0 +1,13 b'' | |||||
|
1 | import imp | |||
|
2 | ||||
|
3 | ||||
|
4 | def create_module(name, path): | |||
|
5 | """ | |||
|
6 | Returns module created *on the fly*. Returned module would have name same | |||
|
7 | as given ``name`` and would contain code read from file at the given | |||
|
8 | ``path`` (it may also be a zip or package containing *__main__* module). | |||
|
9 | """ | |||
|
10 | module = imp.new_module(name) | |||
|
11 | module.__file__ = path | |||
|
12 | execfile(path, module.__dict__) | |||
|
13 | return module |
@@ -0,0 +1,28 b'' | |||||
|
1 | def filesizeformat(bytes, sep=' '): | |||
|
2 | """ | |||
|
3 | Formats the value like a 'human-readable' file size (i.e. 13 KB, 4.1 MB, | |||
|
4 | 102 B, 2.3 GB etc). | |||
|
5 | ||||
|
6 | Grabbed from Django (http://www.djangoproject.com), slightly modified. | |||
|
7 | ||||
|
8 | :param bytes: size in bytes (as integer) | |||
|
9 | :param sep: string separator between number and abbreviation | |||
|
10 | """ | |||
|
11 | try: | |||
|
12 | bytes = float(bytes) | |||
|
13 | except (TypeError, ValueError, UnicodeDecodeError): | |||
|
14 | return '0%sB' % sep | |||
|
15 | ||||
|
16 | if bytes < 1024: | |||
|
17 | size = bytes | |||
|
18 | template = '%.0f%sB' | |||
|
19 | elif bytes < 1024 * 1024: | |||
|
20 | size = bytes / 1024 | |||
|
21 | template = '%.0f%sKB' | |||
|
22 | elif bytes < 1024 * 1024 * 1024: | |||
|
23 | size = bytes / 1024 / 1024 | |||
|
24 | template = '%.1f%sMB' | |||
|
25 | else: | |||
|
26 | size = bytes / 1024 / 1024 / 1024 | |||
|
27 | template = '%.2f%sGB' | |||
|
28 | return template % (size, sep) |
@@ -0,0 +1,252 b'' | |||||
|
1 | """ | |||
|
2 | Utitlites aimed to help achieve mostly basic tasks. | |||
|
3 | """ | |||
|
4 | from __future__ import division | |||
|
5 | ||||
|
6 | import re | |||
|
7 | import time | |||
|
8 | import datetime | |||
|
9 | import os.path | |||
|
10 | from subprocess import Popen, PIPE | |||
|
11 | from rhodecode.lib.vcs.exceptions import VCSError | |||
|
12 | from rhodecode.lib.vcs.exceptions import RepositoryError | |||
|
13 | from rhodecode.lib.vcs.utils.paths import abspath | |||
|
14 | ||||
|
15 | ALIASES = ['hg', 'git'] | |||
|
16 | ||||
|
17 | ||||
|
18 | def get_scm(path, search_recursively=False, explicit_alias=None): | |||
|
19 | """ | |||
|
20 | Returns one of alias from ``ALIASES`` (in order of precedence same as | |||
|
21 | shortcuts given in ``ALIASES``) and top working dir path for the given | |||
|
22 | argument. If no scm-specific directory is found or more than one scm is | |||
|
23 | found at that directory, ``VCSError`` is raised. | |||
|
24 | ||||
|
25 | :param search_recursively: if set to ``True``, this function would try to | |||
|
26 | move up to parent directory every time no scm is recognized for the | |||
|
27 | currently checked path. Default: ``False``. | |||
|
28 | :param explicit_alias: can be one of available backend aliases, when given | |||
|
29 | it will return given explicit alias in repositories under more than one | |||
|
30 | version control, if explicit_alias is different than found it will raise | |||
|
31 | VCSError | |||
|
32 | """ | |||
|
33 | if not os.path.isdir(path): | |||
|
34 | raise VCSError("Given path %s is not a directory" % path) | |||
|
35 | ||||
|
36 | def get_scms(path): | |||
|
37 | return [(scm, path) for scm in get_scms_for_path(path)] | |||
|
38 | ||||
|
39 | found_scms = get_scms(path) | |||
|
40 | while not found_scms and search_recursively: | |||
|
41 | newpath = abspath(path, '..') | |||
|
42 | if newpath == path: | |||
|
43 | break | |||
|
44 | path = newpath | |||
|
45 | found_scms = get_scms(path) | |||
|
46 | ||||
|
47 | if len(found_scms) > 1: | |||
|
48 | for scm in found_scms: | |||
|
49 | if scm[0] == explicit_alias: | |||
|
50 | return scm | |||
|
51 | raise VCSError('More than one [%s] scm found at given path %s' | |||
|
52 | % (','.join((x[0] for x in found_scms)), path)) | |||
|
53 | ||||
|
54 | if len(found_scms) is 0: | |||
|
55 | raise VCSError('No scm found at given path %s' % path) | |||
|
56 | ||||
|
57 | return found_scms[0] | |||
|
58 | ||||
|
59 | ||||
|
60 | def get_scms_for_path(path): | |||
|
61 | """ | |||
|
62 | Returns all scm's found at the given path. If no scm is recognized | |||
|
63 | - empty list is returned. | |||
|
64 | ||||
|
65 | :param path: path to directory which should be checked. May be callable. | |||
|
66 | ||||
|
67 | :raises VCSError: if given ``path`` is not a directory | |||
|
68 | """ | |||
|
69 | from rhodecode.lib.vcs.backends import get_backend | |||
|
70 | if hasattr(path, '__call__'): | |||
|
71 | path = path() | |||
|
72 | if not os.path.isdir(path): | |||
|
73 | raise VCSError("Given path %r is not a directory" % path) | |||
|
74 | ||||
|
75 | result = [] | |||
|
76 | for key in ALIASES: | |||
|
77 | dirname = os.path.join(path, '.' + key) | |||
|
78 | if os.path.isdir(dirname): | |||
|
79 | result.append(key) | |||
|
80 | continue | |||
|
81 | # We still need to check if it's not bare repository as | |||
|
82 | # bare repos don't have working directories | |||
|
83 | try: | |||
|
84 | get_backend(key)(path) | |||
|
85 | result.append(key) | |||
|
86 | continue | |||
|
87 | except RepositoryError: | |||
|
88 | # Wrong backend | |||
|
89 | pass | |||
|
90 | except VCSError: | |||
|
91 | # No backend at all | |||
|
92 | pass | |||
|
93 | return result | |||
|
94 | ||||
|
95 | ||||
|
96 | def get_repo_paths(path): | |||
|
97 | """ | |||
|
98 | Returns path's subdirectories which seems to be a repository. | |||
|
99 | """ | |||
|
100 | repo_paths = [] | |||
|
101 | dirnames = (os.path.abspath(dirname) for dirname in os.listdir(path)) | |||
|
102 | for dirname in dirnames: | |||
|
103 | try: | |||
|
104 | get_scm(dirname) | |||
|
105 | repo_paths.append(dirname) | |||
|
106 | except VCSError: | |||
|
107 | pass | |||
|
108 | return repo_paths | |||
|
109 | ||||
|
110 | ||||
|
111 | def run_command(cmd, *args): | |||
|
112 | """ | |||
|
113 | Runs command on the system with given ``args``. | |||
|
114 | """ | |||
|
115 | command = ' '.join((cmd, args)) | |||
|
116 | p = Popen(command, shell=True, stdout=PIPE, stderr=PIPE) | |||
|
117 | stdout, stderr = p.communicate() | |||
|
118 | return p.retcode, stdout, stderr | |||
|
119 | ||||
|
120 | ||||
|
121 | def get_highlighted_code(name, code, type='terminal'): | |||
|
122 | """ | |||
|
123 | If pygments are available on the system | |||
|
124 | then returned output is colored. Otherwise | |||
|
125 | unchanged content is returned. | |||
|
126 | """ | |||
|
127 | import logging | |||
|
128 | try: | |||
|
129 | import pygments | |||
|
130 | pygments | |||
|
131 | except ImportError: | |||
|
132 | return code | |||
|
133 | from pygments import highlight | |||
|
134 | from pygments.lexers import guess_lexer_for_filename, ClassNotFound | |||
|
135 | from pygments.formatters import TerminalFormatter | |||
|
136 | ||||
|
137 | try: | |||
|
138 | lexer = guess_lexer_for_filename(name, code) | |||
|
139 | formatter = TerminalFormatter() | |||
|
140 | content = highlight(code, lexer, formatter) | |||
|
141 | except ClassNotFound: | |||
|
142 | logging.debug("Couldn't guess Lexer, will not use pygments.") | |||
|
143 | content = code | |||
|
144 | return content | |||
|
145 | ||||
|
146 | def parse_changesets(text): | |||
|
147 | """ | |||
|
148 | Returns dictionary with *start*, *main* and *end* ids. | |||
|
149 | ||||
|
150 | Examples:: | |||
|
151 | ||||
|
152 | >>> parse_changesets('aaabbb') | |||
|
153 | {'start': None, 'main': 'aaabbb', 'end': None} | |||
|
154 | >>> parse_changesets('aaabbb..cccddd') | |||
|
155 | {'start': 'aaabbb', 'main': None, 'end': 'cccddd'} | |||
|
156 | ||||
|
157 | """ | |||
|
158 | text = text.strip() | |||
|
159 | CID_RE = r'[a-zA-Z0-9]+' | |||
|
160 | if not '..' in text: | |||
|
161 | m = re.match(r'^(?P<cid>%s)$' % CID_RE, text) | |||
|
162 | if m: | |||
|
163 | return { | |||
|
164 | 'start': None, | |||
|
165 | 'main': text, | |||
|
166 | 'end': None, | |||
|
167 | } | |||
|
168 | else: | |||
|
169 | RE = r'^(?P<start>%s)?\.{2,3}(?P<end>%s)?$' % (CID_RE, CID_RE) | |||
|
170 | m = re.match(RE, text) | |||
|
171 | if m: | |||
|
172 | result = m.groupdict() | |||
|
173 | result['main'] = None | |||
|
174 | return result | |||
|
175 | raise ValueError("IDs not recognized") | |||
|
176 | ||||
|
177 | def parse_datetime(text): | |||
|
178 | """ | |||
|
179 | Parses given text and returns ``datetime.datetime`` instance or raises | |||
|
180 | ``ValueError``. | |||
|
181 | ||||
|
182 | :param text: string of desired date/datetime or something more verbose, | |||
|
183 | like *yesterday*, *2weeks 3days*, etc. | |||
|
184 | """ | |||
|
185 | ||||
|
186 | text = text.strip().lower() | |||
|
187 | ||||
|
188 | INPUT_FORMATS = ( | |||
|
189 | '%Y-%m-%d %H:%M:%S', | |||
|
190 | '%Y-%m-%d %H:%M', | |||
|
191 | '%Y-%m-%d', | |||
|
192 | '%m/%d/%Y %H:%M:%S', | |||
|
193 | '%m/%d/%Y %H:%M', | |||
|
194 | '%m/%d/%Y', | |||
|
195 | '%m/%d/%y %H:%M:%S', | |||
|
196 | '%m/%d/%y %H:%M', | |||
|
197 | '%m/%d/%y', | |||
|
198 | ) | |||
|
199 | for format in INPUT_FORMATS: | |||
|
200 | try: | |||
|
201 | return datetime.datetime(*time.strptime(text, format)[:6]) | |||
|
202 | except ValueError: | |||
|
203 | pass | |||
|
204 | ||||
|
205 | # Try descriptive texts | |||
|
206 | if text == 'tomorrow': | |||
|
207 | future = datetime.datetime.now() + datetime.timedelta(days=1) | |||
|
208 | args = future.timetuple()[:3] + (23, 59, 59) | |||
|
209 | return datetime.datetime(*args) | |||
|
210 | elif text == 'today': | |||
|
211 | return datetime.datetime(*datetime.datetime.today().timetuple()[:3]) | |||
|
212 | elif text == 'now': | |||
|
213 | return datetime.datetime.now() | |||
|
214 | elif text == 'yesterday': | |||
|
215 | past = datetime.datetime.now() - datetime.timedelta(days=1) | |||
|
216 | return datetime.datetime(*past.timetuple()[:3]) | |||
|
217 | else: | |||
|
218 | days = 0 | |||
|
219 | matched = re.match( | |||
|
220 | r'^((?P<weeks>\d+) ?w(eeks?)?)? ?((?P<days>\d+) ?d(ays?)?)?$', text) | |||
|
221 | if matched: | |||
|
222 | groupdict = matched.groupdict() | |||
|
223 | if groupdict['days']: | |||
|
224 | days += int(matched.groupdict()['days']) | |||
|
225 | if groupdict['weeks']: | |||
|
226 | days += int(matched.groupdict()['weeks']) * 7 | |||
|
227 | past = datetime.datetime.now() - datetime.timedelta(days=days) | |||
|
228 | return datetime.datetime(*past.timetuple()[:3]) | |||
|
229 | ||||
|
230 | raise ValueError('Wrong date: "%s"' % text) | |||
|
231 | ||||
|
232 | ||||
|
233 | def get_dict_for_attrs(obj, attrs): | |||
|
234 | """ | |||
|
235 | Returns dictionary for each attribute from given ``obj``. | |||
|
236 | """ | |||
|
237 | data = {} | |||
|
238 | for attr in attrs: | |||
|
239 | data[attr] = getattr(obj, attr) | |||
|
240 | return data | |||
|
241 | ||||
|
242 | ||||
|
243 | def get_total_seconds(timedelta): | |||
|
244 | """ | |||
|
245 | Backported for Python 2.5. | |||
|
246 | ||||
|
247 | See http://docs.python.org/library/datetime.html. | |||
|
248 | """ | |||
|
249 | return ((timedelta.microseconds + ( | |||
|
250 | timedelta.seconds + | |||
|
251 | timedelta.days * 24 * 60 * 60 | |||
|
252 | ) * 10**6) / 10**6) |
@@ -0,0 +1,12 b'' | |||||
|
1 | """Mercurial libs compatibility | |||
|
2 | ||||
|
3 | """ | |||
|
4 | from mercurial import archival, merge as hg_merge, patch, ui | |||
|
5 | from mercurial.commands import clone, nullid, pull | |||
|
6 | from mercurial.context import memctx, memfilectx | |||
|
7 | from mercurial.error import RepoError, RepoLookupError, Abort | |||
|
8 | from mercurial.hgweb.common import get_contact | |||
|
9 | from mercurial.localrepo import localrepository | |||
|
10 | from mercurial.match import match | |||
|
11 | from mercurial.mdiff import diffopts | |||
|
12 | from mercurial.node import hex |
@@ -0,0 +1,27 b'' | |||||
|
1 | from rhodecode.lib.vcs.exceptions import VCSError | |||
|
2 | ||||
|
3 | ||||
|
4 | def import_class(class_path): | |||
|
5 | """ | |||
|
6 | Returns class from the given path. | |||
|
7 | ||||
|
8 | For example, in order to get class located at | |||
|
9 | ``vcs.backends.hg.MercurialRepository``: | |||
|
10 | ||||
|
11 | try: | |||
|
12 | hgrepo = import_class('vcs.backends.hg.MercurialRepository') | |||
|
13 | except VCSError: | |||
|
14 | # hadle error | |||
|
15 | """ | |||
|
16 | splitted = class_path.split('.') | |||
|
17 | mod_path = '.'.join(splitted[:-1]) | |||
|
18 | class_name = splitted[-1] | |||
|
19 | try: | |||
|
20 | class_mod = __import__(mod_path, {}, {}, [class_name]) | |||
|
21 | except ImportError, err: | |||
|
22 | msg = "There was problem while trying to import backend class. "\ | |||
|
23 | "Original error was:\n%s" % err | |||
|
24 | raise VCSError(msg) | |||
|
25 | cls = getattr(class_mod, class_name) | |||
|
26 | ||||
|
27 | return cls |
@@ -0,0 +1,27 b'' | |||||
|
1 | class LazyProperty(object): | |||
|
2 | """ | |||
|
3 | Decorator for easier creation of ``property`` from potentially expensive to | |||
|
4 | calculate attribute of the class. | |||
|
5 | ||||
|
6 | Usage:: | |||
|
7 | ||||
|
8 | class Foo(object): | |||
|
9 | @LazyProperty | |||
|
10 | def bar(self): | |||
|
11 | print 'Calculating self._bar' | |||
|
12 | return 42 | |||
|
13 | ||||
|
14 | Taken from http://blog.pythonisito.com/2008/08/lazy-descriptors.html and | |||
|
15 | used widely. | |||
|
16 | """ | |||
|
17 | ||||
|
18 | def __init__(self, func): | |||
|
19 | self._func = func | |||
|
20 | self.__name__ = func.__name__ | |||
|
21 | self.__doc__ = func.__doc__ | |||
|
22 | ||||
|
23 | def __get__(self, obj, klass=None): | |||
|
24 | if obj is None: | |||
|
25 | return None | |||
|
26 | result = obj.__dict__[self.__name__] = self._func(obj) | |||
|
27 | return result |
@@ -0,0 +1,72 b'' | |||||
|
1 | import os | |||
|
2 | ||||
|
3 | ||||
|
4 | class LockFile(object): | |||
|
5 | """Provides methods to obtain, check for, and release a file based lock which | |||
|
6 | should be used to handle concurrent access to the same file. | |||
|
7 | ||||
|
8 | As we are a utility class to be derived from, we only use protected methods. | |||
|
9 | ||||
|
10 | Locks will automatically be released on destruction""" | |||
|
11 | __slots__ = ("_file_path", "_owns_lock") | |||
|
12 | ||||
|
13 | def __init__(self, file_path): | |||
|
14 | self._file_path = file_path | |||
|
15 | self._owns_lock = False | |||
|
16 | ||||
|
17 | def __del__(self): | |||
|
18 | self._release_lock() | |||
|
19 | ||||
|
20 | def _lock_file_path(self): | |||
|
21 | """:return: Path to lockfile""" | |||
|
22 | return "%s.lock" % (self._file_path) | |||
|
23 | ||||
|
24 | def _has_lock(self): | |||
|
25 | """:return: True if we have a lock and if the lockfile still exists | |||
|
26 | :raise AssertionError: if our lock-file does not exist""" | |||
|
27 | if not self._owns_lock: | |||
|
28 | return False | |||
|
29 | ||||
|
30 | return True | |||
|
31 | ||||
|
32 | def _obtain_lock_or_raise(self): | |||
|
33 | """Create a lock file as flag for other instances, mark our instance as lock-holder | |||
|
34 | ||||
|
35 | :raise IOError: if a lock was already present or a lock file could not be written""" | |||
|
36 | if self._has_lock(): | |||
|
37 | return | |||
|
38 | lock_file = self._lock_file_path() | |||
|
39 | if os.path.isfile(lock_file): | |||
|
40 | raise IOError("Lock for file %r did already exist, delete %r in case the lock is illegal" % (self._file_path, lock_file)) | |||
|
41 | ||||
|
42 | try: | |||
|
43 | fd = os.open(lock_file, os.O_WRONLY | os.O_CREAT | os.O_EXCL, 0) | |||
|
44 | os.close(fd) | |||
|
45 | except OSError,e: | |||
|
46 | raise IOError(str(e)) | |||
|
47 | ||||
|
48 | self._owns_lock = True | |||
|
49 | ||||
|
50 | def _obtain_lock(self): | |||
|
51 | """The default implementation will raise if a lock cannot be obtained. | |||
|
52 | Subclasses may override this method to provide a different implementation""" | |||
|
53 | return self._obtain_lock_or_raise() | |||
|
54 | ||||
|
55 | def _release_lock(self): | |||
|
56 | """Release our lock if we have one""" | |||
|
57 | if not self._has_lock(): | |||
|
58 | return | |||
|
59 | ||||
|
60 | # if someone removed our file beforhand, lets just flag this issue | |||
|
61 | # instead of failing, to make it more usable. | |||
|
62 | lfp = self._lock_file_path() | |||
|
63 | try: | |||
|
64 | # on bloody windows, the file needs write permissions to be removable. | |||
|
65 | # Why ... | |||
|
66 | if os.name == 'nt': | |||
|
67 | os.chmod(lfp, 0777) | |||
|
68 | # END handle win32 | |||
|
69 | os.remove(lfp) | |||
|
70 | except OSError: | |||
|
71 | pass | |||
|
72 | self._owns_lock = False |
@@ -0,0 +1,102 b'' | |||||
|
1 | """Ordered dict implementation""" | |||
|
2 | from UserDict import DictMixin | |||
|
3 | ||||
|
4 | ||||
|
5 | class OrderedDict(dict, DictMixin): | |||
|
6 | ||||
|
7 | def __init__(self, *args, **kwds): | |||
|
8 | if len(args) > 1: | |||
|
9 | raise TypeError('expected at most 1 arguments, got %d' % len(args)) | |||
|
10 | try: | |||
|
11 | self.__end | |||
|
12 | except AttributeError: | |||
|
13 | self.clear() | |||
|
14 | self.update(*args, **kwds) | |||
|
15 | ||||
|
16 | def clear(self): | |||
|
17 | self.__end = end = [] | |||
|
18 | end += [None, end, end] # sentinel node for doubly linked list | |||
|
19 | self.__map = {} # key --> [key, prev, next] | |||
|
20 | dict.clear(self) | |||
|
21 | ||||
|
22 | def __setitem__(self, key, value): | |||
|
23 | if key not in self: | |||
|
24 | end = self.__end | |||
|
25 | curr = end[1] | |||
|
26 | curr[2] = end[1] = self.__map[key] = [key, curr, end] | |||
|
27 | dict.__setitem__(self, key, value) | |||
|
28 | ||||
|
29 | def __delitem__(self, key): | |||
|
30 | dict.__delitem__(self, key) | |||
|
31 | key, prev, next = self.__map.pop(key) | |||
|
32 | prev[2] = next | |||
|
33 | next[1] = prev | |||
|
34 | ||||
|
35 | def __iter__(self): | |||
|
36 | end = self.__end | |||
|
37 | curr = end[2] | |||
|
38 | while curr is not end: | |||
|
39 | yield curr[0] | |||
|
40 | curr = curr[2] | |||
|
41 | ||||
|
42 | def __reversed__(self): | |||
|
43 | end = self.__end | |||
|
44 | curr = end[1] | |||
|
45 | while curr is not end: | |||
|
46 | yield curr[0] | |||
|
47 | curr = curr[1] | |||
|
48 | ||||
|
49 | def popitem(self, last=True): | |||
|
50 | if not self: | |||
|
51 | raise KeyError('dictionary is empty') | |||
|
52 | if last: | |||
|
53 | key = reversed(self).next() | |||
|
54 | else: | |||
|
55 | key = iter(self).next() | |||
|
56 | value = self.pop(key) | |||
|
57 | return key, value | |||
|
58 | ||||
|
59 | def __reduce__(self): | |||
|
60 | items = [[k, self[k]] for k in self] | |||
|
61 | tmp = self.__map, self.__end | |||
|
62 | del self.__map, self.__end | |||
|
63 | inst_dict = vars(self).copy() | |||
|
64 | self.__map, self.__end = tmp | |||
|
65 | if inst_dict: | |||
|
66 | return (self.__class__, (items,), inst_dict) | |||
|
67 | return self.__class__, (items,) | |||
|
68 | ||||
|
69 | def keys(self): | |||
|
70 | return list(self) | |||
|
71 | ||||
|
72 | setdefault = DictMixin.setdefault | |||
|
73 | update = DictMixin.update | |||
|
74 | pop = DictMixin.pop | |||
|
75 | values = DictMixin.values | |||
|
76 | items = DictMixin.items | |||
|
77 | iterkeys = DictMixin.iterkeys | |||
|
78 | itervalues = DictMixin.itervalues | |||
|
79 | iteritems = DictMixin.iteritems | |||
|
80 | ||||
|
81 | def __repr__(self): | |||
|
82 | if not self: | |||
|
83 | return '%s()' % (self.__class__.__name__,) | |||
|
84 | return '%s(%r)' % (self.__class__.__name__, self.items()) | |||
|
85 | ||||
|
86 | def copy(self): | |||
|
87 | return self.__class__(self) | |||
|
88 | ||||
|
89 | @classmethod | |||
|
90 | def fromkeys(cls, iterable, value=None): | |||
|
91 | d = cls() | |||
|
92 | for key in iterable: | |||
|
93 | d[key] = value | |||
|
94 | return d | |||
|
95 | ||||
|
96 | def __eq__(self, other): | |||
|
97 | if isinstance(other, OrderedDict): | |||
|
98 | return len(self) == len(other) and self.items() == other.items() | |||
|
99 | return dict.__eq__(self, other) | |||
|
100 | ||||
|
101 | def __ne__(self, other): | |||
|
102 | return not self == other |
@@ -0,0 +1,36 b'' | |||||
|
1 | import os | |||
|
2 | ||||
|
3 | abspath = lambda * p: os.path.abspath(os.path.join(*p)) | |||
|
4 | ||||
|
5 | ||||
|
6 | def get_dirs_for_path(*paths): | |||
|
7 | """ | |||
|
8 | Returns list of directories, including intermediate. | |||
|
9 | """ | |||
|
10 | for path in paths: | |||
|
11 | head = path | |||
|
12 | while head: | |||
|
13 | head, tail = os.path.split(head) | |||
|
14 | if head: | |||
|
15 | yield head | |||
|
16 | else: | |||
|
17 | # We don't need to yield empty path | |||
|
18 | break | |||
|
19 | ||||
|
20 | ||||
|
21 | def get_dir_size(path): | |||
|
22 | root_path = path | |||
|
23 | size = 0 | |||
|
24 | for path, dirs, files in os.walk(root_path): | |||
|
25 | for f in files: | |||
|
26 | try: | |||
|
27 | size += os.path.getsize(os.path.join(path, f)) | |||
|
28 | except OSError: | |||
|
29 | pass | |||
|
30 | return size | |||
|
31 | ||||
|
32 | def get_user_home(): | |||
|
33 | """ | |||
|
34 | Returns home path of the user. | |||
|
35 | """ | |||
|
36 | return os.getenv('HOME', os.getenv('USERPROFILE')) |
@@ -0,0 +1,419 b'' | |||||
|
1 | # encoding: UTF-8 | |||
|
2 | import sys | |||
|
3 | import datetime | |||
|
4 | from string import Template | |||
|
5 | from rhodecode.lib.vcs.utils.filesize import filesizeformat | |||
|
6 | from rhodecode.lib.vcs.utils.helpers import get_total_seconds | |||
|
7 | ||||
|
8 | ||||
|
9 | class ProgressBarError(Exception): | |||
|
10 | pass | |||
|
11 | ||||
|
12 | class AlreadyFinishedError(ProgressBarError): | |||
|
13 | pass | |||
|
14 | ||||
|
15 | ||||
|
16 | class ProgressBar(object): | |||
|
17 | ||||
|
18 | default_elements = ['percentage', 'bar', 'steps'] | |||
|
19 | ||||
|
20 | def __init__(self, steps=100, stream=None, elements=None): | |||
|
21 | self.step = 0 | |||
|
22 | self.steps = steps | |||
|
23 | self.stream = stream or sys.stderr | |||
|
24 | self.bar_char = '=' | |||
|
25 | self.width = 50 | |||
|
26 | self.separator = ' | ' | |||
|
27 | self.elements = elements or self.default_elements | |||
|
28 | self.started = None | |||
|
29 | self.finished = False | |||
|
30 | self.steps_label = 'Step' | |||
|
31 | self.time_label = 'Time' | |||
|
32 | self.eta_label = 'ETA' | |||
|
33 | self.speed_label = 'Speed' | |||
|
34 | self.transfer_label = 'Transfer' | |||
|
35 | ||||
|
36 | def __str__(self): | |||
|
37 | return self.get_line() | |||
|
38 | ||||
|
39 | def __iter__(self): | |||
|
40 | start = self.step | |||
|
41 | end = self.steps + 1 | |||
|
42 | for x in xrange(start, end): | |||
|
43 | self.render(x) | |||
|
44 | yield x | |||
|
45 | ||||
|
46 | def get_separator(self): | |||
|
47 | return self.separator | |||
|
48 | ||||
|
49 | def get_bar_char(self): | |||
|
50 | return self.bar_char | |||
|
51 | ||||
|
52 | def get_bar(self): | |||
|
53 | char = self.get_bar_char() | |||
|
54 | perc = self.get_percentage() | |||
|
55 | length = int(self.width * perc / 100) | |||
|
56 | bar = char * length | |||
|
57 | bar = bar.ljust(self.width) | |||
|
58 | return bar | |||
|
59 | ||||
|
60 | def get_elements(self): | |||
|
61 | return self.elements | |||
|
62 | ||||
|
63 | def get_template(self): | |||
|
64 | separator = self.get_separator() | |||
|
65 | elements = self.get_elements() | |||
|
66 | return Template(separator.join((('$%s' % e) for e in elements))) | |||
|
67 | ||||
|
68 | def get_total_time(self, current_time=None): | |||
|
69 | if current_time is None: | |||
|
70 | current_time = datetime.datetime.now() | |||
|
71 | if not self.started: | |||
|
72 | return datetime.timedelta() | |||
|
73 | return current_time - self.started | |||
|
74 | ||||
|
75 | def get_rendered_total_time(self): | |||
|
76 | delta = self.get_total_time() | |||
|
77 | if not delta: | |||
|
78 | ttime = '-' | |||
|
79 | else: | |||
|
80 | ttime = str(delta) | |||
|
81 | return '%s %s' % (self.time_label, ttime) | |||
|
82 | ||||
|
83 | def get_eta(self, current_time=None): | |||
|
84 | if current_time is None: | |||
|
85 | current_time = datetime.datetime.now() | |||
|
86 | if self.step == 0: | |||
|
87 | return datetime.timedelta() | |||
|
88 | total_seconds = get_total_seconds(self.get_total_time()) | |||
|
89 | eta_seconds = total_seconds * self.steps / self.step - total_seconds | |||
|
90 | return datetime.timedelta(seconds=int(eta_seconds)) | |||
|
91 | ||||
|
92 | def get_rendered_eta(self): | |||
|
93 | eta = self.get_eta() | |||
|
94 | if not eta: | |||
|
95 | eta = '--:--:--' | |||
|
96 | else: | |||
|
97 | eta = str(eta).rjust(8) | |||
|
98 | return '%s: %s' % (self.eta_label, eta) | |||
|
99 | ||||
|
100 | def get_percentage(self): | |||
|
101 | return float(self.step) / self.steps * 100 | |||
|
102 | ||||
|
103 | def get_rendered_percentage(self): | |||
|
104 | perc = self.get_percentage() | |||
|
105 | return ('%s%%' % (int(perc))).rjust(5) | |||
|
106 | ||||
|
107 | def get_rendered_steps(self): | |||
|
108 | return '%s: %s/%s' % (self.steps_label, self.step, self.steps) | |||
|
109 | ||||
|
110 | def get_rendered_speed(self, step=None, total_seconds=None): | |||
|
111 | if step is None: | |||
|
112 | step = self.step | |||
|
113 | if total_seconds is None: | |||
|
114 | total_seconds = get_total_seconds(self.get_total_time()) | |||
|
115 | if step <= 0 or total_seconds <= 0: | |||
|
116 | speed = '-' | |||
|
117 | else: | |||
|
118 | speed = filesizeformat(float(step) / total_seconds) | |||
|
119 | return '%s: %s/s' % (self.speed_label, speed) | |||
|
120 | ||||
|
121 | def get_rendered_transfer(self, step=None, steps=None): | |||
|
122 | if step is None: | |||
|
123 | step = self.step | |||
|
124 | if steps is None: | |||
|
125 | steps = self.steps | |||
|
126 | ||||
|
127 | if steps <= 0: | |||
|
128 | return '%s: -' % self.transfer_label | |||
|
129 | total = filesizeformat(float(steps)) | |||
|
130 | if step <= 0: | |||
|
131 | transferred = '-' | |||
|
132 | else: | |||
|
133 | transferred = filesizeformat(float(step)) | |||
|
134 | return '%s: %s / %s' % (self.transfer_label, transferred, total) | |||
|
135 | ||||
|
136 | def get_context(self): | |||
|
137 | return { | |||
|
138 | 'percentage': self.get_rendered_percentage(), | |||
|
139 | 'bar': self.get_bar(), | |||
|
140 | 'steps': self.get_rendered_steps(), | |||
|
141 | 'time': self.get_rendered_total_time(), | |||
|
142 | 'eta': self.get_rendered_eta(), | |||
|
143 | 'speed': self.get_rendered_speed(), | |||
|
144 | 'transfer': self.get_rendered_transfer(), | |||
|
145 | } | |||
|
146 | ||||
|
147 | def get_line(self): | |||
|
148 | template = self.get_template() | |||
|
149 | context = self.get_context() | |||
|
150 | return template.safe_substitute(**context) | |||
|
151 | ||||
|
152 | def write(self, data): | |||
|
153 | self.stream.write(data) | |||
|
154 | ||||
|
155 | def render(self, step): | |||
|
156 | if not self.started: | |||
|
157 | self.started = datetime.datetime.now() | |||
|
158 | if self.finished: | |||
|
159 | raise AlreadyFinishedError | |||
|
160 | self.step = step | |||
|
161 | self.write('\r%s' % self) | |||
|
162 | if step == self.steps: | |||
|
163 | self.finished = True | |||
|
164 | if step == self.steps: | |||
|
165 | self.write('\n') | |||
|
166 | ||||
|
167 | ||||
|
168 | """ | |||
|
169 | termcolors.py | |||
|
170 | ||||
|
171 | Grabbed from Django (http://www.djangoproject.com) | |||
|
172 | """ | |||
|
173 | ||||
|
174 | color_names = ('black', 'red', 'green', 'yellow', 'blue', 'magenta', 'cyan', 'white') | |||
|
175 | foreground = dict([(color_names[x], '3%s' % x) for x in range(8)]) | |||
|
176 | background = dict([(color_names[x], '4%s' % x) for x in range(8)]) | |||
|
177 | ||||
|
178 | RESET = '0' | |||
|
179 | opt_dict = {'bold': '1', 'underscore': '4', 'blink': '5', 'reverse': '7', 'conceal': '8'} | |||
|
180 | ||||
|
181 | def colorize(text='', opts=(), **kwargs): | |||
|
182 | """ | |||
|
183 | Returns your text, enclosed in ANSI graphics codes. | |||
|
184 | ||||
|
185 | Depends on the keyword arguments 'fg' and 'bg', and the contents of | |||
|
186 | the opts tuple/list. | |||
|
187 | ||||
|
188 | Returns the RESET code if no parameters are given. | |||
|
189 | ||||
|
190 | Valid colors: | |||
|
191 | 'black', 'red', 'green', 'yellow', 'blue', 'magenta', 'cyan', 'white' | |||
|
192 | ||||
|
193 | Valid options: | |||
|
194 | 'bold' | |||
|
195 | 'underscore' | |||
|
196 | 'blink' | |||
|
197 | 'reverse' | |||
|
198 | 'conceal' | |||
|
199 | 'noreset' - string will not be auto-terminated with the RESET code | |||
|
200 | ||||
|
201 | Examples: | |||
|
202 | colorize('hello', fg='red', bg='blue', opts=('blink',)) | |||
|
203 | colorize() | |||
|
204 | colorize('goodbye', opts=('underscore',)) | |||
|
205 | print colorize('first line', fg='red', opts=('noreset',)) | |||
|
206 | print 'this should be red too' | |||
|
207 | print colorize('and so should this') | |||
|
208 | print 'this should not be red' | |||
|
209 | """ | |||
|
210 | code_list = [] | |||
|
211 | if text == '' and len(opts) == 1 and opts[0] == 'reset': | |||
|
212 | return '\x1b[%sm' % RESET | |||
|
213 | for k, v in kwargs.iteritems(): | |||
|
214 | if k == 'fg': | |||
|
215 | code_list.append(foreground[v]) | |||
|
216 | elif k == 'bg': | |||
|
217 | code_list.append(background[v]) | |||
|
218 | for o in opts: | |||
|
219 | if o in opt_dict: | |||
|
220 | code_list.append(opt_dict[o]) | |||
|
221 | if 'noreset' not in opts: | |||
|
222 | text = text + '\x1b[%sm' % RESET | |||
|
223 | return ('\x1b[%sm' % ';'.join(code_list)) + text | |||
|
224 | ||||
|
225 | def make_style(opts=(), **kwargs): | |||
|
226 | """ | |||
|
227 | Returns a function with default parameters for colorize() | |||
|
228 | ||||
|
229 | Example: | |||
|
230 | bold_red = make_style(opts=('bold',), fg='red') | |||
|
231 | print bold_red('hello') | |||
|
232 | KEYWORD = make_style(fg='yellow') | |||
|
233 | COMMENT = make_style(fg='blue', opts=('bold',)) | |||
|
234 | """ | |||
|
235 | return lambda text: colorize(text, opts, **kwargs) | |||
|
236 | ||||
|
237 | NOCOLOR_PALETTE = 'nocolor' | |||
|
238 | DARK_PALETTE = 'dark' | |||
|
239 | LIGHT_PALETTE = 'light' | |||
|
240 | ||||
|
241 | PALETTES = { | |||
|
242 | NOCOLOR_PALETTE: { | |||
|
243 | 'ERROR': {}, | |||
|
244 | 'NOTICE': {}, | |||
|
245 | 'SQL_FIELD': {}, | |||
|
246 | 'SQL_COLTYPE': {}, | |||
|
247 | 'SQL_KEYWORD': {}, | |||
|
248 | 'SQL_TABLE': {}, | |||
|
249 | 'HTTP_INFO': {}, | |||
|
250 | 'HTTP_SUCCESS': {}, | |||
|
251 | 'HTTP_REDIRECT': {}, | |||
|
252 | 'HTTP_NOT_MODIFIED': {}, | |||
|
253 | 'HTTP_BAD_REQUEST': {}, | |||
|
254 | 'HTTP_NOT_FOUND': {}, | |||
|
255 | 'HTTP_SERVER_ERROR': {}, | |||
|
256 | }, | |||
|
257 | DARK_PALETTE: { | |||
|
258 | 'ERROR': { 'fg': 'red', 'opts': ('bold',) }, | |||
|
259 | 'NOTICE': { 'fg': 'red' }, | |||
|
260 | 'SQL_FIELD': { 'fg': 'green', 'opts': ('bold',) }, | |||
|
261 | 'SQL_COLTYPE': { 'fg': 'green' }, | |||
|
262 | 'SQL_KEYWORD': { 'fg': 'yellow' }, | |||
|
263 | 'SQL_TABLE': { 'opts': ('bold',) }, | |||
|
264 | 'HTTP_INFO': { 'opts': ('bold',) }, | |||
|
265 | 'HTTP_SUCCESS': { }, | |||
|
266 | 'HTTP_REDIRECT': { 'fg': 'green' }, | |||
|
267 | 'HTTP_NOT_MODIFIED': { 'fg': 'cyan' }, | |||
|
268 | 'HTTP_BAD_REQUEST': { 'fg': 'red', 'opts': ('bold',) }, | |||
|
269 | 'HTTP_NOT_FOUND': { 'fg': 'yellow' }, | |||
|
270 | 'HTTP_SERVER_ERROR': { 'fg': 'magenta', 'opts': ('bold',) }, | |||
|
271 | }, | |||
|
272 | LIGHT_PALETTE: { | |||
|
273 | 'ERROR': { 'fg': 'red', 'opts': ('bold',) }, | |||
|
274 | 'NOTICE': { 'fg': 'red' }, | |||
|
275 | 'SQL_FIELD': { 'fg': 'green', 'opts': ('bold',) }, | |||
|
276 | 'SQL_COLTYPE': { 'fg': 'green' }, | |||
|
277 | 'SQL_KEYWORD': { 'fg': 'blue' }, | |||
|
278 | 'SQL_TABLE': { 'opts': ('bold',) }, | |||
|
279 | 'HTTP_INFO': { 'opts': ('bold',) }, | |||
|
280 | 'HTTP_SUCCESS': { }, | |||
|
281 | 'HTTP_REDIRECT': { 'fg': 'green', 'opts': ('bold',) }, | |||
|
282 | 'HTTP_NOT_MODIFIED': { 'fg': 'green' }, | |||
|
283 | 'HTTP_BAD_REQUEST': { 'fg': 'red', 'opts': ('bold',) }, | |||
|
284 | 'HTTP_NOT_FOUND': { 'fg': 'red' }, | |||
|
285 | 'HTTP_SERVER_ERROR': { 'fg': 'magenta', 'opts': ('bold',) }, | |||
|
286 | } | |||
|
287 | } | |||
|
288 | DEFAULT_PALETTE = DARK_PALETTE | |||
|
289 | ||||
|
290 | # ---------------------------- # | |||
|
291 | # --- End of termcolors.py --- # | |||
|
292 | # ---------------------------- # | |||
|
293 | ||||
|
294 | ||||
|
295 | class ColoredProgressBar(ProgressBar): | |||
|
296 | ||||
|
297 | BAR_COLORS = ( | |||
|
298 | (10, 'red'), | |||
|
299 | (30, 'magenta'), | |||
|
300 | (50, 'yellow'), | |||
|
301 | (99, 'green'), | |||
|
302 | (100, 'blue'), | |||
|
303 | ) | |||
|
304 | ||||
|
305 | def get_line(self): | |||
|
306 | line = super(ColoredProgressBar, self).get_line() | |||
|
307 | perc = self.get_percentage() | |||
|
308 | if perc > 100: | |||
|
309 | color = 'blue' | |||
|
310 | for max_perc, color in self.BAR_COLORS: | |||
|
311 | if perc <= max_perc: | |||
|
312 | break | |||
|
313 | return colorize(line, fg=color) | |||
|
314 | ||||
|
315 | ||||
|
316 | class AnimatedProgressBar(ProgressBar): | |||
|
317 | ||||
|
318 | def get_bar_char(self): | |||
|
319 | chars = '-/|\\' | |||
|
320 | if self.step >= self.steps: | |||
|
321 | return '=' | |||
|
322 | return chars[self.step % len(chars)] | |||
|
323 | ||||
|
324 | ||||
|
325 | class BarOnlyProgressBar(ProgressBar): | |||
|
326 | ||||
|
327 | default_elements = ['bar', 'steps'] | |||
|
328 | ||||
|
329 | def get_bar(self): | |||
|
330 | bar = super(BarOnlyProgressBar, self).get_bar() | |||
|
331 | perc = self.get_percentage() | |||
|
332 | perc_text = '%s%%' % int(perc) | |||
|
333 | text = (' %s%% ' % (perc_text)).center(self.width, '=') | |||
|
334 | L = text.find(' ') | |||
|
335 | R = text.rfind(' ') | |||
|
336 | bar = ' '.join((bar[:L], perc_text, bar[R:])) | |||
|
337 | return bar | |||
|
338 | ||||
|
339 | ||||
|
340 | class AnimatedColoredProgressBar(AnimatedProgressBar, | |||
|
341 | ColoredProgressBar): | |||
|
342 | pass | |||
|
343 | ||||
|
344 | ||||
|
345 | class BarOnlyColoredProgressBar(ColoredProgressBar, | |||
|
346 | BarOnlyProgressBar): | |||
|
347 | pass | |||
|
348 | ||||
|
349 | ||||
|
350 | ||||
|
351 | def main(): | |||
|
352 | import time | |||
|
353 | ||||
|
354 | print "Standard progress bar..." | |||
|
355 | bar = ProgressBar(30) | |||
|
356 | for x in xrange(1, 31): | |||
|
357 | bar.render(x) | |||
|
358 | time.sleep(0.02) | |||
|
359 | bar.stream.write('\n') | |||
|
360 | ||||
|
361 | ||||
|
362 | print "Empty bar..." | |||
|
363 | bar = ProgressBar(50) | |||
|
364 | bar.render(0) | |||
|
365 | ||||
|
366 | ||||
|
367 | ||||
|
368 | print "Colored bar..." | |||
|
369 | bar = ColoredProgressBar(20) | |||
|
370 | for x in bar: | |||
|
371 | time.sleep(0.01) | |||
|
372 | ||||
|
373 | ||||
|
374 | print "Animated char bar..." | |||
|
375 | bar = AnimatedProgressBar(20) | |||
|
376 | for x in bar: | |||
|
377 | time.sleep(0.01) | |||
|
378 | ||||
|
379 | ||||
|
380 | print "Animated + colored char bar..." | |||
|
381 | bar = AnimatedColoredProgressBar(20) | |||
|
382 | for x in bar: | |||
|
383 | time.sleep(0.01) | |||
|
384 | ||||
|
385 | ||||
|
386 | print "Bar only ..." | |||
|
387 | bar = BarOnlyProgressBar(20) | |||
|
388 | for x in bar: | |||
|
389 | time.sleep(0.01) | |||
|
390 | ||||
|
391 | ||||
|
392 | print "Colored, longer bar-only, eta, total time ..." | |||
|
393 | bar = BarOnlyColoredProgressBar(40) | |||
|
394 | bar.width = 60 | |||
|
395 | bar.elements += ['time', 'eta'] | |||
|
396 | for x in bar: | |||
|
397 | time.sleep(0.01) | |||
|
398 | ||||
|
399 | ||||
|
400 | ||||
|
401 | print "File transfer bar, breaks after 2 seconds ..." | |||
|
402 | total_bytes = 1024 * 1024 * 2 | |||
|
403 | bar = ProgressBar(total_bytes) | |||
|
404 | bar.width = 50 | |||
|
405 | bar.elements.remove('steps') | |||
|
406 | bar.elements += ['transfer', 'time', 'eta', 'speed'] | |||
|
407 | for x in xrange(0, bar.steps, 1024): | |||
|
408 | bar.render(x) | |||
|
409 | time.sleep(0.01) | |||
|
410 | now = datetime.datetime.now() | |||
|
411 | if now - bar.started >= datetime.timedelta(seconds=2): | |||
|
412 | break | |||
|
413 | ||||
|
414 | ||||
|
415 | ||||
|
416 | ||||
|
417 | ||||
|
418 | if __name__ == '__main__': | |||
|
419 | main() |
@@ -0,0 +1,200 b'' | |||||
|
1 | """ | |||
|
2 | termcolors.py | |||
|
3 | ||||
|
4 | Grabbed from Django (http://www.djangoproject.com) | |||
|
5 | """ | |||
|
6 | ||||
|
7 | color_names = ('black', 'red', 'green', 'yellow', 'blue', 'magenta', 'cyan', 'white') | |||
|
8 | foreground = dict([(color_names[x], '3%s' % x) for x in range(8)]) | |||
|
9 | background = dict([(color_names[x], '4%s' % x) for x in range(8)]) | |||
|
10 | ||||
|
11 | RESET = '0' | |||
|
12 | opt_dict = {'bold': '1', 'underscore': '4', 'blink': '5', 'reverse': '7', 'conceal': '8'} | |||
|
13 | ||||
|
14 | def colorize(text='', opts=(), **kwargs): | |||
|
15 | """ | |||
|
16 | Returns your text, enclosed in ANSI graphics codes. | |||
|
17 | ||||
|
18 | Depends on the keyword arguments 'fg' and 'bg', and the contents of | |||
|
19 | the opts tuple/list. | |||
|
20 | ||||
|
21 | Returns the RESET code if no parameters are given. | |||
|
22 | ||||
|
23 | Valid colors: | |||
|
24 | 'black', 'red', 'green', 'yellow', 'blue', 'magenta', 'cyan', 'white' | |||
|
25 | ||||
|
26 | Valid options: | |||
|
27 | 'bold' | |||
|
28 | 'underscore' | |||
|
29 | 'blink' | |||
|
30 | 'reverse' | |||
|
31 | 'conceal' | |||
|
32 | 'noreset' - string will not be auto-terminated with the RESET code | |||
|
33 | ||||
|
34 | Examples: | |||
|
35 | colorize('hello', fg='red', bg='blue', opts=('blink',)) | |||
|
36 | colorize() | |||
|
37 | colorize('goodbye', opts=('underscore',)) | |||
|
38 | print colorize('first line', fg='red', opts=('noreset',)) | |||
|
39 | print 'this should be red too' | |||
|
40 | print colorize('and so should this') | |||
|
41 | print 'this should not be red' | |||
|
42 | """ | |||
|
43 | code_list = [] | |||
|
44 | if text == '' and len(opts) == 1 and opts[0] == 'reset': | |||
|
45 | return '\x1b[%sm' % RESET | |||
|
46 | for k, v in kwargs.iteritems(): | |||
|
47 | if k == 'fg': | |||
|
48 | code_list.append(foreground[v]) | |||
|
49 | elif k == 'bg': | |||
|
50 | code_list.append(background[v]) | |||
|
51 | for o in opts: | |||
|
52 | if o in opt_dict: | |||
|
53 | code_list.append(opt_dict[o]) | |||
|
54 | if 'noreset' not in opts: | |||
|
55 | text = text + '\x1b[%sm' % RESET | |||
|
56 | return ('\x1b[%sm' % ';'.join(code_list)) + text | |||
|
57 | ||||
|
58 | def make_style(opts=(), **kwargs): | |||
|
59 | """ | |||
|
60 | Returns a function with default parameters for colorize() | |||
|
61 | ||||
|
62 | Example: | |||
|
63 | bold_red = make_style(opts=('bold',), fg='red') | |||
|
64 | print bold_red('hello') | |||
|
65 | KEYWORD = make_style(fg='yellow') | |||
|
66 | COMMENT = make_style(fg='blue', opts=('bold',)) | |||
|
67 | """ | |||
|
68 | return lambda text: colorize(text, opts, **kwargs) | |||
|
69 | ||||
|
70 | NOCOLOR_PALETTE = 'nocolor' | |||
|
71 | DARK_PALETTE = 'dark' | |||
|
72 | LIGHT_PALETTE = 'light' | |||
|
73 | ||||
|
74 | PALETTES = { | |||
|
75 | NOCOLOR_PALETTE: { | |||
|
76 | 'ERROR': {}, | |||
|
77 | 'NOTICE': {}, | |||
|
78 | 'SQL_FIELD': {}, | |||
|
79 | 'SQL_COLTYPE': {}, | |||
|
80 | 'SQL_KEYWORD': {}, | |||
|
81 | 'SQL_TABLE': {}, | |||
|
82 | 'HTTP_INFO': {}, | |||
|
83 | 'HTTP_SUCCESS': {}, | |||
|
84 | 'HTTP_REDIRECT': {}, | |||
|
85 | 'HTTP_NOT_MODIFIED': {}, | |||
|
86 | 'HTTP_BAD_REQUEST': {}, | |||
|
87 | 'HTTP_NOT_FOUND': {}, | |||
|
88 | 'HTTP_SERVER_ERROR': {}, | |||
|
89 | }, | |||
|
90 | DARK_PALETTE: { | |||
|
91 | 'ERROR': { 'fg': 'red', 'opts': ('bold',) }, | |||
|
92 | 'NOTICE': { 'fg': 'red' }, | |||
|
93 | 'SQL_FIELD': { 'fg': 'green', 'opts': ('bold',) }, | |||
|
94 | 'SQL_COLTYPE': { 'fg': 'green' }, | |||
|
95 | 'SQL_KEYWORD': { 'fg': 'yellow' }, | |||
|
96 | 'SQL_TABLE': { 'opts': ('bold',) }, | |||
|
97 | 'HTTP_INFO': { 'opts': ('bold',) }, | |||
|
98 | 'HTTP_SUCCESS': { }, | |||
|
99 | 'HTTP_REDIRECT': { 'fg': 'green' }, | |||
|
100 | 'HTTP_NOT_MODIFIED': { 'fg': 'cyan' }, | |||
|
101 | 'HTTP_BAD_REQUEST': { 'fg': 'red', 'opts': ('bold',) }, | |||
|
102 | 'HTTP_NOT_FOUND': { 'fg': 'yellow' }, | |||
|
103 | 'HTTP_SERVER_ERROR': { 'fg': 'magenta', 'opts': ('bold',) }, | |||
|
104 | }, | |||
|
105 | LIGHT_PALETTE: { | |||
|
106 | 'ERROR': { 'fg': 'red', 'opts': ('bold',) }, | |||
|
107 | 'NOTICE': { 'fg': 'red' }, | |||
|
108 | 'SQL_FIELD': { 'fg': 'green', 'opts': ('bold',) }, | |||
|
109 | 'SQL_COLTYPE': { 'fg': 'green' }, | |||
|
110 | 'SQL_KEYWORD': { 'fg': 'blue' }, | |||
|
111 | 'SQL_TABLE': { 'opts': ('bold',) }, | |||
|
112 | 'HTTP_INFO': { 'opts': ('bold',) }, | |||
|
113 | 'HTTP_SUCCESS': { }, | |||
|
114 | 'HTTP_REDIRECT': { 'fg': 'green', 'opts': ('bold',) }, | |||
|
115 | 'HTTP_NOT_MODIFIED': { 'fg': 'green' }, | |||
|
116 | 'HTTP_BAD_REQUEST': { 'fg': 'red', 'opts': ('bold',) }, | |||
|
117 | 'HTTP_NOT_FOUND': { 'fg': 'red' }, | |||
|
118 | 'HTTP_SERVER_ERROR': { 'fg': 'magenta', 'opts': ('bold',) }, | |||
|
119 | } | |||
|
120 | } | |||
|
121 | DEFAULT_PALETTE = DARK_PALETTE | |||
|
122 | ||||
|
123 | def parse_color_setting(config_string): | |||
|
124 | """Parse a DJANGO_COLORS environment variable to produce the system palette | |||
|
125 | ||||
|
126 | The general form of a pallete definition is: | |||
|
127 | ||||
|
128 | "palette;role=fg;role=fg/bg;role=fg,option,option;role=fg/bg,option,option" | |||
|
129 | ||||
|
130 | where: | |||
|
131 | palette is a named palette; one of 'light', 'dark', or 'nocolor'. | |||
|
132 | role is a named style used by Django | |||
|
133 | fg is a background color. | |||
|
134 | bg is a background color. | |||
|
135 | option is a display options. | |||
|
136 | ||||
|
137 | Specifying a named palette is the same as manually specifying the individual | |||
|
138 | definitions for each role. Any individual definitions following the pallete | |||
|
139 | definition will augment the base palette definition. | |||
|
140 | ||||
|
141 | Valid roles: | |||
|
142 | 'error', 'notice', 'sql_field', 'sql_coltype', 'sql_keyword', 'sql_table', | |||
|
143 | 'http_info', 'http_success', 'http_redirect', 'http_bad_request', | |||
|
144 | 'http_not_found', 'http_server_error' | |||
|
145 | ||||
|
146 | Valid colors: | |||
|
147 | 'black', 'red', 'green', 'yellow', 'blue', 'magenta', 'cyan', 'white' | |||
|
148 | ||||
|
149 | Valid options: | |||
|
150 | 'bold', 'underscore', 'blink', 'reverse', 'conceal' | |||
|
151 | ||||
|
152 | """ | |||
|
153 | if not config_string: | |||
|
154 | return PALETTES[DEFAULT_PALETTE] | |||
|
155 | ||||
|
156 | # Split the color configuration into parts | |||
|
157 | parts = config_string.lower().split(';') | |||
|
158 | palette = PALETTES[NOCOLOR_PALETTE].copy() | |||
|
159 | for part in parts: | |||
|
160 | if part in PALETTES: | |||
|
161 | # A default palette has been specified | |||
|
162 | palette.update(PALETTES[part]) | |||
|
163 | elif '=' in part: | |||
|
164 | # Process a palette defining string | |||
|
165 | definition = {} | |||
|
166 | ||||
|
167 | # Break the definition into the role, | |||
|
168 | # plus the list of specific instructions. | |||
|
169 | # The role must be in upper case | |||
|
170 | role, instructions = part.split('=') | |||
|
171 | role = role.upper() | |||
|
172 | ||||
|
173 | styles = instructions.split(',') | |||
|
174 | styles.reverse() | |||
|
175 | ||||
|
176 | # The first instruction can contain a slash | |||
|
177 | # to break apart fg/bg. | |||
|
178 | colors = styles.pop().split('/') | |||
|
179 | colors.reverse() | |||
|
180 | fg = colors.pop() | |||
|
181 | if fg in color_names: | |||
|
182 | definition['fg'] = fg | |||
|
183 | if colors and colors[-1] in color_names: | |||
|
184 | definition['bg'] = colors[-1] | |||
|
185 | ||||
|
186 | # All remaining instructions are options | |||
|
187 | opts = tuple(s for s in styles if s in opt_dict.keys()) | |||
|
188 | if opts: | |||
|
189 | definition['opts'] = opts | |||
|
190 | ||||
|
191 | # The nocolor palette has all available roles. | |||
|
192 | # Use that palette as the basis for determining | |||
|
193 | # if the role is valid. | |||
|
194 | if role in PALETTES[NOCOLOR_PALETTE] and definition: | |||
|
195 | palette[role] = definition | |||
|
196 | ||||
|
197 | # If there are no colors specified, return the empty palette. | |||
|
198 | if palette == PALETTES[NOCOLOR_PALETTE]: | |||
|
199 | return None | |||
|
200 | return palette |
@@ -17,3 +17,4 b' syntax: regexp' | |||||
17 | ^test\.db$ |
|
17 | ^test\.db$ | |
18 | ^RhodeCode\.egg-info$ |
|
18 | ^RhodeCode\.egg-info$ | |
19 | ^rc\.ini$ |
|
19 | ^rc\.ini$ | |
|
20 | ^fabfile.py |
@@ -10,9 +10,8 b' celery>=2.2.5,<2.3' | |||||
10 | babel |
|
10 | babel | |
11 | python-dateutil>=1.5.0,<2.0.0 |
|
11 | python-dateutil>=1.5.0,<2.0.0 | |
12 | dulwich>=0.8.0,<0.9.0 |
|
12 | dulwich>=0.8.0,<0.9.0 | |
13 | vcs>=0.2.3.dev |
|
|||
14 | webob==1.0.8 |
|
13 | webob==1.0.8 | |
15 |
markdown==2. |
|
14 | markdown==2.1.1 | |
16 | docutils==0.8.1 |
|
15 | docutils==0.8.1 | |
17 | py-bcrypt |
|
16 | py-bcrypt | |
18 | mercurial>=2.1,<2.2 No newline at end of file |
|
17 | mercurial>=2.1,<2.2 |
@@ -49,9 +49,8 b' requirements = [' | |||||
49 | "babel", |
|
49 | "babel", | |
50 | "python-dateutil>=1.5.0,<2.0.0", |
|
50 | "python-dateutil>=1.5.0,<2.0.0", | |
51 | "dulwich>=0.8.0,<0.9.0", |
|
51 | "dulwich>=0.8.0,<0.9.0", | |
52 | "vcs>=0.2.3.dev", |
|
|||
53 | "webob==1.0.8", |
|
52 | "webob==1.0.8", | |
54 |
"markdown==2. |
|
53 | "markdown==2.1.1", | |
55 | "docutils==0.8.1", |
|
54 | "docutils==0.8.1", | |
56 | ] |
|
55 | ] | |
57 |
|
56 |
@@ -37,7 +37,7 b' from rhodecode.lib.base import BaseRepoC' | |||||
37 | from rhodecode.lib.helpers import RepoPage |
|
37 | from rhodecode.lib.helpers import RepoPage | |
38 | from rhodecode.lib.compat import json |
|
38 | from rhodecode.lib.compat import json | |
39 |
|
39 | |||
40 | from vcs.exceptions import RepositoryError, ChangesetDoesNotExistError |
|
40 | from rhodecode.lib.vcs.exceptions import RepositoryError, ChangesetDoesNotExistError | |
41 | from rhodecode.model.db import Repository |
|
41 | from rhodecode.model.db import Repository | |
42 |
|
42 | |||
43 | log = logging.getLogger(__name__) |
|
43 | log = logging.getLogger(__name__) |
@@ -33,9 +33,9 b' from pylons.i18n.translation import _' | |||||
33 | from pylons.controllers.util import redirect |
|
33 | from pylons.controllers.util import redirect | |
34 | from pylons.decorators import jsonify |
|
34 | from pylons.decorators import jsonify | |
35 |
|
35 | |||
36 | from vcs.exceptions import RepositoryError, ChangesetError, \ |
|
36 | from rhodecode.lib.vcs.exceptions import RepositoryError, ChangesetError, \ | |
37 | ChangesetDoesNotExistError |
|
37 | ChangesetDoesNotExistError | |
38 | from vcs.nodes import FileNode |
|
38 | from rhodecode.lib.vcs.nodes import FileNode | |
39 |
|
39 | |||
40 | import rhodecode.lib.helpers as h |
|
40 | import rhodecode.lib.helpers as h | |
41 | from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator |
|
41 | from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator |
@@ -32,11 +32,11 b' from pylons.i18n.translation import _' | |||||
32 | from pylons.controllers.util import redirect |
|
32 | from pylons.controllers.util import redirect | |
33 | from pylons.decorators import jsonify |
|
33 | from pylons.decorators import jsonify | |
34 |
|
34 | |||
35 | from vcs.conf import settings |
|
35 | from rhodecode.lib.vcs.conf import settings | |
36 | from vcs.exceptions import RepositoryError, ChangesetDoesNotExistError, \ |
|
36 | from rhodecode.lib.vcs.exceptions import RepositoryError, ChangesetDoesNotExistError, \ | |
37 | EmptyRepositoryError, ImproperArchiveTypeError, VCSError, \ |
|
37 | EmptyRepositoryError, ImproperArchiveTypeError, VCSError, \ | |
38 | NodeAlreadyExistsError |
|
38 | NodeAlreadyExistsError | |
39 | from vcs.nodes import FileNode |
|
39 | from rhodecode.lib.vcs.nodes import FileNode | |
40 |
|
40 | |||
41 | from rhodecode.lib.compat import OrderedDict |
|
41 | from rhodecode.lib.compat import OrderedDict | |
42 | from rhodecode.lib import convert_line_endings, detect_mode, safe_str |
|
42 | from rhodecode.lib import convert_line_endings, detect_mode, safe_str |
@@ -31,7 +31,7 b' from datetime import timedelta, date' | |||||
31 | from itertools import product |
|
31 | from itertools import product | |
32 | from urlparse import urlparse |
|
32 | from urlparse import urlparse | |
33 |
|
33 | |||
34 | from vcs.exceptions import ChangesetError, EmptyRepositoryError, \ |
|
34 | from rhodecode.lib.vcs.exceptions import ChangesetError, EmptyRepositoryError, \ | |
35 | NodeDoesNotExistError |
|
35 | NodeDoesNotExistError | |
36 |
|
36 | |||
37 | from pylons import tmpl_context as c, request, url, config |
|
37 | from pylons import tmpl_context as c, request, url, config |
@@ -25,7 +25,7 b'' | |||||
25 |
|
25 | |||
26 | import os |
|
26 | import os | |
27 | import re |
|
27 | import re | |
28 | from vcs.utils.lazy import LazyProperty |
|
28 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |
29 |
|
29 | |||
30 |
|
30 | |||
31 | def __get_lem(): |
|
31 | def __get_lem(): | |
@@ -138,7 +138,6 b' def convert_line_endings(line, mode):' | |||||
138 | line = replace(line, '\r\n', '\r') |
|
138 | line = replace(line, '\r\n', '\r') | |
139 | line = replace(line, '\n', '\r') |
|
139 | line = replace(line, '\n', '\r') | |
140 | elif mode == 2: |
|
140 | elif mode == 2: | |
141 | import re |
|
|||
142 | line = re.sub("\r(?!\n)|(?<!\r)\n", "\r\n", line) |
|
141 | line = re.sub("\r(?!\n)|(?<!\r)\n", "\r\n", line) | |
143 | return line |
|
142 | return line | |
144 |
|
143 | |||
@@ -341,7 +340,8 b' def age(curdate):' | |||||
341 | pos = 1 |
|
340 | pos = 1 | |
342 | for scale in agescales: |
|
341 | for scale in agescales: | |
343 | if scale[1] <= age_seconds: |
|
342 | if scale[1] <= age_seconds: | |
344 |
if pos == 6: |
|
343 | if pos == 6: | |
|
344 | pos = 5 | |||
345 | return '%s %s' % (time_ago_in_words(curdate, |
|
345 | return '%s %s' % (time_ago_in_words(curdate, | |
346 | agescales[pos][0]), _('ago')) |
|
346 | agescales[pos][0]), _('ago')) | |
347 | pos += 1 |
|
347 | pos += 1 | |
@@ -404,8 +404,8 b' def get_changeset_safe(repo, rev):' | |||||
404 | :param repo: |
|
404 | :param repo: | |
405 | :param rev: |
|
405 | :param rev: | |
406 | """ |
|
406 | """ | |
407 | from vcs.backends.base import BaseRepository |
|
407 | from rhodecode.lib.vcs.backends.base import BaseRepository | |
408 | from vcs.exceptions import RepositoryError |
|
408 | from rhodecode.lib.vcs.exceptions import RepositoryError | |
409 | if not isinstance(repo, BaseRepository): |
|
409 | if not isinstance(repo, BaseRepository): | |
410 | raise Exception('You must pass an Repository ' |
|
410 | raise Exception('You must pass an Repository ' | |
411 | 'object as first argument got %s', type(repo)) |
|
411 | 'object as first argument got %s', type(repo)) | |
@@ -427,8 +427,8 b' def get_current_revision(quiet=False):' | |||||
427 | """ |
|
427 | """ | |
428 |
|
428 | |||
429 | try: |
|
429 | try: | |
430 | from vcs import get_repo |
|
430 | from rhodecode.lib.vcs import get_repo | |
431 | from vcs.utils.helpers import get_scm |
|
431 | from rhodecode.lib.vcs.utils.helpers import get_scm | |
432 | repopath = os.path.join(os.path.dirname(__file__), '..', '..') |
|
432 | repopath = os.path.join(os.path.dirname(__file__), '..', '..') | |
433 | scm = get_scm(repopath)[0] |
|
433 | scm = get_scm(repopath)[0] | |
434 | repo = get_repo(path=repopath, alias=scm) |
|
434 | repo = get_repo(path=repopath, alias=scm) |
@@ -11,8 +11,8 b'' | |||||
11 | :license: GPLv3, see COPYING for more details. |
|
11 | :license: GPLv3, see COPYING for more details. | |
12 | """ |
|
12 | """ | |
13 |
|
13 | |||
14 | from vcs.exceptions import VCSError |
|
14 | from rhodecode.lib.vcs.exceptions import VCSError | |
15 | from vcs.nodes import FileNode |
|
15 | from rhodecode.lib.vcs.nodes import FileNode | |
16 | from pygments.formatters import HtmlFormatter |
|
16 | from pygments.formatters import HtmlFormatter | |
17 | from pygments import highlight |
|
17 | from pygments import highlight | |
18 |
|
18 |
@@ -34,7 +34,7 b' from pylons import config' | |||||
34 | from hashlib import md5 |
|
34 | from hashlib import md5 | |
35 | from decorator import decorator |
|
35 | from decorator import decorator | |
36 |
|
36 | |||
37 | from vcs.utils.lazy import LazyProperty |
|
37 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |
38 | from rhodecode import CELERY_ON |
|
38 | from rhodecode import CELERY_ON | |
39 | from rhodecode.lib import str2bool, safe_str |
|
39 | from rhodecode.lib import str2bool, safe_str | |
40 | from rhodecode.lib.pidlock import DaemonLock, LockHeld |
|
40 | from rhodecode.lib.pidlock import DaemonLock, LockHeld |
@@ -37,7 +37,7 b' from string import lower' | |||||
37 | from pylons import config, url |
|
37 | from pylons import config, url | |
38 | from pylons.i18n.translation import _ |
|
38 | from pylons.i18n.translation import _ | |
39 |
|
39 | |||
40 | from vcs import get_backend |
|
40 | from rhodecode.lib.vcs import get_backend | |
41 |
|
41 | |||
42 | from rhodecode import CELERY_ON |
|
42 | from rhodecode import CELERY_ON | |
43 | from rhodecode.lib import LANGUAGES_EXTENSIONS_MAP, safe_str |
|
43 | from rhodecode.lib import LANGUAGES_EXTENSIONS_MAP, safe_str |
@@ -34,10 +34,10 b' from sqlalchemy.ext.hybrid import hybrid' | |||||
34 | from sqlalchemy.orm import relationship, joinedload, class_mapper, validates |
|
34 | from sqlalchemy.orm import relationship, joinedload, class_mapper, validates | |
35 | from beaker.cache import cache_region, region_invalidate |
|
35 | from beaker.cache import cache_region, region_invalidate | |
36 |
|
36 | |||
37 | from vcs import get_backend |
|
37 | from rhodecode.lib.vcs import get_backend | |
38 | from vcs.utils.helpers import get_scm |
|
38 | from rhodecode.lib.vcs.utils.helpers import get_scm | |
39 | from vcs.exceptions import VCSError |
|
39 | from rhodecode.lib.vcs.exceptions import VCSError | |
40 | from vcs.utils.lazy import LazyProperty |
|
40 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |
41 |
|
41 | |||
42 | from rhodecode.lib import str2bool, safe_str, get_changeset_safe, \ |
|
42 | from rhodecode.lib import str2bool, safe_str, get_changeset_safe, \ | |
43 | generate_api_key, safe_unicode |
|
43 | generate_api_key, safe_unicode |
@@ -32,8 +32,8 b' from itertools import tee, imap' | |||||
32 |
|
32 | |||
33 | from pylons.i18n.translation import _ |
|
33 | from pylons.i18n.translation import _ | |
34 |
|
34 | |||
35 | from vcs.exceptions import VCSError |
|
35 | from rhodecode.lib.vcs.exceptions import VCSError | |
36 | from vcs.nodes import FileNode |
|
36 | from rhodecode.lib.vcs.nodes import FileNode | |
37 |
|
37 | |||
38 | from rhodecode.lib.utils import EmptyChangeset |
|
38 | from rhodecode.lib.utils import EmptyChangeset | |
39 |
|
39 |
@@ -318,7 +318,7 b' flash = _Flash()' | |||||
318 | #============================================================================== |
|
318 | #============================================================================== | |
319 | # SCM FILTERS available via h. |
|
319 | # SCM FILTERS available via h. | |
320 | #============================================================================== |
|
320 | #============================================================================== | |
321 | from vcs.utils import author_name, author_email |
|
321 | from rhodecode.lib.vcs.utils import author_name, author_email | |
322 | from rhodecode.lib import credentials_filter, age as _age |
|
322 | from rhodecode.lib import credentials_filter, age as _age | |
323 | from rhodecode.model.db import User |
|
323 | from rhodecode.model.db import User | |
324 |
|
324 |
@@ -43,7 +43,7 b' from rhodecode.model.scm import ScmModel' | |||||
43 | from rhodecode.lib import safe_unicode |
|
43 | from rhodecode.lib import safe_unicode | |
44 | from rhodecode.lib.indexers import INDEX_EXTENSIONS, SCHEMA, IDX_NAME |
|
44 | from rhodecode.lib.indexers import INDEX_EXTENSIONS, SCHEMA, IDX_NAME | |
45 |
|
45 | |||
46 | from vcs.exceptions import ChangesetError, RepositoryError, \ |
|
46 | from rhodecode.lib.vcs.exceptions import ChangesetError, RepositoryError, \ | |
47 | NodeDoesNotExistError |
|
47 | NodeDoesNotExistError | |
48 |
|
48 | |||
49 | from whoosh.index import create_in, open_dir |
|
49 | from whoosh.index import create_in, open_dir |
@@ -30,6 +30,7 b' import traceback' | |||||
30 |
|
30 | |||
31 | from dulwich import server as dulserver |
|
31 | from dulwich import server as dulserver | |
32 |
|
32 | |||
|
33 | ||||
33 | class SimpleGitUploadPackHandler(dulserver.UploadPackHandler): |
|
34 | class SimpleGitUploadPackHandler(dulserver.UploadPackHandler): | |
34 |
|
35 | |||
35 | def handle(self): |
|
36 | def handle(self): |
@@ -154,7 +154,6 b' class SimpleHg(BaseVCSController):' | |||||
154 | baseui = make_ui('db') |
|
154 | baseui = make_ui('db') | |
155 | self.__inject_extras(repo_path, baseui, extras) |
|
155 | self.__inject_extras(repo_path, baseui, extras) | |
156 |
|
156 | |||
157 |
|
||||
158 | # quick check if that dir exists... |
|
157 | # quick check if that dir exists... | |
159 | if is_valid_repo(repo_name, self.basepath) is False: |
|
158 | if is_valid_repo(repo_name, self.basepath) is False: | |
160 | return HTTPNotFound()(environ, start_response) |
|
159 | return HTTPNotFound()(environ, start_response) | |
@@ -221,7 +220,6 b' class SimpleHg(BaseVCSController):' | |||||
221 | else: |
|
220 | else: | |
222 | return 'pull' |
|
221 | return 'pull' | |
223 |
|
222 | |||
224 |
|
||||
225 | def __inject_extras(self, repo_path, baseui, extras={}): |
|
223 | def __inject_extras(self, repo_path, baseui, extras={}): | |
226 | """ |
|
224 | """ | |
227 | Injects some extra params into baseui instance |
|
225 | Injects some extra params into baseui instance |
@@ -40,11 +40,11 b' from mercurial import ui, config' | |||||
40 |
|
40 | |||
41 | from webhelpers.text import collapse, remove_formatting, strip_tags |
|
41 | from webhelpers.text import collapse, remove_formatting, strip_tags | |
42 |
|
42 | |||
43 | from vcs import get_backend |
|
43 | from rhodecode.lib.vcs import get_backend | |
44 | from vcs.backends.base import BaseChangeset |
|
44 | from rhodecode.lib.vcs.backends.base import BaseChangeset | |
45 | from vcs.utils.lazy import LazyProperty |
|
45 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |
46 | from vcs.utils.helpers import get_scm |
|
46 | from rhodecode.lib.vcs.utils.helpers import get_scm | |
47 | from vcs.exceptions import VCSError |
|
47 | from rhodecode.lib.vcs.exceptions import VCSError | |
48 |
|
48 | |||
49 | from rhodecode.lib.caching_query import FromCache |
|
49 | from rhodecode.lib.caching_query import FromCache | |
50 |
|
50 |
@@ -34,10 +34,10 b' from sqlalchemy.ext.hybrid import hybrid' | |||||
34 | from sqlalchemy.orm import relationship, joinedload, class_mapper, validates |
|
34 | from sqlalchemy.orm import relationship, joinedload, class_mapper, validates | |
35 | from beaker.cache import cache_region, region_invalidate |
|
35 | from beaker.cache import cache_region, region_invalidate | |
36 |
|
36 | |||
37 | from vcs import get_backend |
|
37 | from rhodecode.lib.vcs import get_backend | |
38 | from vcs.utils.helpers import get_scm |
|
38 | from rhodecode.lib.vcs.utils.helpers import get_scm | |
39 | from vcs.exceptions import VCSError |
|
39 | from rhodecode.lib.vcs.exceptions import VCSError | |
40 | from vcs.utils.lazy import LazyProperty |
|
40 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |
41 |
|
41 | |||
42 | from rhodecode.lib import str2bool, safe_str, get_changeset_safe, safe_unicode |
|
42 | from rhodecode.lib import str2bool, safe_str, get_changeset_safe, safe_unicode | |
43 | from rhodecode.lib.compat import json |
|
43 | from rhodecode.lib.compat import json |
@@ -28,7 +28,7 b' import logging' | |||||
28 | import traceback |
|
28 | import traceback | |
29 | from datetime import datetime |
|
29 | from datetime import datetime | |
30 |
|
30 | |||
31 | from vcs.backends import get_backend |
|
31 | from rhodecode.lib.vcs.backends import get_backend | |
32 |
|
32 | |||
33 | from rhodecode.lib import LazyProperty |
|
33 | from rhodecode.lib import LazyProperty | |
34 | from rhodecode.lib import safe_str, safe_unicode |
|
34 | from rhodecode.lib import safe_str, safe_unicode |
@@ -28,10 +28,10 b' import traceback' | |||||
28 | import logging |
|
28 | import logging | |
29 | import cStringIO |
|
29 | import cStringIO | |
30 |
|
30 | |||
31 | from vcs import get_backend |
|
31 | from rhodecode.lib.vcs import get_backend | |
32 | from vcs.exceptions import RepositoryError |
|
32 | from rhodecode.lib.vcs.exceptions import RepositoryError | |
33 | from vcs.utils.lazy import LazyProperty |
|
33 | from rhodecode.lib.vcs.utils.lazy import LazyProperty | |
34 | from vcs.nodes import FileNode |
|
34 | from rhodecode.lib.vcs.nodes import FileNode | |
35 |
|
35 | |||
36 | from rhodecode import BACKENDS |
|
36 | from rhodecode import BACKENDS | |
37 | from rhodecode.lib import helpers as h |
|
37 | from rhodecode.lib import helpers as h | |
@@ -359,9 +359,9 b' class ScmModel(BaseModel):' | |||||
359 | content, f_path): |
|
359 | content, f_path): | |
360 |
|
360 | |||
361 | if repo.alias == 'hg': |
|
361 | if repo.alias == 'hg': | |
362 | from vcs.backends.hg import MercurialInMemoryChangeset as IMC |
|
362 | from rhodecode.lib.vcs.backends.hg import MercurialInMemoryChangeset as IMC | |
363 | elif repo.alias == 'git': |
|
363 | elif repo.alias == 'git': | |
364 | from vcs.backends.git import GitInMemoryChangeset as IMC |
|
364 | from rhodecode.lib.vcs.backends.git import GitInMemoryChangeset as IMC | |
365 |
|
365 | |||
366 | # decoding here will force that we have proper encoded values |
|
366 | # decoding here will force that we have proper encoded values | |
367 | # in any other case this will throw exceptions and deny commit |
|
367 | # in any other case this will throw exceptions and deny commit | |
@@ -385,9 +385,9 b' class ScmModel(BaseModel):' | |||||
385 | def create_node(self, repo, repo_name, cs, user, author, message, content, |
|
385 | def create_node(self, repo, repo_name, cs, user, author, message, content, | |
386 | f_path): |
|
386 | f_path): | |
387 | if repo.alias == 'hg': |
|
387 | if repo.alias == 'hg': | |
388 | from vcs.backends.hg import MercurialInMemoryChangeset as IMC |
|
388 | from rhodecode.lib.vcs.backends.hg import MercurialInMemoryChangeset as IMC | |
389 | elif repo.alias == 'git': |
|
389 | elif repo.alias == 'git': | |
390 | from vcs.backends.git import GitInMemoryChangeset as IMC |
|
390 | from rhodecode.lib.vcs.backends.git import GitInMemoryChangeset as IMC | |
391 | # decoding here will force that we have proper encoded values |
|
391 | # decoding here will force that we have proper encoded values | |
392 | # in any other case this will throw exceptions and deny commit |
|
392 | # in any other case this will throw exceptions and deny commit | |
393 |
|
393 |
@@ -1,18 +1,16 b'' | |||||
1 | # -*- coding: utf-8 -*- |
|
1 | # -*- coding: utf-8 -*- | |
2 |
|
2 | |||
3 | import os |
|
3 | import os | |
4 | import vcs |
|
4 | from rhodecode.lib import vcs | |
5 |
|
5 | |||
6 | from rhodecode.model.db import Repository |
|
6 | from rhodecode.model.db import Repository | |
7 | from rhodecode.tests import * |
|
7 | from rhodecode.tests import * | |
8 |
|
8 | |||
9 | class TestAdminReposController(TestController): |
|
9 | class TestAdminReposController(TestController): | |
10 |
|
10 | |||
11 |
|
||||
12 | def __make_repo(self): |
|
11 | def __make_repo(self): | |
13 | pass |
|
12 | pass | |
14 |
|
13 | |||
15 |
|
||||
16 | def test_index(self): |
|
14 | def test_index(self): | |
17 | self.log_user() |
|
15 | self.log_user() | |
18 | response = self.app.get(url('repos')) |
|
16 | response = self.app.get(url('repos')) |
@@ -31,11 +31,10 b'' | |||||
31 | import cookielib |
|
31 | import cookielib | |
32 | import urllib |
|
32 | import urllib | |
33 | import urllib2 |
|
33 | import urllib2 | |
34 | import vcs |
|
|||
35 | import time |
|
34 | import time | |
36 |
|
35 | |||
37 | from os.path import join as jn |
|
36 | from os.path import join as jn | |
38 |
|
37 | from rhodecode.lib import vcs | ||
39 |
|
38 | |||
40 | BASE_URI = 'http://127.0.0.1:5000/%s' |
|
39 | BASE_URI = 'http://127.0.0.1:5000/%s' | |
41 | PROJECT = 'CPython' |
|
40 | PROJECT = 'CPython' | |
@@ -52,7 +51,6 b' o.addheaders = [' | |||||
52 | urllib2.install_opener(o) |
|
51 | urllib2.install_opener(o) | |
53 |
|
52 | |||
54 |
|
53 | |||
55 |
|
||||
56 | def test_changelog_walk(pages=100): |
|
54 | def test_changelog_walk(pages=100): | |
57 | total_time = 0 |
|
55 | total_time = 0 | |
58 | for i in range(1, pages): |
|
56 | for i in range(1, pages): | |
@@ -67,7 +65,6 b' def test_changelog_walk(pages=100):' | |||||
67 | total_time += e |
|
65 | total_time += e | |
68 | print 'visited %s size:%s req:%s ms' % (full_uri, size, e) |
|
66 | print 'visited %s size:%s req:%s ms' % (full_uri, size, e) | |
69 |
|
67 | |||
70 |
|
||||
71 | print 'total_time', total_time |
|
68 | print 'total_time', total_time | |
72 | print 'average on req', total_time / float(pages) |
|
69 | print 'average on req', total_time / float(pages) | |
73 |
|
70 | |||
@@ -103,7 +100,7 b' def test_files_walk(limit=100):' | |||||
103 | repo = vcs.get_repo(jn(PROJECT_PATH, PROJECT)) |
|
100 | repo = vcs.get_repo(jn(PROJECT_PATH, PROJECT)) | |
104 |
|
101 | |||
105 | from rhodecode.lib.compat import OrderedSet |
|
102 | from rhodecode.lib.compat import OrderedSet | |
106 | from vcs.exceptions import RepositoryError |
|
103 | from rhodecode.lib.vcs.exceptions import RepositoryError | |
107 |
|
104 | |||
108 | paths_ = OrderedSet(['']) |
|
105 | paths_ = OrderedSet(['']) | |
109 | try: |
|
106 | try: | |
@@ -141,7 +138,6 b' def test_files_walk(limit=100):' | |||||
141 | print 'average on req', total_time / float(cnt) |
|
138 | print 'average on req', total_time / float(cnt) | |
142 |
|
139 | |||
143 |
|
140 | |||
144 |
|
||||
145 | test_changelog_walk(40) |
|
141 | test_changelog_walk(40) | |
146 | time.sleep(2) |
|
142 | time.sleep(2) | |
147 | test_changeset_walk(limit=100) |
|
143 | test_changeset_walk(limit=100) |
@@ -7,10 +7,7 b' from rhodecode import requirements' | |||||
7 | if __py_version__ < (2, 5): |
|
7 | if __py_version__ < (2, 5): | |
8 | raise Exception('RhodeCode requires python 2.5 or later') |
|
8 | raise Exception('RhodeCode requires python 2.5 or later') | |
9 |
|
9 | |||
10 |
|
||||
11 | dependency_links = [ |
|
10 | dependency_links = [ | |
12 | "https://secure.rhodecode.org/vcs/archive/default.zip#egg=vcs-0.2.3.dev", |
|
|||
13 | "https://bitbucket.org/marcinkuzminski/vcs/get/default.zip#egg=vcs-0.2.3.dev", |
|
|||
14 | ] |
|
11 | ] | |
15 |
|
12 | |||
16 | classifiers = [ |
|
13 | classifiers = [ |
General Comments 0
You need to be logged in to leave comments.
Login now