166 lines
4.2 KiB
Python
166 lines
4.2 KiB
Python
#!/usr/bin/env python
|
|
# coding: utf-8
|
|
|
|
# In[ ]:
|
|
|
|
|
|
from sqlalchemy import create_engine, Column, Integer, String, Float, DateTime, ForeignKey
|
|
from sqlalchemy.orm import sessionmaker, relationship, declarative_base
|
|
from flask import Flask, render_template
|
|
|
|
# In[ ]:
|
|
|
|
|
|
# Create a base class for declarative models
|
|
Base = declarative_base()
|
|
|
|
class Product(Base):
|
|
__tablename__ = 'product'
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(String)
|
|
SKU_PLU_UPC_number = Column(String)
|
|
price_histories = relationship("PriceHistory", back_populates="product")
|
|
|
|
class Retailer(Base):
|
|
__tablename__ = 'retailer'
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(String)
|
|
products = relationship("Product", secondary="retailer_product")
|
|
|
|
class Price(Base):
|
|
__tablename__ = 'price'
|
|
id = Column(Integer, primary_key=True)
|
|
MOQ = Column(Integer)
|
|
price = Column(Float)
|
|
|
|
class ScraperJob(Base):
|
|
__tablename__ = 'scraper_job'
|
|
id = Column(Integer, primary_key=True)
|
|
start_time = Column(DateTime)
|
|
end_time = Column(DateTime)
|
|
parameters = Column(String)
|
|
statistics = Column(String)
|
|
|
|
class PriceHistory(Base):
|
|
__tablename__ = 'price_history'
|
|
id = Column(Integer, primary_key=True)
|
|
location = Column(String)
|
|
zip_code = Column(Integer)
|
|
city_name = Column(String)
|
|
price_at_time = Column(DateTime)
|
|
price = Column(Float)
|
|
scraper_job_id = Column(Integer, ForeignKey('scraper_job.id'))
|
|
product_id = Column(Integer, ForeignKey('product.id'))
|
|
product = relationship("Product", back_populates="price_histories")
|
|
scraper_job = relationship("ScraperJob")
|
|
|
|
class User(Base):
|
|
__tablename__ = 'user'
|
|
id = Column(Integer, primary_key=True)
|
|
username = Column(String)
|
|
credentials = Column(String)
|
|
creation_date = Column(DateTime)
|
|
modification_date = Column(DateTime)
|
|
|
|
class RetailerProduct(Base):
|
|
__tablename__ = 'retailer_product'
|
|
retailer_id = Column(Integer, ForeignKey('retailer.id'), primary_key=True)
|
|
product_id = Column(Integer, ForeignKey('product.id'), primary_key=True)
|
|
|
|
|
|
# In[ ]:
|
|
|
|
|
|
# Create the database engine
|
|
engine = create_engine('sqlite:///user_database.db')
|
|
Base.metadata.create_all(engine)
|
|
|
|
|
|
# In[ ]:
|
|
|
|
|
|
# Create a session factory
|
|
Session = sessionmaker(bind=engine)
|
|
session = Session()
|
|
|
|
|
|
# In[ ]:
|
|
|
|
|
|
import sys
|
|
|
|
# Create a command-line tool that parses sys.argv to allow a user to create/edit/delete user objects.
|
|
|
|
if len(sys.argv) < 2:
|
|
app = Flask(__name__)
|
|
|
|
# Create a database connection
|
|
engine = create_engine('sqlite:///user_database.db')
|
|
Session = sessionmaker(bind=engine)
|
|
session = Session()
|
|
|
|
@app.route('/')
|
|
def home():
|
|
# Get all user objects
|
|
users = session.query(User).all()
|
|
return render_template('index.html', users=users)
|
|
|
|
app.run()
|
|
|
|
|
|
command = sys.argv[1]
|
|
|
|
if command == "create":
|
|
if len(sys.argv) < 4:
|
|
print("Usage: python script.py create [username] [credentials]")
|
|
sys.exit(1)
|
|
|
|
username = sys.argv[2]
|
|
credentials = sys.argv[3]
|
|
|
|
# Create a new user object
|
|
new_user = User(username=username, credentials=credentials)
|
|
session.add(new_user)
|
|
session.commit()
|
|
print("User created successfully!")
|
|
|
|
elif command == "edit":
|
|
if len(sys.argv) < 4:
|
|
print("Usage: python script.py edit [username] [new_credentials]")
|
|
sys.exit(1)
|
|
|
|
username = sys.argv[2]
|
|
new_credentials = sys.argv[3]
|
|
|
|
# Find the user object
|
|
user = session.query(User).filter_by(username=username).first()
|
|
|
|
if user:
|
|
# Update user credentials
|
|
user.credentials = new_credentials
|
|
session.commit()
|
|
print("User edited successfully!")
|
|
else:
|
|
print("User not found!")
|
|
|
|
elif command == "delete":
|
|
if len(sys.argv) < 3:
|
|
print("Usage: python script.py delete [username]")
|
|
sys.exit(1)
|
|
|
|
username = sys.argv[2]
|
|
|
|
# Find the user object
|
|
user = session.query(User).filter_by(username=username).first()
|
|
|
|
if user:
|
|
# Delete the user object
|
|
session.delete(user)
|
|
session.commit()
|
|
print("User deleted successfully!")
|
|
else:
|
|
print("User not found!")
|
|
|
|
else:
|
|
print("Usage: python script.py [create|edit|delete] [username] [credentials]")
|