Minor cleanup in document loader.
This commit is contained in:
parent
b126e178b1
commit
f3e1573677
@ -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)
|
||||
|
Loading…
Reference in New Issue
Block a user