Upload New File

backup/models/location.py

+from database.connect import connection, base
+import sqlalchemy as db
+from sqlalchemy.orm import relationship
+from datetime import date
+
+class LocationModel(base):
+    __tablename__ = "location"
+    id = db.Column(db.Integer, primary_key=True)
+    name = db.Column(db.String(20), nullable=False)
+
+    def __init__(self, name: str):
+        self.name = name
+
+    def __repr__(self):
+         return "<Location %r>" % self.id
+
+class Location:
+    def __init__(self):
+        # Get connection to database as session
+        engine, base, session, meta = connection()
+        self.session = session
+
+    def get_locations(self) -> list:
+        """
+        Get locations from database
+
+        Returns:
+            list: List of all locations
+        """
+        locations = self.session.query(LocationModel).all()
+        return locations
\ No newline at end of file