You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
139 lines
4.0 KiB
Python
139 lines
4.0 KiB
Python
import os
|
|
import time
|
|
|
|
import toml
|
|
from peewee import SqliteDatabase
|
|
|
|
import db_modell as db
|
|
from telegram_api import telegram_bot_api as api
|
|
|
|
|
|
def load_config():
|
|
configfile = os.path.join(SKRIPTPFAD, "config_fotobot.toml")
|
|
with open(configfile) as conffile:
|
|
config = toml.loads(conffile.read())
|
|
return config
|
|
|
|
|
|
SKRIPTPFAD = os.path.abspath(os.path.dirname(__file__))
|
|
CONFIG = load_config()
|
|
# Nur für Testzwecke!
|
|
CONFIG["root_verzeichnis"] = SKRIPTPFAD
|
|
|
|
db.DB_PROXY.initialize(SqliteDatabase(os.path.join(SKRIPTPFAD, "datenbank.db3")))
|
|
|
|
|
|
class FotoBot(api.Bot):
|
|
def __init__(self, token):
|
|
super().__init__(token)
|
|
|
|
self.menue = None
|
|
self.umenue = None
|
|
|
|
self.bot_kommando = {
|
|
|
|
}
|
|
|
|
def message_handler(self, message):
|
|
if self.menue is not None:
|
|
self.menue(message)
|
|
return
|
|
|
|
def bot_kommando_exec(self, kommando):
|
|
self.menue = self.bot_kommando.get(kommando, self.kommando_unbekannt)
|
|
self.menue()
|
|
|
|
def kommando_unbekannt(self, message):
|
|
print("Unbekannt")
|
|
|
|
|
|
def verzeichnis_erstellen(ordner):
|
|
verzeichnis = os.path.join(CONFIG["root_verzeichnis"], *ordner)
|
|
os.makedirs(verzeichnis, mode=0o660, exist_ok=True)
|
|
return verzeichnis
|
|
|
|
|
|
def get_biggest_size(photos):
|
|
max_file_size = 0
|
|
max_index = None
|
|
for index, photo in enumerate(photos):
|
|
if photo["file_size"] > max_file_size:
|
|
max_index = index
|
|
return max_index
|
|
|
|
|
|
def datei_speichern(verzeichnis, file_name, inhalt):
|
|
file = os.path.join(verzeichnis, file_name)
|
|
with open(file, "wb") as file_:
|
|
file_.write(inhalt)
|
|
|
|
|
|
def caption_analysieren(caption):
|
|
caption_split = caption.split(";")
|
|
ordner = caption_split[0].split(",")
|
|
ordner = [ordner_einzel.replace(" ", "") for ordner_einzel in ordner]
|
|
try:
|
|
notizen = caption_split[1]
|
|
if notizen.startswith(" "):
|
|
notizen = notizen.replace(" ", "", 1)
|
|
except IndexError:
|
|
notizen = []
|
|
return ordner, notizen
|
|
|
|
|
|
def get_dateiendung(file_path):
|
|
return file_path.split(".")[-1]
|
|
|
|
|
|
def nachricht_auswerten(bot, message):
|
|
absender = message["message"]["from"].get("id", 0)
|
|
if absender not in CONFIG["erlaubte_telegram_ids"]:
|
|
bot.send_message(absender, "User nicht freigegeben")
|
|
return
|
|
if bot.menue is None:
|
|
if "entities" in message["message"]:
|
|
if "bot_command" in message["message"]["entities"][0]["type"]:
|
|
text = message["message"]["text"]
|
|
kommando = text.replace("/", "")
|
|
bot.bot_kommando_exec(kommando)
|
|
|
|
if "photo" in message["message"]:
|
|
if "caption" in message["message"]:
|
|
ordner, notiz = caption_analysieren(message["message"]["caption"])
|
|
ordner_db_id = db.ordner_struktur_erstellen(ordner)
|
|
verzeichnis = verzeichnis_erstellen(ordner)
|
|
|
|
max_index = get_biggest_size(message["message"]["photo"])
|
|
file, file_path = bot.get_file(message["message"]["photo"][max_index]["file_id"])
|
|
dateiendung = get_dateiendung(file_path)
|
|
file_name = db.file_speichern(ordner_db_id, notiz, dateiendung)
|
|
datei_speichern(verzeichnis, file_name, file)
|
|
|
|
else:
|
|
bot.send_message(chat_id=absender,
|
|
text="Bild kann nicht verabeitet werden, Ordner wurde nicht angegeben",
|
|
reply_to_message_id=message["message"]["message_id"])
|
|
|
|
else:
|
|
bot.menue()
|
|
|
|
|
|
def main():
|
|
bot = FotoBot(CONFIG["token"])
|
|
db.db_create_table()
|
|
db.update_root_verzeichnis(CONFIG["root_verzeichnis"])
|
|
|
|
while True:
|
|
nachricht_vorhanden = False
|
|
for message in bot.get_updates():
|
|
nachricht_vorhanden = True
|
|
nachricht_auswerten(bot, message)
|
|
|
|
if not nachricht_vorhanden and bot.menue is None:
|
|
break
|
|
time.sleep(0.1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|