Implement methods for the visits counter.

Antoine Goutenoir
1 parent e9a989e0
Showing 1 changed file with 33 additions and 5 deletions Show diff stats
flaskr/core.py
 import abc
 import importlib
+from os.path import isfile
 from datetime import datetime
 from uuid import uuid4
 from .content import content
+hit_count_path = "../VISITS"
+
+
 def generate_unique_id():
     """
     :return: a unique identifier that can be sorted chronologically.
@@ -32,8 +36,32 @@ def get_emission_models():
 models = get_emission_models()
-# unused
-class FootprintEstimatorDriver(abc.ABCMeta):
-    @abc.abstractmethod
-    def get_travel_footprint(self, from_location, to_location):  # TBD
-        pass
+def get_hit_counter():
+    hit_count = 1
+    if isfile(hit_count_path):
+        with open(hit_count_path) as hcf:
+            hit_count = int(hcf.read().strip())
+
+    return hit_count
+
+
+def increment_hit_counter():
+    if isfile(hit_count_path):
+        hit_count = int(open(hit_count_path).read())
+        hit_count += 1
+    else:
+        hit_count = 1
+
+    hit_counter_file = open(hit_count_path, 'w')
+    hit_counter_file.write(str(hit_count))
+    hit_counter_file.close()
+
+    return hit_count
+
+
+
+# # unused
+# class FootprintEstimatorDriver(abc.ABCMeta):
+#     @abc.abstractmethod
+#     def get_travel_footprint(self, from_location, to_location):  # TBD
+#         pass