calendarium.py

#!/usr/bin/env python3
from sys import argv, exit, modules
import datetime, os, json, yaml, subprocess, shutil, glob

try: calendar = argv[1]
except:
    print("the proper calling format is ./calendarium.py <calendar-path>.\n"
        +"see the repo's readme.txt for more details")
    exit(1)

if not os.path.isdir(calendar):
    print(f"{calendar} does not exist or is not a directory")
    exit(1)

now = datetime.datetime.now()

checked_years = [now.year]
if now.month == 12:
    checked_years.append(now.year+1)
if now.month == 1:
    checked_years.append(now.year-1)
checked_months = [now.month]
if now.month != 12:
    checked_months.append(now.month+1)
if now.month != 1:
    checked_months.append(now.month-1)

mapping = ["jan", "feb", "mar", "apr", "may", "jun", "jul", "aug", "sep", "oct", "nov", "dec"]
checked_months = [mapping[i-1] for i in checked_months]

past_manifest = []
upcoming_manifest = []

for y in checked_years:
    for m in checked_months:
        for state in ["past", "upcoming"]:
            path = f"{calendar}/{state}/{y}/{m}/"
            
            if not os.path.isdir(path): continue
            days = os.listdir(path)
            for d in days:
                for event in os.listdir(f"{path}/{d}"):
                    manifest = getattr(modules[__name__], state+"_manifest")
                    manifest.append(f"{path}/{d}/{event}")

manifest_path = f"{calendar}/manifest.json"

if not os.path.exists(manifest_path):
    with open(manifest_path, "w") as f:
        f.write("{}")

with open(manifest_path, "r") as f:
    manifest = json.loads(f.read())

for e in past_manifest + upcoming_manifest:
    if not os.path.exists(e) or not e.endswith(".yml"):
        if manifest.get(e): manifest.pop(e)
        continue
    e = e.replace("//", "/").replace("//", "/").replace("//", "/") #f1x shit hack
    #e = e.replace("/upcoming/", "/aaa/")
    #e = e.replace("/past/", "/aaa/")

    exp = e.split("/")
    if len(exp) < 5: continue
    upcoming = exp[-5]
    year = exp[-4]
    month = exp[-3]
    day = exp[-2]
    name = exp[-1]
    
    with open(e, "r") as f:
        info = yaml.safe_load(f.read())

    hour = int(info["time"] / 100)
    minute = info["time"] % 100

    dtime = datetime.datetime(int(year), mapping.index(month)+1, int(day), hour, minute)
    if dtime < now and upcoming == "upcoming":
        ee = e.replace("/upcoming/", "/past/")
        ee = os.path.dirname(ee)
        os.makedirs(os.path.dirname(ee), exist_ok=True)
        shutil.move(os.path.dirname(e), os.path.abspath(os.path.join(ee, "..")))
        if manifest.get(e):
            manifest.pop(e)
        e = e.replace("/upcoming/", "/past/")
        upcoming = "past"

    entry = info
    entry["upcoming"] = upcoming == "upcoming"
    entry["datetime"] = int(dtime.timestamp())
    entry["hooked"] = [] if manifest.get(e) is None else manifest[e]["hooked"]
    manifest[e] = entry

for k in manifest.copy():
    if not os.path.isfile(k): manifest.pop(k)

with open(manifest_path, "w") as f:
    f.write(json.dumps(manifest, indent=4))

hooks = os.path.dirname(os.path.abspath(__file__))+"/hooks"

if not os.path.exists(hooks+"/config.yml"):
    with open(hooks+"/config.yml", "w") as f:
        f.write(yaml.safe_dump({
            "calendar-output-path": "/tmp/vtluug-calendar.html",
            "mediawiki-username": "u",
            "mediawiki-passwd": "p",
            "email-username": "u",
            "email-passwd": "p"
            }))

for script in os.listdir(hooks+"/always"):
    subprocess.run([f"{hooks}/always/{script}", os.path.abspath(calendar+"/manifest.json")])

for script in glob.glob(hooks+"/days-before/**/*.*"):
    subprocess.run([script, os.path.abspath(calendar+"/manifest.json")])