rewrite settings module

.vscode/launch.json

@@ -1,5 +1,12 @@
 {
     "configurations": [
+        {
+            "name": "Debug Mock",
+            "type": "python",
+            "request": "launch",
+            "module": "mock.run",
+            "justMyCode": true
+        },
         {
             "name": "Python: Flask",
             "type": "python",

mock/mock.py

@@ -76,8 +76,8 @@ def mock():
             print(intent.id)
 
             PaymentService.handle_stripe_payment(intent)
-        except Exception as e:
-            print(str(e))
+        except Exception as err:
+            print(err)
 
         # reservation_2 = ReservationService.create_reservation(
         #     user,

settings/model.py

@@ -2,13 +2,13 @@
     settings model
 """
 from database import Base
-from sqlalchemy import Column, Float
+from sqlalchemy import Column, Float, Integer
 
 
 class Setting(Base):
     __tablename__ = 'settings'
 
-    id = Column(Float, primary_key=True)
+    id = Column(Integer, primary_key=True)
     charge_within_hour = Column(Float)
     charge_more_than_a_hour = Column(Float)
     charge_more_than_a_day = Column(Float)

settings/service.py

@@ -2,6 +2,7 @@
     settings service
 """
 from database.database import db_session
+from werkzeug.exceptions import NotFound
 
 from settings.model import Setting
 
@@ -10,15 +11,16 @@ class SettingService:
     @staticmethod
     def set_settings(settings: Setting):
         try:
-            original = SettingService.get_settings()
-
-            if original is None:
-                db_session.add(settings)
-            else:
-                original.charge_within_hour = settings.charge_within_hour
-                original.charge_more_than_a_hour = settings.charge_more_than_a_hour
-                original.charge_more_than_a_day = settings.charge_more_than_a_day
-
+            get_settings = SettingService.get_settings()
+            Setting.query.filter(Setting.id == get_settings.id).update({
+                "charge_within_hour": settings.charge_within_hour,
+                "charge_more_than_a_hour": settings.charge_more_than_a_hour,
+                "charge_more_than_a_day": settings.charge_more_than_a_day
+            })
+            db_session.commit()
+            return settings
+        except NotFound as err:
+            db_session.add(settings)
             db_session.commit()
             return settings
         except Exception as err:
@@ -27,4 +29,7 @@ def set_settings(settings: Setting):
 
     @staticmethod
     def get_settings() -> Setting:
-        return Setting.query.first()
+        settings = Setting.query.first()
+        if settings is None:
+            raise NotFound("Settings not found!")
+        return settings