first commit

models.py

@@ -0,0 +1,51 @@
+# models.py
+from sqlalchemy import Column, Integer, String, Float, Date, Boolean, BigInteger, ForeignKey, DateTime # Added DateTime
+from sqlalchemy.orm import relationship
+from datetime import date, datetime # Import datetime as well
+
+# Import Base from our database.py
+from database import Base
+
+# --- OilPrice Model ---
+class OilPrice(Base):
+    __tablename__ = "oil_prices"
+
+    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
+    state = Column(String(100), index=True)
+    zone = Column(Integer, index=True)
+    name = Column(String(255), index=True)
+    price = Column(Float, nullable=True)
+    date = Column(String(20)) # This is the 'Date Posted' from the website
+
+    # --- UPDATED scrapetimestamp ---
+    # To store both date and time of scraping
+    scrapetimestamp = Column(DateTime, default=datetime.utcnow, index=True)
+    # 'datetime.utcnow' will automatically provide the current UTC date and time
+    # when a new record is created and this field is not explicitly set.
+
+    company_id = Column(Integer, ForeignKey("company.id"), nullable=True)
+
+    def __repr__(self):
+        return (f"<OilPrice(id={self.id}, state='{self.state}', zone='{self.zone}', "
+                f"name='{self.name}', price={self.price}, date='{self.date}', "
+                f"scraped_at='{self.scrapetimestamp}')>") # Added scraped_at to repr
+
+# --- Company Model (remains the same) ---
+class Company(Base):
+    __tablename__ = "company"
+    # ... (fields as before) ...
+    id = Column(Integer, primary_key=True, index=True, autoincrement=True)
+    active = Column(Boolean, nullable=False, default=True)
+    created = Column(Date, nullable=False, default=date.today) # This 'created' is for the company record
+    name = Column(String(255), nullable=False, index=True, unique=True)
+    address = Column(String(500), nullable=True)
+    town = Column(String(100), nullable=True)
+    state = Column(String(50), nullable=True)
+    phone = Column(String(20), nullable=True)
+    owner_name = Column(String(255), nullable=True)
+    owner_phone_number = Column(String(20), nullable=True)
+    email = Column(String(255), nullable=True, unique=True)
+    user_id = Column(Integer, ForeignKey("users.id"), nullable=True, index=True)
+
+    def __repr__(self):
+        return f"<Company(id={self.id}, name='{self.name}', active={self.active})>"