From 263da6e5eb95da3ebe38eb24c31a92cb4798b5c5 Mon Sep 17 00:00:00 2001
From: ynerant <ynerant@zamokv5.crans.org>
Date: Tue, 11 May 2021 14:26:33 +0200
Subject: [PATCH] Ajout d'un fichier de configuration

Signed-off-by: ynerant <ynerant@zamokv5.crans.org>
---
 .gitignore         |  1 +
 chronodose.py      | 75 ++++++++++++++++++++++++++++++++++++----------
 config.yml.example | 22 ++++++++++++++
 3 files changed, 82 insertions(+), 16 deletions(-)
 create mode 100644 config.yml.example

diff --git a/.gitignore b/.gitignore
index bee8a64..2f9121e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1 +1,2 @@
 __pycache__
+config.yml
diff --git a/chronodose.py b/chronodose.py
index 3c056c0..50144da 100755
--- a/chronodose.py
+++ b/chronodose.py
@@ -3,13 +3,50 @@
 from dataclasses import dataclass
 from datetime import date
 from math import acos, cos, pi, sin
+import os
 import requests
+import sys
 from threading import Thread
 from time import sleep
+import yaml
 
 from irc import IRCClient
 
 
+@dataclass
+class Config:
+    @dataclass
+    class IRCConfig:
+        nickname: str = ""
+        host: str = ""
+        channel: str = ""
+
+
+    @dataclass
+    class SearchConfig:
+        position: "Location" = None
+        radius: float = 0
+        departements: list[int] = None
+        mentions: list[str] = None
+
+
+    irc: "IRCConfig" = None
+    search: list["SearchConfig"] = None
+
+    def __init__(self, irc: "IRCConfig", search: list["SearchConfig"]):
+        if isinstance(irc, dict):
+            irc = Config.IRCConfig(**irc)
+        if not search:
+            search = []
+        search = [Config.SearchConfig(**sc) if isinstance(sc, dict) else sc for sc in search]
+        for s in search:
+            if isinstance(s.position, dict):
+                s.position = Location(**s.position)
+
+        self.irc = irc
+        self.search = search
+
+
 @dataclass
 class Location:
     longitude: float = 0.0
@@ -87,36 +124,41 @@ def check_dpt(dpt_number: int, position: Location, radius: int = 20):
 
 
 def main():
-    gif = Location(latitude=48.7090418, longitude=2.1648068, city="Gif-sur-Yvette")
-    lyon = Location(latitude=45.7579502, longitude=4.8001017, city="Lyon")
-    chambéry = Location(latitude=45.5822142, longitude=5.8713341, city="Chambéry")
-    nantes = Location(latitude=47.2382007, longitude=-1.6300954, city="Nantes")
-    marseille = Location(latitude=43.2803692, longitude=5.3104571, city="Marseille")
+    if not os.path.isfile('config.yml'):
+        print("Le fichier de configuration n'existe pas. "
+              "Commencez par copier l'exemple depuis config.yml.example.", file=sys.stderr)
+        exit(1)
 
-    irc_client = IRCClient('irc.crans.org', 'chronodose')
+    # Chargement de la configuration
+    with open('config.yml') as f:
+        config = yaml.safe_load(f)
+    config = Config(**config)
+
+    irc_client = IRCClient(config.irc.host, config.irc.nickname)
     Thread(target=irc_client.start).start()
+    # Connexion à IRC
     sleep(10)
-    irc_client.join('#chronodose')
-    irc_client.privmsg('#chronodose', 'coucou')
+    irc_client.join(config.irc.channel)
+    irc_client.privmsg(config.irc.channel, 'coucou')
 
     already_indicated = []
 
     def msg(*mesg: str) -> None:
         # Afficher un message dans la console et sur IRC
         print(*mesg)
-        irc_client.privmsg('#chronodose', ' '.join(str(a) for a in mesg))
+        irc_client.privmsg(config.irc.channel, ' '.join(str(a) for a in mesg))
 
     while True:
-        places = {}
-        for dpt, ville in [(91, gif), (92, gif), (94, gif), (78, gif), (69, lyon),
-                (73, chambéry), (44, nantes), (13, marseille)]:
-            places[dpt] = check_dpt(dpt, ville)
+        for search in config.search:
+            places = []
+            for dpt in search.departements:
+                places.extend(check_dpt(dpt, search.position, search.radius))
 
-        for dpt, places in places.items():
             if not places:
-                print("Pas de dose disponible dans le", dpt)
+                print("Pas de place disponible autour de", search.position.city)
                 continue
-            print(sum(place[1] for place in places), "doses disponibles dans le", dpt)
+
+            print(sum(place[1] for place in places), "doses disponibles autour de", search.position.city)
             for centre, count in places:
                 if (centre.internal_id, date.today()) in already_indicated:
                     # Message déjà envoyé, on spam pas
@@ -127,6 +169,7 @@ def main():
                 msg("Type de vaccin :", ", ".join(centre.vaccine_type))
                 msg(centre.metadata.address, centre.metadata.phone_number)
                 msg("Réserver sur", centre.url)
+                msg(*search.mentions)
                 msg(" ")
 
         # 5 minutes
diff --git a/config.yml.example b/config.yml.example
new file mode 100644
index 0000000..d7f0c7f
--- /dev/null
+++ b/config.yml.example
@@ -0,0 +1,22 @@
+---
+irc:
+  nickname: chronodose
+  host: irc.crans.org
+  channel: '#chronodose'
+
+search:
+  - position:
+      longitude: 2.2770207
+      latitude: 48.8588377
+       city: Paris
+    radius: 15
+    departements:
+      - 75
+      - 77
+      - 78
+      - 92
+      - 93
+      - 94
+      - 95
+    mentions:
+      - toto