Implemented comments

1 files changed, 94 insertions, 1 deletions

diff --git a/libbe/bugdir.py b/libbe/bugdir.py
index d318690..25a0b66 100644
--- a/libbe/bugdir.py
+++ b/libbe/bugdir.py
@@ -194,6 +194,39 @@ class Bug(object):
         path = self.get_path("values")
         map_save(rcs_by_name(self.rcs_name), path, map)
 
+    def _get_rcs(self):
+        return rcs_by_name(self.rcs_name)
+
+    rcs = property(_get_rcs)
+
+    def new_comment(self):
+        if not os.path.exists(self.get_path("comments")):
+            self.rcs.mkdir(self.get_path("comments"))
+        comm = Comment(None, self)
+        comm.uuid = names.uuid()
+        return comm
+
+    def get_comment(self, uuid):
+        return Comment(uuid, self)
+
+    def iter_comment_ids(self):
+        try:
+            for uuid in os.listdir(self.get_path("comments")):
+                if (uuid.startswith('.')):
+                    continue
+                yield uuid
+        except IOError, e:
+            if e.errno != errno.ENOENT:
+                raise
+            return
+
+    def list_comments(self):
+        comments = [Comment(id, self) for id in self.iter_comment_ids()]
+        comments.sort(cmp_date)
+        return comments
+
+def cmp_date(comm1, comm2):
+    return cmp(comm1.date, comm2.date)
 
 def new_bug(dir):
     bug = dir.new_bug()
@@ -202,8 +235,68 @@ def new_bug(dir):
     bug.status = "open"
     bug.time = time.time()
     return bug
- 
 
+def new_comment(bug, body=None):
+    comm = bug.new_comment()
+    comm.From = names.creator()
+    comm.date = time.time()
+    comm.body = body
+    return comm
+
+def add_headers(obj, map, names):
+    map_names = {}
+    for name in names:
+        map_names[name] = pyname_to_header(name)
+    add_attrs(obj, map, names, map_names)
+
+def add_attrs(obj, map, names, map_names=None):
+    if map_names is None:
+        map_names = {}
+        for name in names:
+            map_names[name] = name 
+        
+    for name in names:
+        value = obj.__getattribute__(name)
+        if value is not None:
+            map[map_names[name]] = value
+
+
+class Comment(object):
+    def __init__(self, uuid, bug):
+        object.__init__(self)
+        self.uuid = uuid 
+        self.bug = bug
+        if self.uuid is not None and self.bug is not None:
+            mapfile = map_load(self.get_path("values"))
+            self.date = utility.str_to_time(mapfile["Date"])
+            self.From = mapfile["From"]
+            self.in_reply_to = mapfile.get("In-reply-to")
+            self.body = file(self.get_path("body")).read()
+        else:
+            self.date = None
+            self.From = None
+            self.in_reply_to = None
+            self.body = None
+
+    def save(self):
+        map_file = {"Date": utility.time_to_str(self.date)}
+        add_headers(self, map_file, ("From", "in_reply_to"))
+        if not os.path.exists(self.get_path(None)):
+            self.bug.rcs.mkdir(self.get_path(None))
+        map_save(self.bug.rcs, self.get_path("values"), map_file)
+        self.bug.rcs.set_file_contents(self.get_path("body"), self.body)
+            
+
+    def get_path(self, name):
+        my_dir = os.path.join(self.bug.get_path("comments"), self.uuid)
+        if name is None:
+            return my_dir
+        return os.path.join(my_dir, name)
+        
+def pyname_to_header(name):
+    return name.capitalize().replace('_', '-')
+    
+    
 def map_save(rcs, path, map):
     """Save the map as a mapfile to the specified path"""
     if not os.path.exists(path):