Task 07

OluwaninsolaAO · Apr 22, 2023 · c02c132 · c02c132
1 parent cd20e11
commit c02c132
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/models/city.py b/models/city.py
@@ -8,8 +8,15 @@
 
 class City(BaseModel, Base):
     """ The city class, contains state ID and name """
-    __tablename__ = 'cities'
-    name = Column(String(128), nullable=False)
-    state_id = Column(String(60), ForeignKey('states.id'), nullable=False)
+    if os.environ.get('HBNB_TYPE_STORAGE') == 'db':
+        __tablename__ = 'cities'
+        name = Column(String(128), nullable=False)
+        state_id = Column(String(60), ForeignKey('states.id'),
+                          nullable=False)
 
-    places = relationship('Place', cascade='all, delete', backref='cities')
+        places = relationship('Place', cascade='all, delete',
+                              backref='cities')
+
+    else:
+        name = ""
+        state_id = ""