models.py 700 B

1234567891011121314151617181920212223
  1. from sqlalchemy import Column, Integer, String, DateTime
  2. from datetime import datetime
  3. from sqlalchemy.ext.declarative import declarative_base
  4. Base = declarative_base()
  5. class Company(Base):
  6. __tablename__ = "companies"
  7. id = Column(Integer, primary_key=True, index=True)
  8. hubspot_id = Column(String, unique=True, index=True)
  9. name = Column(String, nullable=False)
  10. domain = Column(String, nullable=True)
  11. class Token(Base):
  12. __tablename__ = "tokens"
  13. id = Column(Integer, primary_key=True, index=True)
  14. access_token = Column(String, nullable=False)
  15. refresh_token = Column(String, nullable=True)
  16. expires_at = Column(DateTime, nullable=False) # Store expiry time