first commit
This commit is contained in:
Mateusz Gruszczyński
+205
@@ -0,0 +1,205 @@
|
||||
from datetime import datetime
|
||||
from flask_login import UserMixin
|
||||
from werkzeug.security import generate_password_hash, check_password_hash
|
||||
from .extensions import db
|
||||
|
||||
vehicle_users = db.Table(
|
||||
"vehicle_users",
|
||||
db.Column("vehicle_id", db.Integer, db.ForeignKey("vehicle.id"), primary_key=True),
|
||||
db.Column("user_id", db.Integer, db.ForeignKey("user.id"), primary_key=True),
|
||||
)
|
||||
|
||||
user_favorite_stations = db.Table(
|
||||
"user_favorite_stations",
|
||||
db.Column("user_id", db.Integer, db.ForeignKey("user.id"), primary_key=True),
|
||||
db.Column("station_company_id", db.Integer, db.ForeignKey("fuel_station_company.id"), primary_key=True),
|
||||
)
|
||||
|
||||
company_favorite_stations = db.Table(
|
||||
"company_favorite_stations",
|
||||
db.Column("settings_id", db.Integer, db.ForeignKey("company_settings.id"), primary_key=True),
|
||||
db.Column("station_company_id", db.Integer, db.ForeignKey("fuel_station_company.id"), primary_key=True),
|
||||
)
|
||||
|
||||
company_allowed_stations = db.Table(
|
||||
"company_allowed_stations",
|
||||
db.Column("settings_id", db.Integer, db.ForeignKey("company_settings.id"), primary_key=True),
|
||||
db.Column("station_company_id", db.Integer, db.ForeignKey("fuel_station_company.id"), primary_key=True),
|
||||
)
|
||||
|
||||
class User(UserMixin, db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
email = db.Column(db.String(160), unique=True, nullable=False, index=True)
|
||||
name = db.Column(db.String(120), nullable=False)
|
||||
password_hash = db.Column(db.String(255), nullable=False)
|
||||
role = db.Column(db.String(20), nullable=False, default="user")
|
||||
active = db.Column(db.Boolean, default=True, nullable=False)
|
||||
fuel_card_id = db.Column(db.Integer, db.ForeignKey("fuel_card.id"), index=True)
|
||||
company_id = db.Column(db.Integer, db.ForeignKey("company_settings.id"), index=True)
|
||||
created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
|
||||
owned_vehicles = db.relationship("Vehicle", back_populates="owner", foreign_keys="Vehicle.owner_id")
|
||||
shared_vehicles = db.relationship("Vehicle", secondary=vehicle_users, back_populates="drivers")
|
||||
fuel_card = db.relationship("FuelCard", foreign_keys=[fuel_card_id])
|
||||
company = db.relationship("CompanySettings", foreign_keys=[company_id], back_populates="users")
|
||||
|
||||
def set_password(self, password): self.password_hash = generate_password_hash(password)
|
||||
def check_password(self, password): return check_password_hash(self.password_hash, password)
|
||||
@property
|
||||
def is_active(self): return self.active
|
||||
|
||||
class CompanySettings(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
name = db.Column(db.String(160), default="Moja firma", nullable=False)
|
||||
entity_type = db.Column(db.String(30), default="JDG", nullable=False)
|
||||
vat_rate = db.Column(db.Numeric(5, 2), default=23, nullable=False)
|
||||
vat_deduction_percent = db.Column(db.Numeric(5, 2), default=50, nullable=False)
|
||||
region = db.Column(db.String(60), default="mazowieckie", nullable=False)
|
||||
invoice_split_day = db.Column(db.Integer, default=15, nullable=False)
|
||||
active = db.Column(db.Boolean, default=True, nullable=False)
|
||||
users = db.relationship("User", back_populates="company", foreign_keys="User.company_id")
|
||||
|
||||
class AppSetting(db.Model):
|
||||
key = db.Column(db.String(80), primary_key=True)
|
||||
value = db.Column(db.String(255), nullable=False)
|
||||
|
||||
class FuelCard(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
name = db.Column(db.String(120), nullable=False, unique=True, index=True)
|
||||
provider = db.Column(db.String(120), nullable=False)
|
||||
active = db.Column(db.Boolean, default=True, nullable=False)
|
||||
description = db.Column(db.String(255))
|
||||
company_id = db.Column(db.Integer, db.ForeignKey("company_settings.id"), index=True)
|
||||
created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
|
||||
station_rules = db.relationship("FuelCardStationRule", back_populates="fuel_card", cascade="all, delete-orphan")
|
||||
company = db.relationship("CompanySettings", foreign_keys=[company_id])
|
||||
|
||||
class Vehicle(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
name = db.Column(db.String(120), nullable=False)
|
||||
make = db.Column(db.String(80), nullable=False)
|
||||
model = db.Column(db.String(80), nullable=False)
|
||||
registration = db.Column(db.String(20), unique=True, nullable=False)
|
||||
fuel_type = db.Column(db.String(20), nullable=False, default="PB95")
|
||||
owner_id = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
|
||||
company_id = db.Column(db.Integer, db.ForeignKey("company_settings.id"), index=True)
|
||||
has_fuel_card = db.Column(db.Boolean, default=False, nullable=False)
|
||||
card_provider = db.Column(db.String(80))
|
||||
fuel_card_id = db.Column(db.Integer, db.ForeignKey("fuel_card.id"), index=True)
|
||||
current_odometer = db.Column(db.Integer, default=0, nullable=False)
|
||||
owner = db.relationship("User", back_populates="owned_vehicles", foreign_keys=[owner_id])
|
||||
drivers = db.relationship("User", secondary=vehicle_users, back_populates="shared_vehicles")
|
||||
fuel_card = db.relationship("FuelCard", foreign_keys=[fuel_card_id])
|
||||
company = db.relationship("CompanySettings", foreign_keys=[company_id])
|
||||
fuel_entries = db.relationship("FuelEntry", back_populates="vehicle", cascade="all, delete-orphan")
|
||||
|
||||
class FuelStationCompany(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
ure_key = db.Column(db.String(180), unique=True, nullable=False, index=True)
|
||||
brand_name = db.Column(db.String(120), nullable=False, default="", index=True)
|
||||
company_name = db.Column(db.String(220), nullable=False, index=True)
|
||||
nip = db.Column(db.String(20), index=True)
|
||||
regon = db.Column(db.String(20), index=True)
|
||||
station_names = db.Column(db.Text, default="", nullable=False)
|
||||
regions = db.Column(db.Text, default="", nullable=False)
|
||||
selected_region = db.Column(db.String(60), default="", nullable=False)
|
||||
station_count = db.Column(db.Integer, default=0, nullable=False)
|
||||
has_petrol = db.Column(db.Boolean, default=False, nullable=False)
|
||||
has_diesel = db.Column(db.Boolean, default=False, nullable=False)
|
||||
has_lpg = db.Column(db.Boolean, default=False, nullable=False)
|
||||
use_orlen_last_price = db.Column(db.Boolean, default=False, nullable=False)
|
||||
active = db.Column(db.Boolean, default=True, nullable=False)
|
||||
fetched_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
|
||||
points = db.relationship("FuelStationPoint", back_populates="station_company", cascade="all, delete-orphan", lazy="dynamic")
|
||||
|
||||
@property
|
||||
def region_list(self): return [x for x in (self.regions or "").split("|") if x]
|
||||
@property
|
||||
def station_name_list(self): return [x for x in (self.station_names or "").split("|") if x]
|
||||
@property
|
||||
def display_name(self): return self.brand_name or self.company_name
|
||||
|
||||
|
||||
class FuelStationPoint(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
station_company_id = db.Column(db.Integer, db.ForeignKey("fuel_station_company.id"), nullable=False, index=True)
|
||||
ure_dkn = db.Column(db.String(40), nullable=False, index=True)
|
||||
station_name = db.Column(db.String(220), nullable=False, default="")
|
||||
street = db.Column(db.String(180), nullable=False, default="")
|
||||
street_number = db.Column(db.String(40), nullable=False, default="")
|
||||
postal_code = db.Column(db.String(20), nullable=False, default="")
|
||||
city = db.Column(db.String(120), nullable=False, default="")
|
||||
region = db.Column(db.String(60), nullable=False, default="")
|
||||
coordinates = db.Column(db.String(120), nullable=False, default="")
|
||||
has_petrol = db.Column(db.Boolean, default=False, nullable=False)
|
||||
has_diesel = db.Column(db.Boolean, default=False, nullable=False)
|
||||
has_lpg = db.Column(db.Boolean, default=False, nullable=False)
|
||||
station_company = db.relationship("FuelStationCompany", back_populates="points")
|
||||
__table_args__ = (db.UniqueConstraint("station_company_id", "ure_dkn", name="uq_station_company_point_dkn"),)
|
||||
|
||||
@property
|
||||
def address(self):
|
||||
line = " ".join(x for x in (self.street, self.street_number) if x).strip()
|
||||
locality = " ".join(x for x in (self.postal_code, self.city) if x).strip()
|
||||
return ", ".join(x for x in (line, locality) if x) or "Brak adresu"
|
||||
|
||||
class FuelCardStationRule(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
fuel_card_id = db.Column(db.Integer, db.ForeignKey("fuel_card.id"), nullable=False, index=True)
|
||||
station_company_id = db.Column(db.Integer, db.ForeignKey("fuel_station_company.id"), nullable=False, index=True)
|
||||
allowed = db.Column(db.Boolean, default=True, nullable=False)
|
||||
use_orlen_last_price = db.Column(db.Boolean, default=False, nullable=False)
|
||||
discount_net_percent = db.Column(db.Numeric(6, 3), default=0, nullable=False)
|
||||
surcharge_net_per_liter = db.Column(db.Numeric(10, 4), default=0, nullable=False)
|
||||
fuel_card = db.relationship("FuelCard", back_populates="station_rules")
|
||||
station_company = db.relationship("FuelStationCompany")
|
||||
__table_args__ = (db.UniqueConstraint("fuel_card_id", "station_company_id", name="uq_card_station_rule"),)
|
||||
|
||||
class FuelEntry(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
vehicle_id = db.Column(db.Integer, db.ForeignKey("vehicle.id"), nullable=False, index=True)
|
||||
user_id = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
|
||||
fueled_at = db.Column(db.DateTime, nullable=False, default=datetime.utcnow, index=True)
|
||||
liters = db.Column(db.Numeric(10, 3), nullable=False)
|
||||
price_per_liter = db.Column(db.Numeric(10, 4), nullable=False)
|
||||
fuel_type = db.Column(db.String(20), nullable=False)
|
||||
odometer = db.Column(db.Integer, nullable=False)
|
||||
station = db.Column(db.String(120))
|
||||
station_company_id = db.Column(db.Integer, db.ForeignKey("fuel_station_company.id"), index=True)
|
||||
fuel_card_id = db.Column(db.Integer, db.ForeignKey("fuel_card.id"), index=True)
|
||||
invoice_number = db.Column(db.String(80))
|
||||
used_fuel_card = db.Column(db.Boolean, default=False, nullable=False)
|
||||
wholesale_price = db.Column(db.Numeric(10, 4))
|
||||
wholesale_source = db.Column(db.String(120))
|
||||
created_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
|
||||
vehicle = db.relationship("Vehicle", back_populates="fuel_entries")
|
||||
user = db.relationship("User")
|
||||
station_company = db.relationship("FuelStationCompany")
|
||||
fuel_card = db.relationship("FuelCard")
|
||||
@property
|
||||
def gross(self): return float(self.liters) * float(self.price_per_liter)
|
||||
|
||||
class OrlenPrice(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
fuel_type = db.Column(db.String(20), nullable=False, index=True)
|
||||
effective_date = db.Column(db.Date, nullable=False, index=True)
|
||||
price_per_liter = db.Column(db.Numeric(10, 4), nullable=False)
|
||||
raw_value = db.Column(db.Numeric(14, 4), nullable=False)
|
||||
region = db.Column(db.String(60), nullable=False, default="")
|
||||
product_name = db.Column(db.String(80), nullable=False)
|
||||
source = db.Column(db.String(160), nullable=False)
|
||||
fetched_at = db.Column(db.DateTime, default=datetime.utcnow, nullable=False)
|
||||
__table_args__ = (db.UniqueConstraint("fuel_type", "effective_date", "region", name="uq_orlen_price"),)
|
||||
|
||||
class FuelCardPolicy(db.Model):
|
||||
id = db.Column(db.Integer, primary_key=True)
|
||||
vehicle_id = db.Column(db.Integer, db.ForeignKey("vehicle.id"), nullable=False, unique=True, index=True)
|
||||
use_orlen_last_price = db.Column(db.Boolean, default=False, nullable=False)
|
||||
discount_percent = db.Column(db.Numeric(6, 3), default=0, nullable=False)
|
||||
surcharge_per_liter = db.Column(db.Numeric(10, 4), default=0, nullable=False)
|
||||
description = db.Column(db.String(160))
|
||||
vehicle = db.relationship("Vehicle", backref=db.backref("fuel_card_policy", uselist=False, cascade="all, delete-orphan"))
|
||||
|
||||
User.favorite_stations = db.relationship("FuelStationCompany", secondary=user_favorite_stations, backref=db.backref("favorite_users", lazy="dynamic"))
|
||||
CompanySettings.allowed_stations = db.relationship("FuelStationCompany", secondary=company_allowed_stations, backref=db.backref("allowed_for_companies", lazy="dynamic"))
|
||||
|
||||
CompanySettings.favorite_stations = db.relationship("FuelStationCompany", secondary=company_favorite_stations, backref=db.backref("favorite_for_companies", lazy="dynamic"))
|
||||
Reference in New Issue
Block a user