initial commit
This commit is contained in:
commit
cc1dc2c4d8
4 changed files with 522 additions and 0 deletions
174
.gitignore
vendored
Normal file
174
.gitignore
vendored
Normal file
|
|
@ -0,0 +1,174 @@
|
|||
# Byte-compiled / optimized / DLL files
|
||||
__pycache__/
|
||||
*.py[cod]
|
||||
*$py.class
|
||||
|
||||
# C extensions
|
||||
*.so
|
||||
|
||||
# Distribution / packaging
|
||||
.Python
|
||||
build/
|
||||
develop-eggs/
|
||||
dist/
|
||||
downloads/
|
||||
eggs/
|
||||
.eggs/
|
||||
lib/
|
||||
lib64/
|
||||
parts/
|
||||
sdist/
|
||||
var/
|
||||
wheels/
|
||||
share/python-wheels/
|
||||
*.egg-info/
|
||||
.installed.cfg
|
||||
*.egg
|
||||
MANIFEST
|
||||
|
||||
# PyInstaller
|
||||
# Usually these files are written by a python script from a template
|
||||
# before PyInstaller builds the exe, so as to inject date/other infos into it.
|
||||
*.manifest
|
||||
*.spec
|
||||
|
||||
# Installer logs
|
||||
pip-log.txt
|
||||
pip-delete-this-directory.txt
|
||||
|
||||
# Unit test / coverage reports
|
||||
htmlcov/
|
||||
.tox/
|
||||
.nox/
|
||||
.coverage
|
||||
.coverage.*
|
||||
.cache
|
||||
nosetests.xml
|
||||
coverage.xml
|
||||
*.cover
|
||||
*.py,cover
|
||||
.hypothesis/
|
||||
.pytest_cache/
|
||||
cover/
|
||||
|
||||
# Translations
|
||||
*.mo
|
||||
*.pot
|
||||
|
||||
# Django stuff:
|
||||
*.log
|
||||
local_settings.py
|
||||
db.sqlite3
|
||||
db.sqlite3-journal
|
||||
|
||||
# Flask stuff:
|
||||
instance/
|
||||
.webassets-cache
|
||||
|
||||
# Scrapy stuff:
|
||||
.scrapy
|
||||
|
||||
# Sphinx documentation
|
||||
docs/_build/
|
||||
|
||||
# PyBuilder
|
||||
.pybuilder/
|
||||
target/
|
||||
|
||||
# Jupyter Notebook
|
||||
.ipynb_checkpoints
|
||||
|
||||
# IPython
|
||||
profile_default/
|
||||
ipython_config.py
|
||||
|
||||
# pyenv
|
||||
# For a library or package, you might want to ignore these files since the code is
|
||||
# intended to run in multiple environments; otherwise, check them in:
|
||||
# .python-version
|
||||
|
||||
# pipenv
|
||||
# According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
|
||||
# However, in case of collaboration, if having platform-specific dependencies or dependencies
|
||||
# having no cross-platform support, pipenv may install dependencies that don't work, or not
|
||||
# install all needed dependencies.
|
||||
#Pipfile.lock
|
||||
|
||||
# UV
|
||||
# Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
|
||||
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
||||
# commonly ignored for libraries.
|
||||
#uv.lock
|
||||
|
||||
# poetry
|
||||
# Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
|
||||
# This is especially recommended for binary packages to ensure reproducibility, and is more
|
||||
# commonly ignored for libraries.
|
||||
# https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
|
||||
#poetry.lock
|
||||
|
||||
# pdm
|
||||
# Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
|
||||
#pdm.lock
|
||||
# pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
|
||||
# in version control.
|
||||
# https://pdm.fming.dev/latest/usage/project/#working-with-version-control
|
||||
.pdm.toml
|
||||
.pdm-python
|
||||
.pdm-build/
|
||||
|
||||
# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
|
||||
__pypackages__/
|
||||
|
||||
# Celery stuff
|
||||
celerybeat-schedule
|
||||
celerybeat.pid
|
||||
|
||||
# SageMath parsed files
|
||||
*.sage.py
|
||||
|
||||
# Environments
|
||||
.env
|
||||
.venv
|
||||
env/
|
||||
venv/
|
||||
ENV/
|
||||
env.bak/
|
||||
venv.bak/
|
||||
|
||||
# Spyder project settings
|
||||
.spyderproject
|
||||
.spyproject
|
||||
|
||||
# Rope project settings
|
||||
.ropeproject
|
||||
|
||||
# mkdocs documentation
|
||||
/site
|
||||
|
||||
# mypy
|
||||
.mypy_cache/
|
||||
.dmypy.json
|
||||
dmypy.json
|
||||
|
||||
# Pyre type checker
|
||||
.pyre/
|
||||
|
||||
# pytype static type analyzer
|
||||
.pytype/
|
||||
|
||||
# Cython debug symbols
|
||||
cython_debug/
|
||||
|
||||
# PyCharm
|
||||
# JetBrains specific template is maintained in a separate JetBrains.gitignore that can
|
||||
# be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
|
||||
# and can be added to the global gitignore or merged into this file. For a more nuclear
|
||||
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
||||
#.idea/
|
||||
|
||||
# Ruff stuff:
|
||||
.ruff_cache/
|
||||
|
||||
# PyPI configuration file
|
||||
.pypirc
|
||||
38
Dockerfile
Normal file
38
Dockerfile
Normal file
|
|
@ -0,0 +1,38 @@
|
|||
# ┌────────────────────────── build stage ──────────────────────────┐
|
||||
FROM python:3.11-slim as builder
|
||||
WORKDIR /app
|
||||
|
||||
# install build deps if any (none here), copy requirements
|
||||
COPY requirements.txt .
|
||||
RUN pip install --no-cache-dir --upgrade pip \
|
||||
&& pip install --no-cache-dir -r requirements.txt
|
||||
|
||||
# └────────────────────────── final stage ──────────────────────────┘
|
||||
FROM python:3.11-slim
|
||||
WORKDIR /app
|
||||
|
||||
# copy only installed packages from builder (local pip cache)
|
||||
COPY --from=builder /usr/local/lib/python3.11/site-packages \
|
||||
/usr/local/lib/python3.11/site-packages
|
||||
COPY --from=builder /usr/local/bin /usr/local/bin
|
||||
|
||||
# copy app code
|
||||
COPY app.py .
|
||||
|
||||
# expose port
|
||||
EXPOSE 8000
|
||||
|
||||
# default ENV (you can override these in your orchestrator)
|
||||
ENV BEAVERHABITS_URL="http://beaverhabits.lan" \
|
||||
BEAVERHABITS_USERNAME="changeme" \
|
||||
BEAVERHABITS_PASSWORD="changeme"
|
||||
|
||||
# use a non-root user if you like; for brevity we run as root.
|
||||
|
||||
# run via Gunicorn WSGI server with 4 workers
|
||||
CMD ["gunicorn", "app:app", \
|
||||
"--bind", "0.0.0.0:8000", \
|
||||
"--workers", "4", \
|
||||
"--threads", "4", \
|
||||
"--access-logfile", "-", \
|
||||
"--error-logfile", "-"]
|
||||
306
app.py
Normal file
306
app.py
Normal file
|
|
@ -0,0 +1,306 @@
|
|||
import os
|
||||
import time
|
||||
import logging
|
||||
import datetime
|
||||
import threading
|
||||
|
||||
from flask import Flask, request, Response, abort, jsonify
|
||||
from icalendar import Calendar, Todo
|
||||
import xml.etree.ElementTree as ET
|
||||
import requests
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Configuration via ENV
|
||||
# -----------------------------------------------------------------------------
|
||||
BEAVER_URL = os.getenv("BEAVERHABITS_URL", "")
|
||||
USERNAME = os.getenv("BEAVERHABITS_USERNAME", "")
|
||||
PASSWORD = os.getenv("BEAVERHABITS_PASSWORD", "")
|
||||
# How long to reuse a token (secs)
|
||||
TOKEN_TTL = int(os.getenv("BEAVERHABITS_TOKEN_TTL", "3600"))
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Logging
|
||||
# -----------------------------------------------------------------------------
|
||||
logging.basicConfig(
|
||||
level=logging.INFO,
|
||||
format="%(asctime)s %(levelname)s %(name)s %(message)s",
|
||||
)
|
||||
log = logging.getLogger("caldav-proxy")
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Flask app & XML namespaces
|
||||
# -----------------------------------------------------------------------------
|
||||
app = Flask(__name__)
|
||||
NS = {"D": "DAV:", "C": "urn:ietf:params:xml:ns:caldav"}
|
||||
for p, uri in NS.items():
|
||||
ET.register_namespace(p, uri)
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Beaver Habits token cache with TTL
|
||||
# -----------------------------------------------------------------------------
|
||||
_token_lock = threading.Lock()
|
||||
_token: str | None = None
|
||||
_token_ts: float = 0.0
|
||||
_session = requests.Session()
|
||||
|
||||
def get_token() -> str:
|
||||
global _token, _token_ts
|
||||
with _token_lock:
|
||||
now = time.time()
|
||||
if _token and (now - _token_ts) < TOKEN_TTL:
|
||||
return _token
|
||||
|
||||
# fetch a new token
|
||||
log.info("Requesting new Beaver Habits token")
|
||||
r = _session.post(
|
||||
f"{BEAVER_URL}/auth/login",
|
||||
headers={"Accept": "application/json",
|
||||
"Content-Type": "application/x-www-form-urlencoded"},
|
||||
data={
|
||||
"grant_type": "password",
|
||||
"username": USERNAME,
|
||||
"password": PASSWORD,
|
||||
},
|
||||
timeout=5,
|
||||
)
|
||||
r.raise_for_status()
|
||||
js = r.json()
|
||||
_token = js["access_token"]
|
||||
_token_ts = now
|
||||
return _token
|
||||
|
||||
def beaver_headers() -> dict[str,str]:
|
||||
return {"Accept": "application/json",
|
||||
"Authorization": f"Bearer {get_token()}"}
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Utility: build iCalendar VTODO
|
||||
# -----------------------------------------------------------------------------
|
||||
def make_vtodo(uid: str, summary: str, completed: bool) -> bytes:
|
||||
cal = Calendar()
|
||||
cal.add("prodid", "-//CalDAV⇆BeaverHabits Proxy//EN")
|
||||
cal.add("version", "2.0")
|
||||
todo = Todo()
|
||||
todo.add("uid", uid)
|
||||
todo.add("summary", summary)
|
||||
now = datetime.datetime.now(datetime.timezone.utc)
|
||||
todo.add("dtstamp", now)
|
||||
todo.add("due", now + datetime.timedelta(days=1))
|
||||
todo.add("status", "COMPLETED" if completed else "NEEDS-ACTION")
|
||||
cal.add_component(todo)
|
||||
return cal.to_ical()
|
||||
|
||||
def to_bytes(elem: ET.Element) -> bytes:
|
||||
return ET.tostring(elem, encoding="utf-8", xml_declaration=True)
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Helper: fetch habits + today's completion from Beaver
|
||||
# -----------------------------------------------------------------------------
|
||||
def fetch_tasks() -> dict[str, dict]:
|
||||
"""Return a dict of {habit_id: {SUMMARY, STATUS, DTSTAMP, DUE}}."""
|
||||
try:
|
||||
r = _session.get(f"{BEAVER_URL}/api/v1/habits", headers=beaver_headers(), timeout=5)
|
||||
r.raise_for_status()
|
||||
habits = r.json()
|
||||
except Exception as e:
|
||||
log.error("Failed to list habits: %s", e)
|
||||
return {}
|
||||
|
||||
today = datetime.date.today().strftime("%d-%m-%Y")
|
||||
out: dict[str, dict] = {}
|
||||
for h in habits:
|
||||
hid = h["id"]
|
||||
name = h["name"]
|
||||
# check completion for today
|
||||
try:
|
||||
params = {
|
||||
"date_fmt": "%d-%m-%Y",
|
||||
"date_start": today,
|
||||
"date_end": today,
|
||||
"sort": "asc"
|
||||
}
|
||||
rc = _session.get(
|
||||
f"{BEAVER_URL}/api/v1/habits/{hid}/completions",
|
||||
headers=beaver_headers(),
|
||||
params=params,
|
||||
timeout=5
|
||||
)
|
||||
rc.raise_for_status()
|
||||
done_list = rc.json()
|
||||
completed = today in done_list
|
||||
except Exception:
|
||||
log.warning("Failed to get completions for %s", hid)
|
||||
completed = False
|
||||
|
||||
out[hid] = {
|
||||
"SUMMARY": name,
|
||||
"STATUS": "COMPLETED" if completed else "NEEDS-ACTION",
|
||||
"DTSTAMP": datetime.datetime.now(datetime.timezone.utc),
|
||||
"DUE": datetime.datetime.now(datetime.timezone.utc) + datetime.timedelta(days=1),
|
||||
}
|
||||
return out
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Standard PROPFIND / discovery endpoints
|
||||
# -----------------------------------------------------------------------------
|
||||
@app.route("/", methods=["PROPFIND"])
|
||||
def propfind_root():
|
||||
ms = ET.Element(f"{{{NS['D']}}}multistatus")
|
||||
# current-user-principal
|
||||
r1 = ET.SubElement(ms, f"{{{NS['D']}}}response")
|
||||
ET.SubElement(r1, f"{{{NS['D']}}}href").text = "/"
|
||||
ps1 = ET.SubElement(r1, f"{{{NS['D']}}}propstat")
|
||||
p1 = ET.SubElement(ps1, f"{{{NS['D']}}}prop")
|
||||
cup = ET.SubElement(p1, f"{{{NS['D']}}}current-user-principal")
|
||||
ET.SubElement(cup, f"{{{NS['D']}}}href").text = "/"
|
||||
ET.SubElement(ps1, f"{{{NS['D']}}}status").text = "HTTP/1.1 200 OK"
|
||||
# calendar-home-set
|
||||
r2 = ET.SubElement(ms, f"{{{NS['D']}}}response")
|
||||
ET.SubElement(r2, f"{{{NS['D']}}}href").text = "/"
|
||||
ps2 = ET.SubElement(r2, f"{{{NS['D']}}}propstat")
|
||||
p2 = ET.SubElement(ps2, f"{{{NS['D']}}}prop")
|
||||
chs = ET.SubElement(p2, f"{{{NS['C']}}}calendar-home-set")
|
||||
ET.SubElement(chs, f"{{{NS['D']}}}href").text = "/caldav/"
|
||||
ET.SubElement(ps2, f"{{{NS['D']}}}status").text = "HTTP/1.1 200 OK"
|
||||
return Response(to_bytes(ms), status=207, mimetype="application/xml")
|
||||
|
||||
@app.route("/caldav/", methods=["PROPFIND"])
|
||||
def propfind_caldav():
|
||||
ms = ET.Element(f"{{{NS['D']}}}multistatus")
|
||||
# home collection
|
||||
r1 = ET.SubElement(ms, f"{{{NS['D']}}}response")
|
||||
ET.SubElement(r1, f"{{{NS['D']}}}href").text = "/caldav/"
|
||||
ps1 = ET.SubElement(r1, f"{{{NS['D']}}}propstat")
|
||||
p1 = ET.SubElement(ps1, f"{{{NS['D']}}}prop")
|
||||
rt1 = ET.SubElement(p1, f"{{{NS['D']}}}resourcetype")
|
||||
ET.SubElement(rt1, f"{{{NS['D']}}}collection")
|
||||
ET.SubElement(ps1, f"{{{NS['D']}}}status").text = "HTTP/1.1 200 OK"
|
||||
# our single tasks calendar
|
||||
href="/caldav/tasks/"
|
||||
r2 = ET.SubElement(ms, f"{{{NS['D']}}}response")
|
||||
ET.SubElement(r2, f"{{{NS['D']}}}href").text = href
|
||||
ps2 = ET.SubElement(r2, f"{{{NS['D']}}}propstat")
|
||||
p2 = ET.SubElement(ps2, f"{{{NS['D']}}}prop")
|
||||
rt2 = ET.SubElement(p2, f"{{{NS['D']}}}resourcetype")
|
||||
ET.SubElement(rt2, f"{{{NS['D']}}}collection")
|
||||
ET.SubElement(rt2, f"{{{NS['C']}}}calendar")
|
||||
sccs=ET.SubElement(p2, f"{{{NS['C']}}}supported-calendar-component-set")
|
||||
ET.SubElement(sccs, f"{{{NS['C']}}}comp", name="VTODO")
|
||||
ET.SubElement(p2, f"{{{NS['D']}}}displayname").text="Tasks"
|
||||
ET.SubElement(ps2,f"{{{NS['D']}}}status").text="HTTP/1.1 200 OK"
|
||||
return Response(to_bytes(ms), status=207, mimetype="application/xml")
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# PROPFIND + REPORT + GET/PUT on /caldav/tasks/
|
||||
# -----------------------------------------------------------------------------
|
||||
@app.route("/caldav/tasks/", methods=["PROPFIND", "REPORT"])
|
||||
def tasks_collection():
|
||||
"""
|
||||
PROPFIND: list resource URLs + etags
|
||||
REPORT: return full calendar-data for each VTODO
|
||||
"""
|
||||
tasks = fetch_tasks()
|
||||
multistatus = ET.Element(f"{{{NS['D']}}}multistatus")
|
||||
|
||||
# Collection itself (must advertise calendar + VTODO support)
|
||||
r0 = ET.SubElement(multistatus, f"{{{NS['D']}}}response")
|
||||
ET.SubElement(r0, f"{{{NS['D']}}}href").text = "/caldav/tasks/"
|
||||
ps0 = ET.SubElement(r0, f"{{{NS['D']}}}propstat")
|
||||
p0 = ET.SubElement(ps0, f"{{{NS['D']}}}prop")
|
||||
rt0 = ET.SubElement(p0, f"{{{NS['D']}}}resourcetype")
|
||||
ET.SubElement(rt0, f"{{{NS['D']}}}collection")
|
||||
ET.SubElement(rt0, f"{{{NS['C']}}}calendar")
|
||||
sccs0 = ET.SubElement(p0, f"{{{NS['C']}}}supported-calendar-component-set")
|
||||
ET.SubElement(sccs0, f"{{{NS['C']}}}comp", name="VTODO")
|
||||
ET.SubElement(ps0, f"{{{NS['D']}}}status").text="HTTP/1.1 200 OK"
|
||||
|
||||
for uid, props in tasks.items():
|
||||
href = f"/caldav/tasks/{uid}.ics"
|
||||
etag = f"\"{uid}-{props['STATUS']}\""
|
||||
body = make_vtodo(uid, props["SUMMARY"], props["STATUS"]=="COMPLETED")
|
||||
|
||||
# PROPFIND section
|
||||
if request.method == "PROPFIND":
|
||||
r = ET.SubElement(multistatus, f"{{{NS['D']}}}response")
|
||||
ET.SubElement(r, f"{{{NS['D']}}}href").text = href
|
||||
ps = ET.SubElement(r, f"{{{NS['D']}}}propstat")
|
||||
p = ET.SubElement(ps, f"{{{NS['D']}}}prop")
|
||||
ET.SubElement(p, f"{{{NS['D']}}}getcontentlength").text = str(len(body))
|
||||
ET.SubElement(p, f"{{{NS['D']}}}getcontenttype").text = "text/calendar; charset=utf-8"
|
||||
ET.SubElement(p, f"{{{NS['D']}}}getetag").text = etag
|
||||
ET.SubElement(ps,f"{{{NS['D']}}}status").text = "HTTP/1.1 200 OK"
|
||||
|
||||
# REPORT section
|
||||
elif request.method == "REPORT":
|
||||
r = ET.SubElement(multistatus, f"{{{NS['D']}}}response")
|
||||
ET.SubElement(r, f"{{{NS['D']}}}href").text = href
|
||||
ps = ET.SubElement(r, f"{{{NS['D']}}}propstat")
|
||||
p = ET.SubElement(ps, f"{{{NS['D']}}}prop")
|
||||
cd = ET.SubElement(p, f"{{{NS['C']}}}calendar-data")
|
||||
cd.text = body.decode("utf-8")
|
||||
ET.SubElement(p, f"{{{NS['D']}}}getetag").text = etag
|
||||
ET.SubElement(ps,f"{{{NS['D']}}}status").text="HTTP/1.1 200 OK"
|
||||
|
||||
return Response(to_bytes(multistatus), status=207, mimetype="application/xml")
|
||||
|
||||
@app.route("/caldav/tasks/<uid>.ics", methods=["GET", "PUT"])
|
||||
def single_task(uid):
|
||||
"""
|
||||
GET: return VTODO
|
||||
PUT: update completion for today in Beaver Habits
|
||||
"""
|
||||
# verify habit exists
|
||||
tasks_map = fetch_tasks()
|
||||
if uid not in tasks_map:
|
||||
abort(404)
|
||||
|
||||
summary = tasks_map[uid]["SUMMARY"]
|
||||
completed = (tasks_map[uid]["STATUS"] == "COMPLETED")
|
||||
today = datetime.date.today()
|
||||
|
||||
if request.method == "GET":
|
||||
return Response(make_vtodo(uid, summary, completed),
|
||||
mimetype="text/calendar")
|
||||
|
||||
# PUT: parse STATUS and call Beaver Habits
|
||||
cal = Calendar.from_ical(request.data)
|
||||
new_status = None
|
||||
for comp in cal.walk():
|
||||
if comp.name == "VTODO":
|
||||
new_status = str(comp.get("STATUS"))
|
||||
break
|
||||
|
||||
if new_status not in ("NEEDS-ACTION", "COMPLETED"):
|
||||
abort(400, "Unsupported STATUS")
|
||||
|
||||
done = (new_status == "COMPLETED")
|
||||
try:
|
||||
_session.post(
|
||||
f"{BEAVER_URL}/api/v1/habits/{uid}/completions",
|
||||
headers={**beaver_headers(), "Content-Type": "application/json"},
|
||||
json={
|
||||
"date_fmt": "%d-%m-%Y",
|
||||
"date": today.strftime("%d-%m-%Y"),
|
||||
"done": done
|
||||
},
|
||||
timeout=5
|
||||
).raise_for_status()
|
||||
except Exception as e:
|
||||
log.error("Failed to set completion: %s", e)
|
||||
abort(500)
|
||||
|
||||
return ("", 204)
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Health‐check
|
||||
# -----------------------------------------------------------------------------
|
||||
@app.route("/healthz", methods=["GET"])
|
||||
def healthz():
|
||||
return jsonify({"status": "ok"}), 200
|
||||
|
||||
# -----------------------------------------------------------------------------
|
||||
# Start via WSGI
|
||||
# -----------------------------------------------------------------------------
|
||||
if __name__ == "__main__":
|
||||
# Development only
|
||||
app.run(host="0.0.0.0", port=8000, debug=True)
|
||||
4
requirements.txt
Normal file
4
requirements.txt
Normal file
|
|
@ -0,0 +1,4 @@
|
|||
Flask==2.3.2
|
||||
icalendar==4.0.9
|
||||
requests==2.31.0
|
||||
gunicorn
|
||||
Loading…
Add table
Add a link
Reference in a new issue