Save prices to database (sqlite)

fioul.py

@@ -1,10 +1,13 @@
 #! /usr/bin/env python3
 
 # from bottle import hook, request, route, run, static_file, view
+
+from datetime import date
+from models import Price, IntegrityError
+
 import requests
 import json
 
-
 query = """query {
   getProductGroups(category: 1, zipcode: 1184) {
     name
@@ -28,8 +31,21 @@ def get_price():
     return product["default_price_tax"]
 
 
+def store_current_price():
+    today = date.today()
+    # Get price returns a float (price per L)
+    # We store an int (price per m^3)
+    price = int(get_price() * 1000)
+    try:
+        Price.create(date=today, price=price)
+    except IntegrityError:
+        print("Price for today already in database, not writing")
+    return price
+
+
 def main():
     print(get_price())
+    store_current_price()
 
 
 if __name__ == "__main__":