| 77 | | |
|---|
| 78 | | |
|---|
| 79 | | class BranchChangeset(object): |
|---|
| 80 | | """Represents a Subversion revision with additionnal properties""" |
|---|
| 81 | | |
|---|
| 82 | | def __init__(self, repos, changeset): |
|---|
| 83 | | # Repository |
|---|
| 84 | | self.repos = repos |
|---|
| 85 | | # Trac changeset |
|---|
| 86 | | self.changeset = changeset |
|---|
| 87 | | ## revision number |
|---|
| 88 | | self.rev = self.changeset.rev |
|---|
| 89 | | # branch name |
|---|
| 90 | | self.branchname = None |
|---|
| 91 | | # clone information (if any) |
|---|
| 92 | | self.clone = None |
|---|
| 93 | | # last changeset of a branch |
|---|
| 94 | | self.last = False |
|---|
| 95 | | # SVN properties |
|---|
| 96 | | self.properties = None |
|---|
| 97 | | |
|---|
| 98 | | def __cmp__(self, other): |
|---|
| 99 | | """Compares to another changeset, based on the revision number""" |
|---|
| 100 | | return cmp(self.rev, other.rev) |
|---|
| 101 | | |
|---|
| 102 | | def build(self, bcre): |
|---|
| 103 | | """Loads a changeset from a SVN repository""" |
|---|
| 104 | | """ |
|---|
| 105 | | cre should define two named groups 'branch' and 'path' |
|---|
| 106 | | """ |
|---|
| 107 | | try: |
|---|
| 108 | | if not self._find_simple_branch(bcre): |
|---|
| 109 | | self._find_plain_branch(bcre) |
|---|
| 110 | | except AssertionError, e: |
|---|
| 111 | | raise AssertionError, "%s rev: %d" % (e, self.rev or 0) |
|---|
| 112 | | |
|---|
| 113 | | def _load_properties(self): |
|---|
| 114 | | self.properties = self.repos.get_revision_properties(self.rev) |
|---|
| 115 | | |
|---|
| 116 | | def prop(self, prop): |
|---|
| 117 | | if not isinstance(self.properties, dict): |
|---|
| 118 | | self._load_properties() |
|---|
| 119 | | return self.properties.has_key(prop) and self.properties[prop] or '' |
|---|
| 120 | | |
|---|
| 121 | | def props(self, majtype=None): |
|---|
| 122 | | if not isinstance(self.properties, dict): |
|---|
| 123 | | self._load_properties() |
|---|
| 124 | | if majtype is None: |
|---|
| 125 | | return self.properties |
|---|
| 126 | | else: |
|---|
| 127 | | props = {} |
|---|
| 128 | | for (k,v) in self.properties.items(): |
|---|
| 129 | | items = k.split(':') |
|---|
| 130 | | if len(items) and (items[0] == majtype): |
|---|
| 131 | | props[items[1]] = v |
|---|
| 132 | | return props |
|---|
| 133 | | |
|---|
| 134 | | def _find_simple_branch(self, bcre): |
|---|
| 135 | | change_gen = self.changeset.get_changes() |
|---|
| 136 | | item = change_gen.next() |
|---|
| 137 | | try: |
|---|
| 138 | | change_gen.next() |
|---|
| 139 | | except StopIteration: |
|---|
| 140 | | pass |
|---|
| 141 | | else: |
|---|
| 142 | | return False |
|---|
| 143 | | (path, kind, change, base_path, base_rev) = item |
|---|
| 144 | | if kind is not Node.DIRECTORY: |
|---|
| 145 | | return False |
|---|
| 146 | | if change is Changeset.COPY: |
|---|
| 147 | | path_mo = bcre.match(path) |
|---|
| 148 | | src_mo = bcre.match(base_path) |
|---|
| 149 | | elif change is Changeset.DELETE: |
|---|
| 150 | | self.last = True |
|---|
| 151 | | path_mo = bcre.match(base_path) |
|---|
| 152 | | src_mo = False |
|---|
| 153 | | else: |
|---|
| 154 | | return False |
|---|
| 155 | | if not path_mo: |
|---|
| 156 | | return False |
|---|
| 157 | | if path_mo.group('path'): |
|---|
| 158 | | return False |
|---|
| 159 | | self.branchname = path_mo.group('branch').lower() |
|---|
| 160 | | if src_mo: |
|---|
| 161 | | self.clone = (int(base_rev), src_mo.group('branch').lower()) |
|---|
| 162 | | return True |
|---|
| 163 | | |
|---|
| 164 | | def _find_plain_branch(self, bcre): |
|---|
| 165 | | branch = None |
|---|
| 166 | | for item in self.changeset.get_changes(): |
|---|
| 167 | | (path, kind, change, base_path, base_rev) = item |
|---|
| 168 | | mo = bcre.match(path) |
|---|
| 169 | | if mo: |
|---|
| 170 | | try: |
|---|
| 171 | | br = mo.group('branch').lower() |
|---|
| 172 | | except IndexError: |
|---|
| 173 | | raise AssertionError, "Invalid RE: missing 'branch' group" |
|---|
| 174 | | else: |
|---|
| 175 | | return False |
|---|
| 176 | | if not branch: |
|---|
| 177 | | branch = br |
|---|
| 178 | | elif branch != br: |
|---|
| 179 | | raise AssertionError, 'Incoherent path [%s] != [%s]' \ |
|---|
| 180 | | % (br, branch) |
|---|
| 181 | | self.branchname = branch |
|---|
| 182 | | return True |
|---|
| 183 | | |
|---|
| 184 | | class Branch(object): |
|---|
| 185 | | """Represents a branch in Subversion, tracking the associated |
|---|
| 186 | | changesets""" |
|---|
| 187 | | |
|---|
| 188 | | def __init__(self, name): |
|---|
| 189 | | # Name (path) |
|---|
| 190 | | self.name = name |
|---|
| 191 | | # Source |
|---|
| 192 | | self._source = None |
|---|
| 193 | | # Changesets instances tied to the branch |
|---|
| 194 | | self._changesets = [] |
|---|
| 195 | | |
|---|
| 196 | | def add_changeset(self, changeset): |
|---|
| 197 | | """Adds a new changeset to the branch""" |
|---|
| 198 | | self._changesets.append(changeset) |
|---|
| 199 | | self._changesets.sort() |
|---|
| 200 | | |
|---|
| 201 | | def __len__(self): |
|---|
| 202 | | """Counts the number of tracked changesets""" |
|---|
| 203 | | return len(self._changesets) |
|---|
| 204 | | |
|---|
| 205 | | def changesets(self, revrange=None): |
|---|
| 206 | | """Returns the tracked changeset as a sequence""" |
|---|
| 207 | | if revrange is None: |
|---|
| 208 | | return self._changesets |
|---|
| 209 | | else: |
|---|
| 210 | | return filter(lambda c,mn=revrange[0],mx=revrange[1]: \ |
|---|
| 211 | | mn <= c.rev <= mx, self._changesets) |
|---|
| 212 | | |
|---|
| 213 | | def revision_range(self): |
|---|
| 214 | | """Returns a tuple representing the extent of tracked revisions |
|---|
| 215 | | (first, last)""" |
|---|
| 216 | | if not self._changesets: |
|---|
| 217 | | return (0, 0) |
|---|
| 218 | | return (self._changesets[0].revision, self._changesets[-1].revision) |
|---|
| 219 | | |
|---|
| 220 | | def authors(self): |
|---|
| 221 | | """Returns a list of authors that have committed to the branch""" |
|---|
| 222 | | authors = [] |
|---|
| 223 | | for chg in self._changesets: |
|---|
| 224 | | author = chg.changeset.author |
|---|
| 225 | | if author not in authors: |
|---|
| 226 | | authors.append(author) |
|---|
| 227 | | return authors |
|---|
| 228 | | |
|---|
| 229 | | def source(self): |
|---|
| 230 | | """Search for the origin of the branch""" |
|---|
| 231 | | return self._source |
|---|
| 232 | | |
|---|
| 233 | | def youngest(self): |
|---|
| 234 | | if len(self._changesets) > 0: |
|---|
| 235 | | return self._changesets[-1] |
|---|
| 236 | | else: |
|---|
| 237 | | return None |
|---|
| 238 | | |
|---|
| 239 | | def oldest(self): |
|---|
| 240 | | if len(self._changesets) > 0: |
|---|
| 241 | | return self._changesets[0] |
|---|
| 242 | | else: |
|---|
| 243 | | return None |
|---|
| 244 | | |
|---|
| 245 | | def is_active(self, range): |
|---|
| 246 | | y = self.youngest() |
|---|
| 247 | | if not y: |
|---|
| 248 | | return False |
|---|
| 249 | | if not (range[0] <= y.rev <= range[1]): |
|---|
| 250 | | return False |
|---|
| 251 | | if y.last: |
|---|
| 252 | | return False |
|---|
| 253 | | return True |
|---|
| 254 | | |
|---|
| 255 | | def build(self, repos): |
|---|
| 256 | | if len(self._changesets) > 0: |
|---|
| 257 | | clone = self._changesets[0].clone |
|---|
| 258 | | if clone: |
|---|
| 259 | | node = repos.find_node(clone[1], clone[0]) |
|---|
| 260 | | self._source = (int(node[1]), node[0]) |
|---|
| 261 | | |
|---|
| 262 | | class Repository(object): |
|---|
| 263 | | """Represents a Subversion repositories as a set of branches and a set |
|---|
| 264 | | of changesets""" |
|---|
| 265 | | |
|---|
| 266 | | def __init__(self, env, authname): |
|---|
| 267 | | # Environment |
|---|
| 268 | | self.env = env |
|---|
| 269 | | # Logger |
|---|
| 270 | | self.log = env.log |
|---|
| 271 | | # Trac version control |
|---|
| 272 | | self._crepos = self.env.get_repository(authname) |
|---|
| 273 | | # Dictionnary of changesets |
|---|
| 274 | | self._changesets = {} |
|---|
| 275 | | # Dictionnary of branches |
|---|
| 276 | | self._branches = {} |
|---|
| 277 | | |
|---|
| 278 | | def _build_branches(self): |
|---|
| 279 | | """Constructs the branch dictionnary from the changeset dictionnary""" |
|---|
| 280 | | for chgset in self._changesets.values(): |
|---|
| 281 | | br = chgset.branchname |
|---|
| 282 | | if not self._branches.has_key(br): |
|---|
| 283 | | self._branches[br] = Branch(br) |
|---|
| 284 | | self._branches[br].add_changeset(chgset) |
|---|
| 285 | | map(lambda b: b.build(self), self._branches.values()) |
|---|
| 286 | | |
|---|
| 287 | | def changeset(self, revision): |
|---|
| 288 | | """Returns a tracked changeset from the revision number""" |
|---|
| 289 | | if self._changesets.has_key(revision): |
|---|
| 290 | | return self._changesets[revision] |
|---|
| 291 | | else: |
|---|
| 292 | | return None |
|---|
| 293 | | |
|---|
| 294 | | def branch(self, branchname): |
|---|
| 295 | | """Returns a tracked branch from its name (path)""" |
|---|
| 296 | | if not self._branches.has_key(branchname): |
|---|
| 297 | | return None |
|---|
| 298 | | else: |
|---|
| 299 | | return self._branches[branchname] |
|---|
| 300 | | |
|---|
| 301 | | def changesets(self): |
|---|
| 302 | | """Returns the dictionnary of changesets (keys are rev. numbers)""" |
|---|
| 303 | | return self._changesets |
|---|
| 304 | | |
|---|
| 305 | | def branches(self): |
|---|
| 306 | | """Returns the dictionnary of branches (keys are branch names)""" |
|---|
| 307 | | return self._branches |
|---|
| 308 | | |
|---|
| 309 | | def revision_range(self): |
|---|
| 310 | | """Returns a tuple representing the extent of tracked revisions |
|---|
| 311 | | (first, last)""" |
|---|
| 312 | | return (self._revrange) |
|---|
| 313 | | |
|---|
| 314 | | def authors(self): |
|---|
| 315 | | """Returns a list of authors that have committed to the repository""" |
|---|
| 316 | | authors = [] |
|---|
| 317 | | for chg in self._changesets.values(): |
|---|
| 318 | | author = chg.changeset.author |
|---|
| 319 | | if author not in authors: |
|---|
| 320 | | authors.append(author) |
|---|
| 321 | | return authors |
|---|
| 322 | | |
|---|
| 323 | | def get_revision_properties(self, revision): |
|---|
| 324 | | """Returns the revision properties |
|---|
| 325 | | Ideally, this should be implemented by Trac core...""" |
|---|
| 326 | | return fs.svn_fs_revision_proplist(self._crepos.repos.fs_ptr, revision, |
|---|
| 327 | | self._crepos.repos.pool()) |
|---|
| 328 | | |
|---|
| 329 | | def find_node(self, path, rev): |
|---|
| 330 | | node = self._crepos.get_node(path, rev) |
|---|
| 331 | | return (node.get_name(), node.rev) |
|---|
| 332 | | |
|---|
| 333 | | def build(self, bcre, revrange=None, timerange=None): |
|---|
| 334 | | """Builds an internal representation of the repository, which |
|---|
| 335 | | is used to generate a graphical view of it""" |
|---|
| 336 | | start = 0 |
|---|
| 337 | | stop = int(time.time()) |
|---|
| 338 | | if timerange: |
|---|
| 339 | | if timerange[0]: |
|---|
| 340 | | start = timerange[0] |
|---|
| 341 | | if timerange[1]: |
|---|
| 342 | | stop = timerange[1] |
|---|
| 343 | | vcchangesets = self._crepos.get_changesets(start, stop) |
|---|
| 344 | | if revrange: |
|---|
| 345 | | revmin = self._crepos.get_oldest_rev() |
|---|
| 346 | | revmax = self._crepos.get_youngest_rev() |
|---|
| 347 | | if revrange[0]: |
|---|
| 348 | | revmin = revrange[0] |
|---|
| 349 | | if revrange[1]: |
|---|
| 350 | | revmax = revrange[1] |
|---|
| 351 | | vcsort = [(c.rev, c) for c in vcchangesets \ |
|---|
| 352 | | if revmin <= c.rev <= revmax] |
|---|
| 353 | | else: |
|---|
| 354 | | vcsort = [(c.rev, c) for c in vcchangesets] |
|---|
| 355 | | if len(vcsort) < 1: |
|---|
| 356 | | raise EmptyRangeError |
|---|
| 357 | | vcsort.sort() |
|---|
| 358 | | self._revrange = (vcsort[0][1].rev,vcsort[-1][1].rev) |
|---|
| 359 | | vcsort.reverse() |
|---|
| 360 | | for (rev, vc) in vcsort: |
|---|
| 361 | | chgset = BranchChangeset(self, vc) |
|---|
| 362 | | chgset.build(bcre) |
|---|
| 363 | | self._changesets[rev] = chgset |
|---|
| 364 | | self._build_branches() |
|---|
| 365 | | |
|---|
| 366 | | def __str__(self): |
|---|
| 367 | | """Returns a string representation of the repository""" |
|---|
| 368 | | msg = "Revision counter: %d\n" % len(self._changesets) |
|---|
| 369 | | for br in self._branches.keys(): |
|---|
| 370 | | msg += "Branch %s, %d revisions\n" % \ |
|---|
| 371 | | (br, len(self._branches[br])) |
|---|
| 372 | | return msg |
|---|
| 373 | | |
|---|
