107 lines
2.9 KiB
Python
107 lines
2.9 KiB
Python
|
import logging
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from datetime import datetime
|
||
|
|
from typing import List
|
||
|
|
|
||
|
|
import org_rw
|
||
|
|
from org_rw import OrgTime
|
||
|
|
|
||
|
|
|
||
|
|
def is_today(ot: OrgTime):
|
||
|
|
now = datetime.now()
|
||
|
|
return (
|
||
|
|
(ot.time.year == now.year)
|
||
|
|
and (ot.time.month == now.month)
|
||
|
|
and (ot.time.day == now.day)
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
class Agenda:
|
||
|
|
def __init__(self, /,
|
||
|
|
with_hour: List[org_rw.Headline],
|
||
|
|
no_hour: List[org_rw.Headline],
|
||
|
|
):
|
||
|
|
self.with_hour = with_hour
|
||
|
|
self.no_hour = no_hour
|
||
|
|
|
||
|
|
def print(self):
|
||
|
|
for item in self.with_hour:
|
||
|
|
print(item.scheduled.time, item.state, item.title)
|
||
|
|
|
||
|
|
if len(self.with_hour) > 0:
|
||
|
|
print("--------")
|
||
|
|
|
||
|
|
for item in self.no_hour:
|
||
|
|
print(item.scheduled.time, item.state, item.title)
|
||
|
|
|
||
|
|
|
||
|
|
class DocumentManager:
|
||
|
|
def __init__(self, basepath):
|
||
|
|
self.basepath = basepath
|
||
|
|
|
||
|
|
def load(self):
|
||
|
|
top = os.path.abspath(self.basepath)
|
||
|
|
|
||
|
|
docs = []
|
||
|
|
|
||
|
|
for root, dirs, files in os.walk(top):
|
||
|
|
for name in files:
|
||
|
|
if ".org" not in name:
|
||
|
|
continue
|
||
|
|
|
||
|
|
path = os.path.join(root, name)
|
||
|
|
|
||
|
|
try:
|
||
|
|
doc = org_rw.load(open(path), extra_cautious=True)
|
||
|
|
docs.append(doc)
|
||
|
|
except Exception as err:
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
traceback.print_exc()
|
||
|
|
print(f"== On {path}")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
logging.info("Loaded {} files".format(len(docs)))
|
||
|
|
|
||
|
|
self.docs = docs
|
||
|
|
|
||
|
|
def get_agenda(self) -> Agenda:
|
||
|
|
|
||
|
|
headline_count = 0
|
||
|
|
items_in_agenda = []
|
||
|
|
now = datetime.now()
|
||
|
|
|
||
|
|
for doc in self.docs:
|
||
|
|
for hl in doc.getAllHeadlines():
|
||
|
|
headline_count += 1
|
||
|
|
|
||
|
|
if hl.scheduled and isinstance(hl.scheduled, OrgTime):
|
||
|
|
if is_today(hl.scheduled):
|
||
|
|
items_in_agenda.append(hl)
|
||
|
|
elif (hl.scheduled.time.to_datetime() < now) and hl.is_todo:
|
||
|
|
items_in_agenda.append(hl)
|
||
|
|
|
||
|
|
logging.info("Read {} items".format(headline_count))
|
||
|
|
logging.info("{} items in agenda today".format(len(items_in_agenda)))
|
||
|
|
|
||
|
|
items_with_hour = [
|
||
|
|
item
|
||
|
|
for item in items_in_agenda
|
||
|
|
if item.scheduled and is_today(item.scheduled) and item.scheduled.time.hour
|
||
|
|
]
|
||
|
|
other_items = [
|
||
|
|
item
|
||
|
|
for item in items_in_agenda
|
||
|
|
if not (
|
||
|
|
item.scheduled and is_today(item.scheduled) and item.scheduled.time.hour
|
||
|
|
)
|
||
|
|
]
|
||
|
|
|
||
|
|
logging.info("{} items today for a specific hour".format(len(items_with_hour)))
|
||
|
|
|
||
|
|
return Agenda(
|
||
|
|
with_hour=sorted(items_with_hour, key=lambda x: x.scheduled.time),
|
||
|
|
no_hour=other_items
|
||
|
|
)
|