added parsing for all labels in the project

Author: dogeek

RTE/models/labels.py

@@ -0,0 +1,5 @@
+class Label:
+    def __init__(self, name, linenum, fname):
+        self.name = name
+        self.linenum = linenum
+        self.filename = fname

RTE/models/project.py

@@ -1,5 +1,8 @@
 import git
 import os
+import re
+import glob
+from .labels import Label
 
 class Project:
     def __init__(self, path):
@@ -10,8 +13,28 @@ def __init__(self, path):
                 self.notes = f.read()
         else:
             self.notes = ""
+
+        self.labels = self.get_all_labels()
         pass
 
     def save_notes(self):
         with open(os.path.join(self.path, ".rte-notes"), "w") as f:
             f.write(self.notes)
+
+    def get_all_labels(self):
+        label_re = re.compile(r'(label) (\S+) ?:')
+        labels = []
+        for filename in self.get_all_files("rpy"):
+            path = os.path.join(self.path, filename)
+            with open(path, "r") as f:
+                for i, line in enumerate(f):
+                    match = label_re.match(line)
+                    if match:
+                        labels.append(Label(match.group(2), i, path))
+        return labels
+
+    def get_all_files(self, ext="rpy"):
+        rv = []
+        for filename in glob.iglob(os.path.join(self.path, f'**/*.{ext}'), recursive=True):
+            rv.append(filename)
+        return rv

config/config.json

@@ -1,6 +1,6 @@
 {
-    "wm_width": 1597,
-    "wm_height": 752,
+    "wm_width": 1270,
+    "wm_height": 575,
     "theme_name": "monokai",
     "insert_spaces_instead_of_tabs": true,
     "tabs_length": 4,
@@ -9,6 +9,6 @@
     "start_maximized": false,
     "locale": "en-US",
     "path_to_git": "",
-    "side_notebook_width": 200,
-    "scrollbar_width": 20
+    "side_notebook_width": 200
+}    "scrollbar_width": 20
 }