basic features

1 files changed, 34 insertions, 0 deletions

diff --git a/dbstore/dbm_store.py b/dbstore/dbm_store.py
new file mode 100644
index 0000000..fedb505
--- /dev/null
+++ b/dbstore/dbm_store.py
@@ -0,0 +1,34 @@
+import dbm
+
+db = None
+store_file = "fsq_poi.db"
+
+
+def get_loc(fsq_id):
+    global db
+    if db is None:
+        db = dbm.open(store_file, 'c')
+    if fsq_id in db:
+        res = db[fsq_id].decode("utf-8").split("|")
+        return {
+            "name": res[0],
+            "locality": res[1],
+            "region": res[2],
+            "latitude": res[3],
+            "longitude": res[4],
+            "osm_url": res[5],
+        }
+    else:
+        return None
+
+
+def store_loc(loc):
+    global db
+    if db is None:
+        db = dbm.open(store_file, 'c')
+    db[loc["fsq_id"]] = "{}|{}|{}|{}|{}|{}".format(loc["name"],
+                                                   loc["locality"],
+                                                   loc["region"],
+                                                   loc["latitude"],
+                                                   loc["longitude"],
+                                                   loc["osm_url"])