Merge remote-tracking branch 'notes/master'

notes/__init__.py

@@ -0,0 +1,60 @@
+from glob import glob
+from typing import List, Sequence
+from pathlib import Path
+
+from kython.ktyping import PathIsh
+
+from porg import Org # type: ignore
+
+
+# TODO enc stuff?
+def get_org_paths():
+    return [
+        '***REMOVED***',
+        '***REMOVED***',
+    ]
+
+def _get_org_files_in(path, archived: bool=False) -> List[Path]:
+    ppp = Path(path)
+    assert ppp.exists()
+    # TODO try/catch??
+    if ppp.is_file():
+        return [ppp]
+
+    path = str(path) # TODO FIXME use pathlib
+    res = []
+    res.extend(glob(path + '/**/*.org', recursive=True))
+    if archived:
+        res.extend(glob(path + '/**/*.org_archive', recursive=True))
+    return list(map(Path, res))
+
+
+def get_org_files(archived: bool = False) -> List[Path]:
+    res = []
+    for p in get_org_paths():
+        res.extend(_get_org_files_in(p, archived=archived))
+    return res
+
+
+# TODO move to porg?
+class PorgAll:
+    def __init__(self, paths: Sequence[PathIsh]) -> None:
+        self.paths = [Path(p) for p in paths]
+
+    def xpath_all(self, query: str):
+        return self.query_all(lambda x: x.xpath_all(query))
+
+    def get_all(self):
+        return self.xpath_all('//*')
+
+    def query_all(self, query):
+        res: List[Org] = []
+        for p in self.paths:
+            for of in _get_org_files_in(p):
+                org = Org.from_file(str(of))
+                res.extend(query(org))
+        return res
+
+
+def get_notes():
+    return PorgAll(get_org_paths())