class Column(db.Model):
MAPPINGS = {
"FORENAMES": ("FORENAMES", "FORENAME", "FIRSTNAME", "FIRSTNAMES",
"GIVENNAME", "GIVENNAMES"),
"SURNAME": ("SURNAME", "LASTNAME", "FAMILYNAME"),
"SEX": ("SEX", "GENDER"),
"DOB": ("DOB", "DATEOFBIRTH", "BIRTHDATE"),
"POSTCODE": ("POSTCODE", "POSTALCODE"),
"NHS_NUMBER": ("NHSNUMBER", ),
"SYSTEM_NUMBER":
("SYSTEMNUMBER", "SNUMBER", "UHLSNUMBER", "UHLSYSTEMNUMBER"),
"ADDRESS1": ("ADDRESS1", ),
"ADDRESS2": ("ADDRESS2", ),
"ADDRESS3": ("ADDRESS3", ),
"ADDRESS4": ("ADDRESS4", ),
"ADDRESS5": ("ADDRESS5", ),
"LOCAL_ID": ("LOCALID", ),
}
id = db.Column(db.Integer(), primary_key=True)
column_index = db.Column(db.Integer())
name = db.Column(db.String(100))
mapping = db.Column(db.String(100))
created_date = db.Column(db.DateTime, default=datetime.utcnow)
batch_id = db.Column(db.Integer, db.ForeignKey('batch.id'))
batch = db.relationship("Batch",
backref=db.backref(
'columns',
order_by=id,
cascade="all, delete-orphan",
))
@property
def full_name(self):
column_index_name = "Column {}".format(self.column_index)
if self.name:
return "{}: {}".format(column_index_name, self.name)
else:
return column_index_name
@staticmethod
def suggest_mapping(name):
name = name.translate({
ord(c): None
for c in string.punctuation + string.whitespace + '£€¬¦'
})
name = name.upper()
for mapping, options in Column.MAPPINGS.items():
if name in options:
return mapping
return ''
@staticmethod
def get_select_options():
result = [(m, m) for m in Column.MAPPINGS.keys()]
result.insert(0, ('', 'Not mapped'))
return result
class Role(db.Model, RoleMixin):
ADMIN_ROLENAME = "admin"
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(80), unique=True)
description = db.Column(db.String(255))
created_date = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
def __str__(self):
return self.name or ""
class User(db.Model, UserMixin):
id = db.Column(db.Integer, primary_key=True)
email = db.Column(db.String(255), unique=True)
password = db.Column(db.String(255),
nullable=False,
default=random_password)
first_name = db.Column(db.String(255))
last_name = db.Column(db.String(255))
active = db.Column(db.Boolean())
confirmed_at = db.Column(db.DateTime())
last_login_at = db.Column(db.DateTime())
current_login_at = db.Column(db.DateTime())
last_login_ip = db.Column(db.String(50))
current_login_ip = db.Column(db.String(50))
login_count = db.Column(db.Integer())
created_date = db.Column(db.DateTime,
nullable=False,
default=datetime.utcnow)
roles = db.relationship("Role",
secondary=roles_users,
backref=db.backref("users", lazy="dynamic"))
def __str__(self):
return self.email
@property
def full_name(self):
full_name = " ".join(filter(None, [self.first_name, self.last_name]))
return full_name or self.email
class Batch(db.Model):
id = db.Column(db.Integer(), primary_key=True)
name = db.Column(db.String(100))
filename = db.Column(db.String(500))
created_date = db.Column(db.DateTime, default=datetime.utcnow)
deleted = db.Column(db.Boolean, default=False)
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
user = db.relationship("User",
backref=db.backref(
'batches',
order_by=id,
cascade="all, delete-orphan",
))
def automap_columns(self):
for c in self.columns:
mapping = Column.suggest_mapping(c.name)
if not mapping or mapping in (c.mapping for c in self.columns):
c.mapping = ''
else:
c.mapping = mapping
class Details(db.Model):
id = db.Column(db.Integer(), primary_key=True)
created_date = db.Column(db.DateTime, default=datetime.utcnow)
forename = db.Column(db.String(100))
surname = db.Column(db.String(100))
dob = db.Column(db.Date())
sex = db.Column(db.String(10))
postcode = db.Column(db.String(10))
nhs_number = db.Column(db.String(10))
system_number = db.Column(db.String(10))
address1 = db.Column(db.String(100))
address2 = db.Column(db.String(100))
address3 = db.Column(db.String(100))
address4 = db.Column(db.String(100))
address5 = db.Column(db.String(100))
local_id = db.Column(db.String(100))
batch_id = db.Column(db.Integer, db.ForeignKey('batch.id'))
batch = db.relationship("Batch",
backref=db.backref(
'details',
order_by=id,
cascade="all, delete-orphan",
))
def __repr__(self):
return "<Details nhs_number:%s>" % self.__dict__
def __eq__(self, other):
return self.__dict__ == other.__dict__
def __lt__(self, other):
return self.nhs_number < other.nhs_number
def __le__(self, other):
return self.__lt__(other) or self.__eq__(other)
def __ne__(self, other):
return not self.__eq__(other)
def __gt__(self, other):
return not self.__le__(other)
def __ge__(self, other):
return not self.__lt__(other)