Create and modify drinks function.

2018-03-14 14:08:19 +01:00
parent ec71d4415f
commit ac01a4441b
4 changed files with 22 additions and 10 deletions
--- a/dsst/dsst_server/func_write.py
+++ b/dsst/dsst_server/func_write.py
@@ -24,6 +24,14 @@ class WriteFunctions:
                              sql.Player.hex_id: player.hex_id})
         .execute())

+    @staticmethod
+    def update_drink(drink: 'models.Drink', *_):
+        (sql.Drink
+         .insert(id=drink.id, name=drink.name, vol=drink.vol)
+         .on_conflict(update={sql.Drink.name: drink.name,
+                              sql.Drink.vol: drink.vol})
+         .execute())
+
    @staticmethod
    def update_season(season: 'models.Season', *_):
        (sql.Season