Task 077

OluwaninsolaAO · Apr 22, 2023 · cd20e11 · cd20e11
1 parent 21b0b3e
commit cd20e11
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 12 deletions.
diff --git a/models/state.py b/models/state.py
@@ -9,9 +9,13 @@
 class State(BaseModel, Base):
     """ State class """
 
-    __tablename__ = 'states'
-    name = Column(String(128), nullable=False)
-    cities = relationship('City', cascade='all, delete', backref='state')
+    if os.environ.get('HBNB_TYPE_STORAGE') == 'db':
+        __tablename__ = 'states'
+        name = Column(String(128), nullable=False)
+        cities = relationship('City', cascade='all, delete', backref='state')
+    else:
+        name = ""
+        cities = []
 
     @property
     def cities(self):

diff --git a/models/user.py b/models/user.py
@@ -10,13 +10,22 @@
 class User(BaseModel, Base):
     """This class defines a user by various attributes"""
 
-    __tablename__ = 'users'
-    email = Column(String(128), nullable=False)
-    password = Column(String(128), nullable=False)
-    first_name = Column(String(128))
-    last_name = Column(String(128))
+    if os.environ.get('HBNB_TYPE_STORAGE') == 'db':
+        __tablename__ = 'users'
+        email = Column(String(128), nullable=False)
+        password = Column(String(128), nullable=False)
+        first_name = Column(String(128))
+        last_name = Column(String(128))
 
-    places = relationship('Place', cascade='all, delete',
-                          backref='user')
-    reviews = relationship('Review', cascade='all, delete',
-                           backref='user')
+        places = relationship('Place', cascade='all, delete',
+                              backref='user')
+        reviews = relationship('Review', cascade='all, delete',
+                               backref='user')
+    else:
+        email = ""
+        password = ""
+        first_name = ""
+        last_name = ""
+
+        places = []
+        reviews = []