initial commit
This commit is contained in:
commit
d9ae6527a1
GPG Key ID:
8D83DE470F8544E7
8
.idea/.gitignore
vendored
Normal file
8
.idea/.gitignore
vendored
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
# Default ignored files
|
||||||
|
/shelf/
|
||||||
|
/workspace.xml
|
||||||
|
# Editor-based HTTP Client requests
|
||||||
|
/httpRequests/
|
||||||
|
# Datasource local storage ignored files
|
||||||
|
/dataSources/
|
||||||
|
/dataSources.local.xml
|
||||||
8
.idea/beebot-docker.iml
Normal file
8
.idea/beebot-docker.iml
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<module type="PYTHON_MODULE" version="4">
|
||||||
|
<component name="NewModuleRootManager">
|
||||||
|
<content url="file://$MODULE_DIR$" />
|
||||||
|
<orderEntry type="inheritedJdk" />
|
||||||
|
<orderEntry type="sourceFolder" forTests="false" />
|
||||||
|
</component>
|
||||||
|
</module>
|
||||||
189
.idea/icon.svg
Normal file
189
.idea/icon.svg
Normal file
@ -0,0 +1,189 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8" standalone="no"?>
|
||||||
|
<!-- Created with Inkscape (http://www.inkscape.org/) -->
|
||||||
|
|
||||||
|
<svg
|
||||||
|
width="64"
|
||||||
|
height="64"
|
||||||
|
viewBox="0 0 16.933333 16.933333"
|
||||||
|
version="1.1"
|
||||||
|
id="svg1"
|
||||||
|
inkscape:version="1.3 (0e150ed6c4, 2023-07-21)"
|
||||||
|
sodipodi:docname="logo.svg"
|
||||||
|
inkscape:export-filename="logo.png"
|
||||||
|
inkscape:export-xdpi="768"
|
||||||
|
inkscape:export-ydpi="768"
|
||||||
|
xmlns:inkscape="http://www.inkscape.org/namespaces/inkscape"
|
||||||
|
xmlns:sodipodi="http://sodipodi.sourceforge.net/DTD/sodipodi-0.dtd"
|
||||||
|
xmlns="http://www.w3.org/2000/svg"
|
||||||
|
xmlns:svg="http://www.w3.org/2000/svg">
|
||||||
|
<sodipodi:namedview
|
||||||
|
id="namedview1"
|
||||||
|
pagecolor="#ffffff"
|
||||||
|
bordercolor="#000000"
|
||||||
|
borderopacity="0.25"
|
||||||
|
inkscape:showpageshadow="2"
|
||||||
|
inkscape:pageopacity="0.0"
|
||||||
|
inkscape:pagecheckerboard="0"
|
||||||
|
inkscape:deskcolor="#d1d1d1"
|
||||||
|
inkscape:document-units="px"
|
||||||
|
showgrid="true"
|
||||||
|
inkscape:zoom="12.015825"
|
||||||
|
inkscape:cx="26.132205"
|
||||||
|
inkscape:cy="33.122986"
|
||||||
|
inkscape:window-width="1858"
|
||||||
|
inkscape:window-height="1016"
|
||||||
|
inkscape:window-x="1982"
|
||||||
|
inkscape:window-y="27"
|
||||||
|
inkscape:window-maximized="1"
|
||||||
|
inkscape:current-layer="layer1">
|
||||||
|
<inkscape:grid
|
||||||
|
id="grid1"
|
||||||
|
units="px"
|
||||||
|
originx="0"
|
||||||
|
originy="0"
|
||||||
|
spacingx="0.26458333"
|
||||||
|
spacingy="0.26458333"
|
||||||
|
empcolor="#0099e5"
|
||||||
|
empopacity="0.30196078"
|
||||||
|
color="#0099e5"
|
||||||
|
opacity="0.14901961"
|
||||||
|
empspacing="8"
|
||||||
|
dotted="false"
|
||||||
|
gridanglex="30"
|
||||||
|
gridanglez="30"
|
||||||
|
visible="true" />
|
||||||
|
</sodipodi:namedview>
|
||||||
|
<defs
|
||||||
|
id="defs1">
|
||||||
|
<inkscape:path-effect
|
||||||
|
effect="powermask"
|
||||||
|
id="path-effect7"
|
||||||
|
is_visible="true"
|
||||||
|
lpeversion="1"
|
||||||
|
uri="#mask-powermask-path-effect7"
|
||||||
|
invert="false"
|
||||||
|
hide_mask="false"
|
||||||
|
background="true"
|
||||||
|
background_color="#ffffffff" />
|
||||||
|
<filter
|
||||||
|
id="mask-powermask-path-effect5_inverse"
|
||||||
|
inkscape:label="filtermask-powermask-path-effect5"
|
||||||
|
style="color-interpolation-filters:sRGB"
|
||||||
|
height="100"
|
||||||
|
width="100"
|
||||||
|
x="-50"
|
||||||
|
y="-50">
|
||||||
|
<feColorMatrix
|
||||||
|
id="mask-powermask-path-effect5_primitive1"
|
||||||
|
values="1"
|
||||||
|
type="saturate"
|
||||||
|
result="fbSourceGraphic" />
|
||||||
|
<feColorMatrix
|
||||||
|
id="mask-powermask-path-effect5_primitive2"
|
||||||
|
values="-1 0 0 0 1 0 -1 0 0 1 0 0 -1 0 1 0 0 0 1 0 "
|
||||||
|
in="fbSourceGraphic" />
|
||||||
|
</filter>
|
||||||
|
<mask
|
||||||
|
maskUnits="userSpaceOnUse"
|
||||||
|
id="mask-powermask-path-effect7">
|
||||||
|
<path
|
||||||
|
id="mask-powermask-path-effect7_box"
|
||||||
|
style="fill:#ffffff;fill-opacity:1"
|
||||||
|
d="m 5.01963,2.175 h 6.894073 V 15.816666 H 5.01963 Z" />
|
||||||
|
<g
|
||||||
|
id="g7"
|
||||||
|
style="">
|
||||||
|
<rect
|
||||||
|
style="fill:#000000;fill-opacity:1;stroke:none;stroke-width:1.33906;stroke-linecap:round;stroke-linejoin:round;stroke-dasharray:none;stroke-dashoffset:0.799748;stroke-opacity:1"
|
||||||
|
id="rect5"
|
||||||
|
width="6.0584898"
|
||||||
|
height="0.73989791"
|
||||||
|
x="5.5562496"
|
||||||
|
y="9.948801"
|
||||||
|
d="m 5.5562496,9.948801 h 6.0584894 v 0.739898 H 5.5562496 Z" />
|
||||||
|
<rect
|
||||||
|
style="fill:#000000;fill-opacity:1;stroke:none;stroke-width:1.33906;stroke-linecap:round;stroke-linejoin:round;stroke-dasharray:none;stroke-dashoffset:0.799748;stroke-opacity:1"
|
||||||
|
id="rect6"
|
||||||
|
width="6.0584898"
|
||||||
|
height="0.73989791"
|
||||||
|
x="5.5562496"
|
||||||
|
y="11.536301"
|
||||||
|
d="m 5.5562496,11.536301 h 6.0584894 v 0.739898 H 5.5562496 Z" />
|
||||||
|
<rect
|
||||||
|
style="fill:#000000;fill-opacity:1;stroke:none;stroke-width:1.33906;stroke-linecap:round;stroke-linejoin:round;stroke-dasharray:none;stroke-dashoffset:0.799748;stroke-opacity:1"
|
||||||
|
id="rect7"
|
||||||
|
width="6.0584898"
|
||||||
|
height="0.73989791"
|
||||||
|
x="5.5562496"
|
||||||
|
y="13.1238"
|
||||||
|
d="m 5.5562496,13.1238 h 6.0584894 v 0.739898 H 5.5562496 Z" />
|
||||||
|
</g>
|
||||||
|
</mask>
|
||||||
|
<filter
|
||||||
|
id="mask-powermask-path-effect7_inverse"
|
||||||
|
inkscape:label="filtermask-powermask-path-effect7"
|
||||||
|
style="color-interpolation-filters:sRGB"
|
||||||
|
height="100"
|
||||||
|
width="100"
|
||||||
|
x="-50"
|
||||||
|
y="-50">
|
||||||
|
<feColorMatrix
|
||||||
|
id="mask-powermask-path-effect7_primitive1"
|
||||||
|
values="1"
|
||||||
|
type="saturate"
|
||||||
|
result="fbSourceGraphic" />
|
||||||
|
<feColorMatrix
|
||||||
|
id="mask-powermask-path-effect7_primitive2"
|
||||||
|
values="-1 0 0 0 1 0 -1 0 0 1 0 0 -1 0 1 0 0 0 1 0 "
|
||||||
|
in="fbSourceGraphic" />
|
||||||
|
</filter>
|
||||||
|
</defs>
|
||||||
|
<g
|
||||||
|
inkscape:label="Layer 1"
|
||||||
|
inkscape:groupmode="layer"
|
||||||
|
id="layer1">
|
||||||
|
<path
|
||||||
|
sodipodi:type="star"
|
||||||
|
style="fill:#e8d6ac;fill-opacity:1;stroke:none;stroke-width:0.264583;stroke-linecap:round;stroke-linejoin:round;stroke-dasharray:none;stroke-dashoffset:0.799748;stroke-opacity:1"
|
||||||
|
id="path8"
|
||||||
|
inkscape:flatsided="true"
|
||||||
|
sodipodi:sides="6"
|
||||||
|
sodipodi:cx="8.4666662"
|
||||||
|
sodipodi:cy="8.4666662"
|
||||||
|
sodipodi:r1="8.2020829"
|
||||||
|
sodipodi:r2="7.3787446"
|
||||||
|
sodipodi:arg1="-2.1657387e-16"
|
||||||
|
sodipodi:arg2="0.5235988"
|
||||||
|
inkscape:rounded="0"
|
||||||
|
inkscape:randomized="0"
|
||||||
|
d="m 16.668749,8.4666662 -4.101041,7.1032118 -8.2020831,0 -4.10104131,-7.1032118 4.10104131,-7.1032119 8.2020831,0 z" />
|
||||||
|
<path
|
||||||
|
style="fill:#181716;stroke-width:1.05833;stroke-linecap:round;stroke-linejoin:round;stroke-dashoffset:0.799748;fill-opacity:1"
|
||||||
|
d="m 8.4666663,14.816666 c 2.9104167,0 3.1749997,-7.4083331 1.0583333,-8.4666663 C 10.81418,5.4168093 10.583333,3.1749998 8.4666663,3.175 6.3499998,3.1750002 6.0854163,5.5562497 7.408333,6.3499997 5.2916664,7.4083329 5.5562499,14.816666 8.4666663,14.816666 Z"
|
||||||
|
id="path1"
|
||||||
|
sodipodi:nodetypes="zczcz"
|
||||||
|
mask="url(#mask-powermask-path-effect7)"
|
||||||
|
inkscape:path-effect="#path-effect7"
|
||||||
|
inkscape:original-d="m 8.4666663,14.816666 c 2.9104167,0 3.1749997,-7.4083331 1.0583333,-8.4666663 C 10.81418,5.4168093 10.583333,3.1749998 8.4666663,3.175 6.3499998,3.1750002 6.0854163,5.5562497 7.408333,6.3499997 5.2916664,7.4083329 5.5562499,14.816666 8.4666663,14.816666 Z" />
|
||||||
|
<path
|
||||||
|
style="fill:none;stroke:#181716;stroke-width:0.529167;stroke-linecap:round;stroke-linejoin:round;stroke-dasharray:none;stroke-dashoffset:0.799748;stroke-opacity:1"
|
||||||
|
d="m 9.7895828,6.3499997 c 0,-1.5875 4.7625002,0.2645833 4.4979162,1.8520832 C 14.022916,9.7895828 9.7895828,7.9374996 9.7895828,6.3499997 Z"
|
||||||
|
id="path2"
|
||||||
|
sodipodi:nodetypes="zzz" />
|
||||||
|
<path
|
||||||
|
style="fill:none;stroke:#181716;stroke-width:0.529167;stroke-linecap:round;stroke-linejoin:round;stroke-dasharray:none;stroke-dashoffset:0.799748;stroke-opacity:1"
|
||||||
|
d="m 7.1437496,6.3499997 c 0,-1.5875 -4.7625002,0.2645833 -4.4979162,1.8520832 0.264583,1.5875 4.4979162,-0.2645833 4.4979162,-1.8520832 z"
|
||||||
|
id="path2-6"
|
||||||
|
sodipodi:nodetypes="zzz" />
|
||||||
|
<path
|
||||||
|
style="fill:none;fill-opacity:1;stroke:#181716;stroke-width:0.26458333;stroke-linecap:round;stroke-linejoin:round;stroke-dasharray:none;stroke-dashoffset:0.799748;stroke-opacity:1"
|
||||||
|
d="M 9.2604162,3.4395832 C 9.7603245,2.3048106 10.945507,2.0921784 11.641666,2.1166666"
|
||||||
|
id="path7"
|
||||||
|
sodipodi:nodetypes="cc" />
|
||||||
|
<path
|
||||||
|
style="fill:none;fill-opacity:1;stroke:#181716;stroke-width:0.264583;stroke-linecap:round;stroke-linejoin:round;stroke-dasharray:none;stroke-dashoffset:0.799748;stroke-opacity:1"
|
||||||
|
d="M 7.6729163,3.4395832 C 7.173008,2.3048106 5.9878255,2.0921784 5.2916665,2.1166666"
|
||||||
|
id="path7-2"
|
||||||
|
sodipodi:nodetypes="cc" />
|
||||||
|
</g>
|
||||||
|
</svg>
|
||||||
|
After Width: | Height: | Size: 7.5 KiB |
25
.idea/inspectionProfiles/Project_Default.xml
Normal file
25
.idea/inspectionProfiles/Project_Default.xml
Normal file
@ -0,0 +1,25 @@
|
|||||||
|
<component name="InspectionProjectProfileManager">
|
||||||
|
<profile version="1.0">
|
||||||
|
<option name="myName" value="Project Default" />
|
||||||
|
<inspection_tool class="PyCompatibilityInspection" enabled="true" level="WARNING" enabled_by_default="true">
|
||||||
|
<option name="ourVersions">
|
||||||
|
<value>
|
||||||
|
<list size="3">
|
||||||
|
<item index="0" class="java.lang.String" itemvalue="3.13" />
|
||||||
|
<item index="1" class="java.lang.String" itemvalue="3.10" />
|
||||||
|
<item index="2" class="java.lang.String" itemvalue="3.8" />
|
||||||
|
</list>
|
||||||
|
</value>
|
||||||
|
</option>
|
||||||
|
</inspection_tool>
|
||||||
|
<inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
|
||||||
|
<option name="ignoredPackages">
|
||||||
|
<value>
|
||||||
|
<list size="1">
|
||||||
|
<item index="0" class="java.lang.String" itemvalue="PyYAML" />
|
||||||
|
</list>
|
||||||
|
</value>
|
||||||
|
</option>
|
||||||
|
</inspection_tool>
|
||||||
|
</profile>
|
||||||
|
</component>
|
||||||
7
.idea/inspectionProfiles/profiles_settings.xml
Normal file
7
.idea/inspectionProfiles/profiles_settings.xml
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
<component name="InspectionProjectProfileManager">
|
||||||
|
<settings>
|
||||||
|
<option name="PROJECT_PROFILE" value="Default" />
|
||||||
|
<option name="USE_PROJECT_PROFILE" value="false" />
|
||||||
|
<version value="1.0" />
|
||||||
|
</settings>
|
||||||
|
</component>
|
||||||
7
.idea/misc.xml
Normal file
7
.idea/misc.xml
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="Black">
|
||||||
|
<option name="sdkName" value="Python 3.10" />
|
||||||
|
</component>
|
||||||
|
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.10" project-jdk-type="Python SDK" />
|
||||||
|
</project>
|
||||||
8
.idea/modules.xml
Normal file
8
.idea/modules.xml
Normal file
@ -0,0 +1,8 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="ProjectModuleManager">
|
||||||
|
<modules>
|
||||||
|
<module fileurl="file://$PROJECT_DIR$/.idea/beebot-docker.iml" filepath="$PROJECT_DIR$/.idea/beebot-docker.iml" />
|
||||||
|
</modules>
|
||||||
|
</component>
|
||||||
|
</project>
|
||||||
7
.idea/sqldialects.xml
Normal file
7
.idea/sqldialects.xml
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="SqlDialectMappings">
|
||||||
|
<file url="file://$PROJECT_DIR$/src/beebot.py" dialect="GenericSQL" />
|
||||||
|
<file url="PROJECT" dialect="SQLite" />
|
||||||
|
</component>
|
||||||
|
</project>
|
||||||
6
.idea/vcs.xml
Normal file
6
.idea/vcs.xml
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="VcsDirectoryMappings">
|
||||||
|
<mapping directory="$PROJECT_DIR$" vcs="Git" />
|
||||||
|
</component>
|
||||||
|
</project>
|
||||||
18
Dockerfile
Normal file
18
Dockerfile
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
FROM python:3.10
|
||||||
|
LABEL authors="Lord Baryhobal"
|
||||||
|
LABEL maintainer="Lord Baryhobal <lordbaryhobal@gmail.com>"
|
||||||
|
|
||||||
|
RUN echo "Installing Typst" \
|
||||||
|
&& wget -q -O /tmp/typst.tar.xz https://github.com/typst/typst/releases/download/v0.11.1/typst-x86_64-unknown-linux-musl.tar.xz \
|
||||||
|
&& tar -x /tmp/typst.tar.xz -C /tmp/ \
|
||||||
|
&& mv /tmp/typst-x86_64-unknown-linux-musl/typst /usr/bin/typst \
|
||||||
|
&& chmod +x /usr/bin/typst \
|
||||||
|
&& rm -r /tmp/typst-x86_64-unknown-linux-musl \
|
||||||
|
&& rm /tmp/typst.tar.xz
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
COPY src /app
|
||||||
|
|
||||||
|
RUN pip install -r requirements.txt
|
||||||
|
|
||||||
|
ENTRYPOINT ["python", "beebot.py"]
|
||||||
263
src/beebot.py
Normal file
263
src/beebot.py
Normal file
@ -0,0 +1,263 @@
|
|||||||
|
import asyncio
|
||||||
|
import datetime
|
||||||
|
import json
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
import sqlite3
|
||||||
|
import subprocess
|
||||||
|
from sqlite3 import Connection
|
||||||
|
from typing import Optional
|
||||||
|
|
||||||
|
import requests
|
||||||
|
import telegram.constants
|
||||||
|
from bs4 import BeautifulSoup
|
||||||
|
from telegram import Update
|
||||||
|
from telegram.ext import ApplicationBuilder, CommandHandler, ContextTypes
|
||||||
|
|
||||||
|
logging.basicConfig(
|
||||||
|
format="[%(asctime)s] %(name)s/%(levelname)s - %(message)s",
|
||||||
|
level=logging.INFO
|
||||||
|
)
|
||||||
|
logging.getLogger("httpx").setLevel(logging.WARNING)
|
||||||
|
logger = logging.getLogger(__name__)
|
||||||
|
logger.setLevel(logging.DEBUG)
|
||||||
|
|
||||||
|
class BeeBot:
|
||||||
|
MENU_URL = "https://www.epfl.ch/campus/restaurants-shops-hotels/fr/industrie21-epfl-valais-wallis/?date={date}"
|
||||||
|
RESTO_ID = 545
|
||||||
|
BASE_DIR = os.path.dirname(__file__)
|
||||||
|
CACHE_PATH = os.path.join(BASE_DIR, "cache.json")
|
||||||
|
DB_PATH = os.path.join(BASE_DIR, "database.db")
|
||||||
|
TEMPLATE_PATH = os.path.join(BASE_DIR, "menu.typ")
|
||||||
|
IMG_DIR = "/tmp/menu_images"
|
||||||
|
|
||||||
|
def __init__(self, token: str):
|
||||||
|
self.tg_token: str = token
|
||||||
|
self.tg_app = None
|
||||||
|
self.cache: dict[str, str] = {}
|
||||||
|
self.db_con: Optional[Connection] = None
|
||||||
|
self.locks: dict[str, bool] = {}
|
||||||
|
self.fetch_lock = asyncio.Lock()
|
||||||
|
self.gen_lock = asyncio.Lock()
|
||||||
|
|
||||||
|
def mainloop(self):
|
||||||
|
logger.info("Initialization")
|
||||||
|
|
||||||
|
self.db_con = sqlite3.connect(self.DB_PATH)
|
||||||
|
self.check_database()
|
||||||
|
self.load_cache()
|
||||||
|
|
||||||
|
app = ApplicationBuilder().token(self.tg_token).build()
|
||||||
|
app.add_handler(CommandHandler("week", self.cmd_week))
|
||||||
|
app.add_handler(CommandHandler("today", self.cmd_today))
|
||||||
|
|
||||||
|
logger.info("Starting bot")
|
||||||
|
app.run_polling()
|
||||||
|
|
||||||
|
async def cmd_week(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||||
|
logger.debug("Received /week")
|
||||||
|
await self.request_menu(update, context, False)
|
||||||
|
|
||||||
|
async def cmd_today(self, update: Update, context: ContextTypes.DEFAULT_TYPE) -> None:
|
||||||
|
logger.debug("Received /today")
|
||||||
|
await self.request_menu(update, context, True)
|
||||||
|
|
||||||
|
async def request_menu(self, update: Update, context: ContextTypes.DEFAULT_TYPE, today_only: bool) -> None:
|
||||||
|
chat_id = update.effective_chat.id
|
||||||
|
await context.bot.send_chat_action(chat_id=chat_id, action=telegram.constants.ChatAction.TYPING)
|
||||||
|
prefs = self.get_user_pref(update)
|
||||||
|
logger.debug(f"User prefs: {prefs}")
|
||||||
|
categories = prefs["categories"]
|
||||||
|
|
||||||
|
img_id = self.get_img_id(today_only, categories)
|
||||||
|
filename = img_id + ".png"
|
||||||
|
img_path = os.path.join(self.IMG_DIR, filename)
|
||||||
|
|
||||||
|
menu_id = "today_menu" if today_only else "week_menu"
|
||||||
|
menu_path = os.path.join(self.BASE_DIR, "menus_today.json" if today_only else "menus_week.json")
|
||||||
|
|
||||||
|
# If menu needs to be fetched
|
||||||
|
if not os.path.exists(menu_path) or self.is_outdated(menu_id, today_only):
|
||||||
|
# Notify user
|
||||||
|
msg = await update.message.reply_text("The menu is being updated, please wait...")
|
||||||
|
async with self.fetch_lock:
|
||||||
|
if today_only:
|
||||||
|
await self.fetch_today_menu()
|
||||||
|
else:
|
||||||
|
await self.fetch_week_menu()
|
||||||
|
await msg.delete()
|
||||||
|
|
||||||
|
# If image needs to be (re)generated
|
||||||
|
if not os.path.exists(img_path) or self.is_outdated(img_id, today_only):
|
||||||
|
await self.gen_image(today_only, categories, img_path, img_id)
|
||||||
|
|
||||||
|
await context.bot.send_photo(chat_id=chat_id, photo=img_path)
|
||||||
|
|
||||||
|
def check_database(self):
|
||||||
|
cur = self.db_con.cursor()
|
||||||
|
cur.execute("""
|
||||||
|
CREATE TABLE IF NOT EXISTS "user" (
|
||||||
|
"id" INTEGER,
|
||||||
|
"telegram_id" INTEGER NOT NULL UNIQUE,
|
||||||
|
"categories" TEXT,
|
||||||
|
"created_at" TEXT NOT NULL DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
PRIMARY KEY("id" AUTOINCREMENT)
|
||||||
|
);
|
||||||
|
""")
|
||||||
|
self.db_con.commit()
|
||||||
|
cur.close()
|
||||||
|
|
||||||
|
def get_user_pref(self, update: Update) -> dict:
|
||||||
|
user_id = update.effective_user.id
|
||||||
|
cur = self.db_con.cursor()
|
||||||
|
res = cur.execute("SELECT categories FROM user WHERE telegram_id=?", (user_id,))
|
||||||
|
user = res.fetchone()
|
||||||
|
cur.close()
|
||||||
|
|
||||||
|
if user is None:
|
||||||
|
self.create_user(user_id)
|
||||||
|
return {
|
||||||
|
"categories": {"E", "D", "C", "V"}
|
||||||
|
}
|
||||||
|
|
||||||
|
categories = set(user[0].split(","))
|
||||||
|
return {
|
||||||
|
"categories": categories
|
||||||
|
}
|
||||||
|
|
||||||
|
def create_user(self, telegram_id: int) -> None:
|
||||||
|
logger.debug(f"New user with id {telegram_id}")
|
||||||
|
cur = self.db_con.cursor()
|
||||||
|
cur.execute(
|
||||||
|
"INSERT INTO user (telegram_id, categories) VALUES (?, ?)",
|
||||||
|
(telegram_id, "E,D,C,V")
|
||||||
|
)
|
||||||
|
self.db_con.commit()
|
||||||
|
cur.close()
|
||||||
|
|
||||||
|
def load_cache(self):
|
||||||
|
if os.path.exists(self.CACHE_PATH):
|
||||||
|
with open(self.CACHE_PATH, "r") as f:
|
||||||
|
self.cache = json.load(f)
|
||||||
|
|
||||||
|
def save_cache(self):
|
||||||
|
with open(self.CACHE_PATH, "w") as f:
|
||||||
|
json.dump(self.cache, f)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def get_img_id(today_only: bool, categories: set[str]) -> str:
|
||||||
|
categs = "".join(sorted(categories))
|
||||||
|
layout = "today" if today_only else "week"
|
||||||
|
return f"{layout}-{categs}"
|
||||||
|
|
||||||
|
def is_outdated(self, obj_id: str, today_only: bool) -> bool:
|
||||||
|
if obj_id not in self.cache:
|
||||||
|
return True
|
||||||
|
gen_date = datetime.datetime.strptime(self.cache[obj_id], "%Y-%m-%d")
|
||||||
|
today = datetime.datetime.today()
|
||||||
|
if today_only:
|
||||||
|
if (today - gen_date).days > 0:
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
delta1 = datetime.timedelta(days=gen_date.weekday())
|
||||||
|
delta2 = datetime.timedelta(days=today.weekday())
|
||||||
|
gen_monday = gen_date - delta1
|
||||||
|
cur_monday = today - delta2
|
||||||
|
if (cur_monday - gen_monday).days > 0:
|
||||||
|
return True
|
||||||
|
return False
|
||||||
|
|
||||||
|
async def gen_image(self, today_only: bool, categories: set[str], outpath: str, img_id: str) -> None:
|
||||||
|
if not os.path.exists(self.IMG_DIR):
|
||||||
|
os.mkdir(self.IMG_DIR)
|
||||||
|
|
||||||
|
logger.info(f"Generating {'today' if today_only else 'week'} menu image for categories {categories}")
|
||||||
|
menu_path = "menus_today.json" if today_only else "menus_week.json"
|
||||||
|
subprocess.call([
|
||||||
|
"typst",
|
||||||
|
"compile",
|
||||||
|
"--root", self.BASE_DIR,
|
||||||
|
"--font-path", os.path.join(self.BASE_DIR, "fonts"),
|
||||||
|
"--input", f"path={menu_path}",
|
||||||
|
"--input", f"categories={','.join(categories)}",
|
||||||
|
self.TEMPLATE_PATH,
|
||||||
|
outpath
|
||||||
|
])
|
||||||
|
|
||||||
|
self.cache[img_id] = datetime.datetime.today().strftime("%Y-%m-%d")
|
||||||
|
self.save_cache()
|
||||||
|
|
||||||
|
def save_menu(self, days: list, menu_id: str, filename: str) -> None:
|
||||||
|
with open(os.path.join(self.BASE_DIR, filename), "w") as f:
|
||||||
|
json.dump(days, f)
|
||||||
|
|
||||||
|
self.cache[menu_id] = datetime.datetime.today().strftime("%Y-%m-%d")
|
||||||
|
self.save_cache()
|
||||||
|
|
||||||
|
async def fetch_week_menu(self) -> None:
|
||||||
|
logger.info("Fetching week menu")
|
||||||
|
today = datetime.datetime.today()
|
||||||
|
delta = datetime.timedelta(days=today.weekday())
|
||||||
|
monday = today - delta
|
||||||
|
days = []
|
||||||
|
for i in range(5):
|
||||||
|
dt = datetime.timedelta(days=i)
|
||||||
|
date = monday + dt
|
||||||
|
menus = await self.fetch_menu(date)
|
||||||
|
days.append({
|
||||||
|
"date": date.strftime("%Y-%m-%d"),
|
||||||
|
"menus": menus
|
||||||
|
})
|
||||||
|
|
||||||
|
self.save_menu(days, "week_menu", "menus_week.json")
|
||||||
|
|
||||||
|
async def fetch_today_menu(self) -> None:
|
||||||
|
logger.info("Fetching today menu")
|
||||||
|
today = datetime.datetime.today()
|
||||||
|
menus = await self.fetch_menu(today)
|
||||||
|
days = [{
|
||||||
|
"date": today.strftime("%Y-%m-%d"),
|
||||||
|
"menus": menus
|
||||||
|
}]
|
||||||
|
|
||||||
|
self.save_menu(days, "today_menu", "menus_today.json")
|
||||||
|
|
||||||
|
async def fetch_menu(self, date: datetime.date) -> list:
|
||||||
|
url = self.MENU_URL.format(date=date.strftime("%Y-%m-%d"))
|
||||||
|
headers = {
|
||||||
|
"User-Agent": "BeeBot 1.0"
|
||||||
|
}
|
||||||
|
res = requests.get(url, headers=headers)
|
||||||
|
|
||||||
|
if res.status_code != 200:
|
||||||
|
return []
|
||||||
|
|
||||||
|
bs = BeautifulSoup(res.content, features="lxml")
|
||||||
|
table = bs.find("table", {"id": "menuTable"})
|
||||||
|
lines = table.find("tbody").findAll("tr", {"data-restoid": str(self.RESTO_ID)})
|
||||||
|
|
||||||
|
menus = []
|
||||||
|
|
||||||
|
for line in lines:
|
||||||
|
menu = line.find("td", class_="menu").find("div", class_="descr").contents[0].text
|
||||||
|
menu_lines = menu.split("\n")
|
||||||
|
|
||||||
|
price_cells = line.find("td", class_="prices").findAll("span", class_="price")
|
||||||
|
prices = {}
|
||||||
|
for cell in price_cells:
|
||||||
|
category = cell.find("abbr").text
|
||||||
|
price = cell.contents[-1].replace("CHF", "").strip()
|
||||||
|
prices[category] = float(price)
|
||||||
|
|
||||||
|
menus.append({
|
||||||
|
"name": menu_lines[0],
|
||||||
|
"extra": menu_lines[1:],
|
||||||
|
"prices": prices
|
||||||
|
})
|
||||||
|
|
||||||
|
return menus
|
||||||
|
|
||||||
|
if __name__ == '__main__':
|
||||||
|
logger.info("Welcome to BeeBot !")
|
||||||
|
bot = BeeBot(os.getenv("TELEGRAM_TOKEN"))
|
||||||
|
bot.mainloop()
|
||||||
154
src/menu.typ
Normal file
154
src/menu.typ
Normal file
@ -0,0 +1,154 @@
|
|||||||
|
#let path = sys.inputs.at("path", default: "menus.json")
|
||||||
|
#let categories = sys.inputs.at("categories", default: "E,D,C,V").split(",")
|
||||||
|
#let days = json(path)
|
||||||
|
#let nbsp = sym.space.nobreak
|
||||||
|
|
||||||
|
#let width = if days.len() > 1 {20cm} else {10cm}
|
||||||
|
|
||||||
|
#set page(margin: 0cm, height: auto, width: width)
|
||||||
|
|
||||||
|
#let parse-date(date-txt) = {
|
||||||
|
let (year, month, day) = date-txt.split("-")
|
||||||
|
return datetime(year: int(year), month: int(month), day: int(day))
|
||||||
|
}
|
||||||
|
|
||||||
|
#let fmt-number(
|
||||||
|
number,
|
||||||
|
decimals: 2,
|
||||||
|
decimal-sep: ".",
|
||||||
|
thousands-sep: "'"
|
||||||
|
) = {
|
||||||
|
let integer = calc.trunc(number)
|
||||||
|
let decimal = calc.fract(number)
|
||||||
|
let res = str(integer).clusters()
|
||||||
|
.rev()
|
||||||
|
.chunks(3)
|
||||||
|
.map(c => c.join(""))
|
||||||
|
.join(thousands-sep)
|
||||||
|
.rev()
|
||||||
|
|
||||||
|
if decimals != 0 {
|
||||||
|
res += decimal-sep
|
||||||
|
decimal = decimal * calc.pow(10, decimals)
|
||||||
|
decimal = calc.round(decimal)
|
||||||
|
decimal = str(decimal) + ("0" * decimals)
|
||||||
|
res += decimal.slice(0, decimals)
|
||||||
|
}
|
||||||
|
|
||||||
|
return res
|
||||||
|
}
|
||||||
|
|
||||||
|
#let day-names = (
|
||||||
|
"Lundi",
|
||||||
|
"Mardi",
|
||||||
|
"Mercredi",
|
||||||
|
"Jeudi",
|
||||||
|
"Vendredi",
|
||||||
|
"Samedi",
|
||||||
|
"Dimanche"
|
||||||
|
)
|
||||||
|
|
||||||
|
#let display-menus(
|
||||||
|
days,
|
||||||
|
categories,
|
||||||
|
day-sep-stroke: rgb("#82BC00") + 1pt,
|
||||||
|
title-underline-stroke: rgb("#e34243") + 2pt,
|
||||||
|
menu-sep-stroke: rgb("#ababab") + 1pt,
|
||||||
|
price-categ-color: rgb("#577F25")
|
||||||
|
) = {
|
||||||
|
set text(font: "Liberation Sans", hyphenate: true)
|
||||||
|
let cols = calc.min(2, days.len())
|
||||||
|
|
||||||
|
let cells = ()
|
||||||
|
|
||||||
|
for day in days {
|
||||||
|
let weekday = parse-date(day.date).weekday() - 1
|
||||||
|
let day-name = day-names.at(weekday)
|
||||||
|
let menus = ()
|
||||||
|
for menu in day.menus {
|
||||||
|
let price-cell = grid.cell(rowspan: 2)[]
|
||||||
|
let prices = menu.at(
|
||||||
|
"prices",
|
||||||
|
default: (:)
|
||||||
|
)
|
||||||
|
if menu.keys().contains("price") {
|
||||||
|
prices = ("": menu.price)
|
||||||
|
}
|
||||||
|
|
||||||
|
let pairs = prices.pairs()
|
||||||
|
.filter(p => p.first() in categories)
|
||||||
|
.sorted(key: p => p.last())
|
||||||
|
|
||||||
|
let prices = pairs.map(p => {
|
||||||
|
let categ = emph(
|
||||||
|
text(
|
||||||
|
fill: price-categ-color,
|
||||||
|
p.first()
|
||||||
|
)
|
||||||
|
)
|
||||||
|
let price = fmt-number(p.last())
|
||||||
|
categ + ":" + nbsp + "CHF" + nbsp + price
|
||||||
|
})
|
||||||
|
price-cell = grid.cell(
|
||||||
|
rowspan: menu.extra.len() + 1,
|
||||||
|
stack(dir: ttb, spacing: .4em, ..prices)
|
||||||
|
)
|
||||||
|
//let price = "CHF" + sym.space.nobreak + str(menu.price)
|
||||||
|
menus.push(
|
||||||
|
grid(
|
||||||
|
columns: (1fr, auto),
|
||||||
|
column-gutter: .4em,
|
||||||
|
row-gutter: .4em,
|
||||||
|
[*#menu.name*],
|
||||||
|
price-cell,
|
||||||
|
..menu.extra.map(l => text(10pt, l))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
let title = align(
|
||||||
|
center,
|
||||||
|
text(size: 14pt, weight: "bold", day-name)
|
||||||
|
)
|
||||||
|
let title-box = box(
|
||||||
|
width: 50%,
|
||||||
|
inset: .6em,
|
||||||
|
stroke: (bottom: title-underline-stroke),
|
||||||
|
title
|
||||||
|
)
|
||||||
|
let title-cell = grid.cell(align: center, title-box)
|
||||||
|
|
||||||
|
cells.push(box(width: 100%, grid(
|
||||||
|
columns: (100%),
|
||||||
|
inset: (x, y) => (
|
||||||
|
top: if y == 0 {0pt} else {.8em},
|
||||||
|
bottom: if y == 0 {0pt} else {2em}
|
||||||
|
),
|
||||||
|
stroke: (_, y) => (top: if y < 2 {none} else {menu-sep-stroke}),
|
||||||
|
title-cell,
|
||||||
|
..menus
|
||||||
|
)))
|
||||||
|
}
|
||||||
|
|
||||||
|
if calc.rem(cells.len(), 2) == 1 {
|
||||||
|
cells.last() = grid.cell(colspan: calc.min(2, cols), cells.last())
|
||||||
|
}
|
||||||
|
|
||||||
|
grid(
|
||||||
|
columns: (100% / cols,) * cols,
|
||||||
|
inset: (x: 1.5em, y: .8em),
|
||||||
|
stroke: (x, y) => {
|
||||||
|
let borders = (:)
|
||||||
|
if x != 0 {
|
||||||
|
borders.insert("left", day-sep-stroke)
|
||||||
|
}
|
||||||
|
if y != 0 {
|
||||||
|
borders.insert("top", day-sep-stroke)
|
||||||
|
}
|
||||||
|
return borders
|
||||||
|
},
|
||||||
|
..cells
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
#display-menus(days, categories)
|
||||||
4
src/requirements.txt
Normal file
4
src/requirements.txt
Normal file
@ -0,0 +1,4 @@
|
|||||||
|
requests==2.25.1
|
||||||
|
beautifulsoup4==4.10.0
|
||||||
|
lxml==4.8.0
|
||||||
|
python-telegram-bot==21.5
|
||||||
Loading…
Reference in New Issue
Block a user