bot: change create_poi to create_or_update_poi

1 files changed, 3 insertions, 4 deletions

diff --git a/dbstore/peewee_store.py b/dbstore/peewee_store.py
index 00897d7..89e4b4b 100644
--- a/dbstore/peewee_store.py
+++ b/dbstore/peewee_store.py
@@ -52,14 +52,13 @@ def get_poi_by_fsq_id(fsq_id) -> dict:
             return {}
 
 
-def create_poi(poi: dict):
+def create_or_update_poi(poi: dict) -> int:
     with db.connection_context():
-        poi = Location.create(
+        return Location.insert(
             fsq_id=poi["fsq_id"],
             name=poi["name"],
             locality=poi["locality"],
             region=poi["region"],
             latitude=poi["latitude"],
             longitude=poi["longitude"],
-        )
-        poi.save()
+        ).on_conflict_replace().execute()