CHIEFSOFT\ameye
74 lines
2.2 KiB
Python
74 lines
2.2 KiB
Python
from datetime import datetime, timezone, timedelta
|
|
from app.extensions import db
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.sql import func
|
|
|
|
|
|
class Country(db.Model):
|
|
__tablename__ = 'country'
|
|
|
|
id = db.Column(db.Integer, primary_key=True, autoincrement=True)
|
|
uid = db.Column(db.String(150), nullable=False)
|
|
code = db.Column(db.String(2), nullable=False)
|
|
description = db.Column(db.String(100), nullable=False)
|
|
status = db.Column(db.Integer, nullable=False, default=0)
|
|
signup = db.Column(db.Integer, nullable=False, default=0)
|
|
added = db.Column(db.DateTime(timezone=True), server_default=func.now())
|
|
|
|
@classmethod
|
|
def get_signup_countries(cls, signup):
|
|
country_list = cls.query.filter_by(signup=signup).order_by(cls.description.asc()).all()
|
|
if not country_list:
|
|
return None
|
|
return country_list
|
|
|
|
@classmethod
|
|
def get_all_countries(cls):
|
|
country_list = cls.query.all()
|
|
if not country_list:
|
|
return None
|
|
return country_list
|
|
|
|
@classmethod
|
|
def set_country_status(cls, country_uid, code, val):
|
|
|
|
country_data = cls.query.filter_by(uid=str(country_uid), code=str(code)).first()
|
|
if not country_data:
|
|
return None
|
|
country_data.status = val
|
|
return country_data
|
|
|
|
@classmethod
|
|
def set_country_signup(cls, country_uid, code, val):
|
|
country_data = cls.query.filter_by(uid=str(country_uid), code=str(code)).first()
|
|
if not country_data:
|
|
return None
|
|
country_data.signup = val
|
|
return country_data
|
|
|
|
def to_dict(self):
|
|
return {
|
|
"id": self.id,
|
|
"uid": str(self.uid),
|
|
"code": self.code,
|
|
"description": self.description,
|
|
"status": self.status,
|
|
"signup": self.signup,
|
|
"added": self.added.isoformat() if self.added else None
|
|
}
|
|
|
|
'''
|
|
CREATE TABLE country (
|
|
id SERIAL,
|
|
uid uuid DEFAULT uuid_generate_v4(),
|
|
code VARCHAR(2) UNIQUE NOT NULL,
|
|
description VARCHAR(100) UNIQUE NOT NULL,
|
|
status INT DEFAULT 0,
|
|
signup INT DEFAULT 0,
|
|
added timestamp without time zone DEFAULT now()
|
|
);
|
|
ALTER TABLE ONLY country
|
|
ADD CONSTRAINT country_id_key UNIQUE (id);
|
|
|
|
'''
|