kenkeiras/org-editor
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Minor cleanup in document loader.

Browse Source
This commit is contained in:
Sergio Martínez Portela 2021-04-03 01:47:33 +02:00
parent b126e178b1
commit f3e1573677
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
doc_manager.py
View File

@ -5,8 +5,9 @@ from datetime import datetime
from typing import List from typing import List
import org_rw import org_rw
from org_rw import OrgTime from org_rw import OrgTime, OrgDoc
EXTENSIONS = ( ".org", ".org.txt" )
def is_today(ot: OrgTime): def is_today(ot: OrgTime):
now = datetime.now() now = datetime.now()
@ -37,17 +38,19 @@ class Agenda:
class DocumentManager: class DocumentManager:
def __init__(self, basepath): docs: list[OrgDoc]
self.basepath = basepath
def __init__(self, base_path: os.PathLike):
self.base_path = base_path
def load(self): def load(self):
top = os.path.abspath(self.basepath) top = os.path.abspath(self.base_path)
docs = [] docs = []
for root, dirs, files in os.walk(top): for root, dirs, files in os.walk(top):
for name in files: for name in files:
if ".org" not in name: if all(map(lambda ext: not name.endswith(ext), EXTENSIONS)):
continue continue
path = os.path.join(root, name) path = os.path.join(root, name)