Minor cleanup in document loader.

This commit is contained in:
Sergio Martínez Portela 2021-04-03 01:47:33 +02:00
parent b126e178b1
commit f3e1573677

View File

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