Compare commits
7 Commits
243bdbb685
...
main
| Author | SHA1 | Date | |
|---|---|---|---|
| e4bd1bbc40 | |||
| 1fa8b21882 | |||
| 61f6070729 | |||
|
|
5f841ddbf9 | ||
|
|
11a91eb213 | ||
|
|
8d80985d79 | ||
|
|
069a2d2c63 |
82
API.py
82
API.py
@@ -78,4 +78,84 @@ class Webserver:
|
|||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
print(f"🚀 Starting server on {self.ip}:{self.port}")
|
print(f"🚀 Starting server on {self.ip}:{self.port}")
|
||||||
self.app.run(host=self.ip, port=self.port)
|
from flask import Flask, Response
|
||||||
|
from icalendar import Calendar, Event
|
||||||
|
from datetime import datetime
|
||||||
|
import pytz
|
||||||
|
import uuid
|
||||||
|
|
||||||
|
|
||||||
|
class Webserver:
|
||||||
|
def __init__(self, ip, port, db_instance):
|
||||||
|
self.ip = ip
|
||||||
|
self.port = port
|
||||||
|
self.db = db_instance
|
||||||
|
|
||||||
|
self.app = Flask(__name__)
|
||||||
|
self.tz = pytz.timezone("Europe/Amsterdam")
|
||||||
|
|
||||||
|
self._register_routes()
|
||||||
|
|
||||||
|
def _register_routes(self):
|
||||||
|
self.app.add_url_rule("/", "index", lambda: "Calendar server running ✅")
|
||||||
|
self.app.add_url_rule("/calendar.ics", "calendar", self._calendar_endpoint)
|
||||||
|
|
||||||
|
def _fetch_shifts(self):
|
||||||
|
con = self.db._get_connection()
|
||||||
|
cur = con.cursor()
|
||||||
|
|
||||||
|
cur.execute("""
|
||||||
|
SELECT shift_start, shift_end, department, description
|
||||||
|
FROM shifts
|
||||||
|
ORDER BY shift_start ASC
|
||||||
|
""")
|
||||||
|
|
||||||
|
rows = cur.fetchall()
|
||||||
|
con.close()
|
||||||
|
|
||||||
|
shifts = []
|
||||||
|
for row in rows:
|
||||||
|
start_str, end_str, department, description = row
|
||||||
|
|
||||||
|
# Convert string → datetime
|
||||||
|
start = self._parse_datetime(start_str)
|
||||||
|
end = self._parse_datetime(end_str)
|
||||||
|
|
||||||
|
shifts.append({
|
||||||
|
"start": start,
|
||||||
|
"end": end,
|
||||||
|
"description": f"{description} ",
|
||||||
|
"department": f"{department} "
|
||||||
|
})
|
||||||
|
|
||||||
|
return shifts
|
||||||
|
|
||||||
|
def _parse_datetime(self, dt_str):
|
||||||
|
dt = datetime.fromisoformat(dt_str)
|
||||||
|
if dt.tzinfo is None:
|
||||||
|
dt = self.tz.localize(dt)
|
||||||
|
return dt
|
||||||
|
|
||||||
|
def _calendar_endpoint(self):
|
||||||
|
cal = Calendar()
|
||||||
|
cal.add('prodid', '-//Shift Calendar//example//')
|
||||||
|
cal.add('version', '2.0')
|
||||||
|
|
||||||
|
shifts = self._fetch_shifts()
|
||||||
|
|
||||||
|
for shift in shifts:
|
||||||
|
event = Event()
|
||||||
|
event.add('uid', str(uuid.uuid4()))
|
||||||
|
event.add('dtstart', shift["start"])
|
||||||
|
event.add('dtend', shift["end"])
|
||||||
|
event.add('summary', shift["department"])
|
||||||
|
event.add('description', shift["description"])
|
||||||
|
event.add('dtstamp', datetime.now(tz=self.tz))
|
||||||
|
|
||||||
|
cal.add_component(event)
|
||||||
|
|
||||||
|
return Response(cal.to_ical(), mimetype="text/calendar")
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
print(f"🚀 Starting server on {self.ip}:{self.port}")
|
||||||
|
self.app.run(host=self.ip, port=self.port, threaded=True)
|
||||||
17
docker-compose.yml
Normal file
17
docker-compose.yml
Normal file
@@ -0,0 +1,17 @@
|
|||||||
|
services:
|
||||||
|
app:
|
||||||
|
build: .
|
||||||
|
container_name: pmt
|
||||||
|
ports:
|
||||||
|
- "8080:8080"
|
||||||
|
restart: unless-stopped
|
||||||
|
networks:
|
||||||
|
- proxy-network
|
||||||
|
volumes:
|
||||||
|
- "./data:/app/data"
|
||||||
|
env_file:
|
||||||
|
- .env
|
||||||
|
|
||||||
|
networks:
|
||||||
|
proxy-network:
|
||||||
|
external: true
|
||||||
8
main.py
8
main.py
@@ -40,12 +40,14 @@ def main():
|
|||||||
|
|
||||||
|
|
||||||
# Every hour at minute 0
|
# Every hour at minute 0
|
||||||
scheduler = BackgroundScheduler()
|
scheduler = BackgroundScheduler(timezone="Europe/Amsterdam")
|
||||||
scheduler.add_job(
|
scheduler.add_job(
|
||||||
lambda: update_shifts(pmt, db),
|
lambda: update_shifts(pmt, db),
|
||||||
trigger='cron',
|
trigger="interval",
|
||||||
minute=0
|
hours=1,
|
||||||
|
next_run_time=datetime.now()
|
||||||
)
|
)
|
||||||
|
scheduler.start()
|
||||||
|
|
||||||
webserver.run()
|
webserver.run()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user