Mercurial > hg > TagInTheMiddle
comparison taginthemiddle/model.py @ 0:1c5cbbde4299
initial commit of middleware tagging; doesnt yet work
| author | Jeff Hammel <jhammel@mozilla.com> |
|---|---|
| date | Tue, 04 May 2010 08:37:15 -0700 |
| parents | |
| children | 837cfc05d4d9 |
comparison
equal
deleted
inserted
replaced
| -1:000000000000 | 0:1c5cbbde4299 |
|---|---|
| 1 import os | |
| 2 | |
| 3 class Tags(object): | |
| 4 """file (e.g. .ini) based tagging scheme""" | |
| 5 | |
| 6 def __init__(self, path): | |
| 7 """ | |
| 8 - path: path to file | |
| 9 """ | |
| 10 self.path = path | |
| 11 if not os.path.exists(path): | |
| 12 f = file(path, 'w') | |
| 13 f.close() | |
| 14 self.tags = {} | |
| 15 self.urls = {} | |
| 16 self.read(path) | |
| 17 | |
| 18 | |
| 19 def read(self): | |
| 20 for line in file(path).readlines(): | |
| 21 line = line.strip() | |
| 22 if not line: | |
| 23 continue | |
| 24 tag, urls = line.split('=', 1) # asserts '=' in line | |
| 25 tag = tag.strip() | |
| 26 urls = urls.strip() | |
| 27 self.tags[tag] = set([urls]) | |
| 28 for url in urls: | |
| 29 self.urls.setdefault(url, set()).add(tag) | |
| 30 | |
| 31 def write(self): | |
| 32 f = file(self.path, 'w') | |
| 33 for tag in sorted(self.tags.keys()): | |
| 34 print >> f, '%s = %s' % (tag, ' '.join(self.tags[tag])) | |
| 35 f.close() | |
| 36 | |
| 37 def add(self, tag, url): | |
| 38 if url in self.tags[tag]: | |
| 39 return | |
| 40 self.tags[tag].add(url) | |
| 41 self.urls[url].add(tag) | |
| 42 self.write() | |
| 43 | |
| 44 def remove(self, tag, url): | |
| 45 if url not in self.tags[tag]: | |
| 46 return | |
| 47 self.tags[tag].remove(url) | |
| 48 self.urls[url].remove(tag) | |
| 49 self.write() |
