class User(UserMixin, db.Model, Mixin): __tablename__ = "users" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(200), unique=True) password_hash = db.Column(db.String(128)) last_seen = db.Column(db.DateTime, index=True, default=datetime.utcnow) role_id = db.Column(db.Integer, db.ForeignKey('roles.id')) def __init__(self, name, password, role_id): self.password = password self.name = name self.role_id = role_id def __str__(self): return "<%r>" % self.name @property def password(self): raise AttributeError('password not for reading !!') @password.setter def password(self, password): self.password_hash = generate_password_hash(password) def verify_password(self, password): return check_password_hash(self.password_hash, password) @classmethod def has_default_password(cls): return cls.query.filter_by( id=1).first().verify_password(DEFAULT_PASSWORD)
class Task(db.Model, Mixin): __tablename__ = "tasks" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(300)) timestamp = db.Column(db.DateTime(), index=True, default=datetime.utcnow) def __init__(self, name): self.name = name @property def common(self): return len(self.offices) > 1 def least_tickets_office(self): self.offices.sort(key=lambda o: Serial.query.filter_by(office_id=o.id).count()) return self.offices[0] def migrate_tickets(self, from_office, to_office): params = dict(office_id=from_office.id, task_id=self.id) tickets = Serial.query.filter_by(**params).all() tickets += Waiting.query.filter_by(**params).all() for ticket in tickets: ticket.office_id = to_office.id db.session.commit()
class Printer(db.Model, Mixin): __tablename__ = "printers" id = db.Column(db.Integer, primary_key=True) vendor = db.Column(db.Integer, nullable=True, unique=True) product = db.Column(db.Integer, nullable=True, unique=True) in_ep = db.Column(db.Integer, nullable=True) out_ep = db.Column(db.Integer, nullable=True) active = db.Column(db.Boolean()) langu = db.Column(db.String(100)) value = db.Column(db.Integer) scale = db.Column(db.Integer, default=1) name = db.Column(db.String(100), nullable=True) def __init__(self, vendor=0, product=0, in_ep=None, out_ep=None, active=False, langu='en', value=1, scale=1, name=None): self.vendor = vendor self.product = product self.in_ep = in_ep self.out_ep = out_ep self.active = active self.value = value self.scale = scale self.name = name
class Settings(db.Model): __tablename__ = 'settings' id = db.Column(db.Integer, primary_key=True) notifications = db.Column(db.Boolean) def __init__(self, notifications=True): self.id = 0 self.notifications = notifications
class Operators(db.Model, Mixin): __tablename__ = "operators" crap = db.Column(db.Integer, primary_key=True) id = db.Column(db.Integer) office_id = db.Column(db.Integer, db.ForeignKey('offices.id')) def __init__(self, id, office_id): self.id = id self.office_id = office_id
class BackgroundTask(db.Model, Mixin): __tablename__ = 'background_tasks' id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(100), unique=True) enabled = db.Column(db.Boolean, default=False) every = db.Column(db.String(10)) time = db.Column(db.Time, nullable=True) def __init__(self, name, enabled=False, every=None, time=None): self.name = name self.enabled = enabled self.every = every self.time = time
class Office(db.Model, Mixin): __tablename__ = "offices" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.Integer, unique=True) timestamp = db.Column(db.DateTime(), index=True, default=datetime.utcnow) prefix = db.Column(db.String(2)) operators = db.relationship('Operators', backref='operators') tasks = db.relationship('Task', secondary=mtasks, lazy='subquery', backref=db.backref('offices', lazy=True)) def __init__(self, name, prefix): self.name = name self.prefix = prefix
class Slides_c(db.Model): __tablename__ = "slides_c" id = db.Column(db.Integer, primary_key=True) rotation = db.Column(db.String(100)) navigation = db.Column(db.Integer) effect = db.Column(db.String(100)) status = db.Column(db.Integer) def __init__(self, rotation="3000", navigation=2, effect="slide", status=2): self.id = 0 self.rotation = rotation self.effect = effect self.status = status self.navigation = navigation
class Settings(db.Model, Mixin): __tablename__ = 'settings' id = db.Column(db.Integer, primary_key=True) notifications = db.Column(db.Boolean, nullable=True) strict_pulling = db.Column(db.Boolean, nullable=True) visual_effects = db.Column(db.Boolean, nullable=True) lp_printing = db.Column(db.Boolean, nullable=True) def __init__(self, notifications=True, strict_pulling=True, visual_effects=True, lp_printing=False): self.id = 0 self.notifications = notifications self.strict_pulling = strict_pulling self.visual_effects = visual_effects self.lp_printing = lp_printing
class Roles(db.Model): __tablename__ = "roles" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(10), unique=True) def __init__(self, id, name): self.id = id self.name = name @classmethod def load_roles(cls): for _id, name in USER_ROLES.items(): existing = cls.query.filter_by(id=_id, name=name)\ .first() if not existing: db.session.add(Roles(id=_id, name=name)) db.session.commit()
class Task(db.Model, Mixin): __tablename__ = "tasks" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(300)) timestamp = db.Column(db.DateTime(), index=True, default=datetime.utcnow) hidden = db.Column(db.Boolean, default=False, nullable=True) def __init__(self, name='Generic', hidden=False): self.name = name self.hidden = hidden @property def common(self): return len(self.offices) > 1 @classmethod def get_first_common(cls): for task in cls.query: if task.common: return task @classmethod def in_offices(cls, office_id): return cls.query.filter(cls.offices.contains(Office.get(office_id))) def least_tickets_office(self): self.offices.sort( key=lambda o: Serial.query.filter_by(office_id=o.id).count()) return self.offices[0] @property def tickets(self): return Serial.query.filter(Serial.task_id == self.id, Serial.number != 100) def migrate_tickets(self, from_office, to_office): params = dict(office_id=from_office.id, task_id=self.id) tickets = Serial.query.filter_by(**params).all() for ticket in tickets: ticket.office_id = to_office.id db.session.commit()
class Settings(db.Model): __tablename__ = 'settings' id = db.Column(db.Integer, primary_key=True) notifications = db.Column(db.Boolean, nullable=True) strict_pulling = db.Column(db.Boolean, nullable=True) visual_effects = db.Column(db.Boolean, nullable=True) def __init__(self, notifications=True, strict_pulling=True, visual_effects=True): self.id = 0 self.notifications = notifications self.strict_pulling = strict_pulling self.visual_effects = visual_effects @classmethod def get(cls): return cls.query.first()
class Office(db.Model, Mixin): __tablename__ = "offices" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.Integer, unique=True) timestamp = db.Column(db.DateTime(), index=True, default=datetime.utcnow) prefix = db.Column(db.String(2)) operators = db.relationship('Operators', backref='operators') tasks = db.relationship('Task', secondary=mtasks, lazy='subquery', backref=db.backref('offices', lazy=True)) def __init__(self, name, prefix): self.name = name self.prefix = prefix @classmethod def get_first_available_prefix(cls): letters = list(map(lambda i: chr(i), range(97, 123))) prefix = None while letters and not prefix: match_letter = letters.pop() if not cls.query.filter_by(prefix=match_letter).first(): prefix = match_letter return prefix and prefix.upper() @property def tickets(self): return Serial.query.filter(Serial.office_id == self.id, Serial.number != 100) @property def display_text(self): show_prefix = Display_store.query.first()\ .prefix return f'{self.prefix if show_prefix else ""}{self.name}'
class Media(db.Model, Mixin): __tablename__ = "media" id = db.Column(db.Integer, primary_key=True) vid = db.Column(db.Boolean()) audio = db.Column(db.Boolean()) img = db.Column(db.Boolean()) used = db.Column(db.Boolean()) name = db.Column(db.String(50)) def __init__(self, vid=False, audio=False, img=False, used=False, name=" "): self.vid = vid self.audio = audio self.img = img self.used = used self.name = name def is_used(self): return any([ Vid.query.filter_by(vkey=self.id).first(), Slides.query.filter_by(ikey=self.id).first(), Display_store.query.filter( or_(Display_store.ikey == self.id, Display_store.akey == self.id)).first(), Touch_store.query.filter( or_(Touch_store.ikey == self.id, Touch_store.akey == self.id)).first() ])
class AuthTokens(db.Model, Mixin): __tablename__ = 'auth_tokens' id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(100)) description = db.Column(db.String(300), nullable=True) token = db.Column(db.String(32), unique=True) role_id = db.Column(db.Integer, db.ForeignKey('roles.id')) def __init__(self, name, description=None): self.name = name self.description = description self.token = self.get_unique_token() # NOTE: defaulting role to Adminstator, most likely in the future # we would want this to be customizeable, with varied API permissions. self.role_id = USER_ROLE_ADMIN @classmethod def get_unique_token(cls): token = f'{uuid4()}'.replace('-', '') while cls.get(token=token): token = f'{uuid4()}'.replace('-', '') return token
class Printer(db.Model): __tablename__ = "printers" id = db.Column(db.Integer, primary_key=True) vendor = db.Column(db.String(100), nullable=True, unique=True) product = db.Column(db.String(100), unique=True) in_ep = db.Column(db.Integer, nullable=True) out_ep = db.Column(db.Integer, nullable=True) active = db.Column(db.Boolean()) langu = db.Column(db.String(100)) value = db.Column(db.Integer) def __init__(self, vendor=" ", product=" ", in_ep=0, out_ep=0, active=False, langu='en', value=1): self.vendor = vendor self.product = product self.in_ep = in_ep self.out_ep = out_ep self.active = active self.value = value
class Waiting(db.Model): __tablename__ = "waitings" id = db.Column(db.Integer, primary_key=True) number = db.Column(db.Integer) name = db.Column(db.String(300), nullable=True) n = db.Column(db.Boolean) office_id = db.Column(db.Integer, db.ForeignKey('offices.id')) task_id = db.Column(db.Integer, db.ForeignKey('tasks.id')) def __init__(self, number, office_id, task_id, name=None, n=False): self.number = number self.office_id = office_id self.task_id = task_id self.name = name self.n = n
class Vid(db.Model, Mixin): __tablename__ = "vids" id = db.Column(db.Integer, primary_key=True) vname = db.Column(db.String(300)) enable = db.Column(db.Integer) ar = db.Column(db.Integer) controls = db.Column(db.Integer) mute = db.Column(db.Integer) vkey = db.Column(db.Integer, db.ForeignKey("media.id", ondelete='cascade'), nullable=True) def __init__(self, vname="", enable=0, ar=1, controls=1, mute=2, vkey=6): self.vname = vname self.enable = enable self.ar = ar self.controls = controls self.mute = mute self.vkey = vkey
class Media(db.Model): __tablename__ = "media" id = db.Column(db.Integer, primary_key=True) vid = db.Column(db.Boolean()) audio = db.Column(db.Boolean()) img = db.Column(db.Boolean()) used = db.Column(db.Boolean()) name = db.Column(db.String(50)) def __init__(self, vid=False, audio=False, img=False, used=False, name=" "): self.vid = vid self.audio = audio self.img = img self.used = used self.name = name
class Waiting_c(db.Model): __tablename__ = "waitings_c" id = db.Column(db.Integer, primary_key=True) ticket = db.Column(db.String) oname = db.Column(db.String) tname = db.Column(db.String) n = db.Column(db.Boolean) name = db.Column(db.String(300), nullable=True) def __init__(self, ticket="Empty", oname="Empty", tname="Empty", n=False, name=None): self.id = 0 self.ticket = ticket self.oname = oname self.tname = tname self.n = n self.name = name
class Aliases(db.Model, Mixin): __tablename__ = "aliases" id = db.Column(db.Integer, primary_key=True) office = db.Column(db.String(100)) task = db.Column(db.String(100)) ticket = db.Column(db.String(100)) name = db.Column(db.String(100)) number = db.Column(db.String(100)) def __init__(self, office="office", task="task", ticket="ticket", name="name", number="number"): self.id = 0 self.office = office self.task = task self.ticket = ticket self.name = name self.number = number
class Office(db.Model, Mixin): __tablename__ = "offices" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(300), unique=True) timestamp = db.Column(db.DateTime(), index=True, default=datetime.utcnow) prefix = db.Column(db.String(2)) operators = db.relationship('Operators', backref='operators') tasks = db.relationship('Task', secondary=mtasks, lazy='subquery', backref=db.backref('offices', lazy=True)) def __init__(self, name=None, prefix=None): self.name = name or self.get_generic_available_name() self.prefix = prefix or self.get_first_available_prefix() @classmethod def get_all_used_prefixes(cls): return [o.prefix for o in cls.query.all()] @classmethod def get_all_available_prefixes(cls): used_prefixes = cls.get_all_used_prefixes() return [p for p in PREFIXES if p not in used_prefixes] @classmethod def get_first_available_prefix(cls): available_prefixes = cls.get_all_available_prefixes() if not available_prefixes: raise AssertionError('No prefixes left!') return available_prefixes[0] @classmethod def get_generic_available_name(cls): name = 0 while not name: temp_name = str(randint(1, 1000)) office = cls.query.filter_by(name=temp_name).first() if not office: name = temp_name return name @property def tickets(self): return Serial.query.filter(Serial.office_id == self.id, Serial.number != 100) @property def display_text(self): show_prefix = Display_store.query.first()\ .prefix return f'{self.prefix if show_prefix else ""}{self.name}' def delete_all(self): for ticket in self.tickets: db.session.delete(ticket) for task in self.tasks: if task.common: self.tasks.remove(task) else: db.session.delete(task) db.session.delete(self) db.session.commit() def is_valid_new_name(self, name): return not self.query.filter(Office.name == name, Office.id != self.id).first()
class Office(db.Model, Mixin): __tablename__ = "offices" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.Integer, unique=True) timestamp = db.Column(db.DateTime(), index=True, default=datetime.utcnow) prefix = db.Column(db.String(2)) operators = db.relationship('Operators', backref='operators') tasks = db.relationship('Task', secondary=mtasks, lazy='subquery', backref=db.backref('offices', lazy=True)) def __init__(self, name=None, prefix=None): self.name = name or self.get_generic_available_name() self.prefix = prefix or self.get_first_available_prefix() @classmethod def get_first_available_prefix(cls): letters = list(reversed(list(map(lambda i: chr(i), range(97, 123))))) prefix = None while letters and not prefix: match_letter = letters.pop() if not cls.query.filter_by(prefix=match_letter).first(): prefix = match_letter return prefix and prefix.upper() @classmethod def get_generic_available_name(cls): name = 0 while not name: temp_name = randint(1, 1000) office = cls.query.filter_by(name=temp_name).first() if not office: name = temp_name return name @property def tickets(self): return Serial.query.filter(Serial.office_id == self.id, Serial.number != 100) @property def display_text(self): show_prefix = Display_store.query.first()\ .prefix return f'{self.prefix if show_prefix else ""}{self.name}' def delete_all(self): for ticket in self.tickets: db.session.delete(ticket) for task in self.tasks: if task.common: self.tasks.remove(task) else: db.session.delete(task) db.session.delete(self) db.session.commit()
class Slides(db.Model, Mixin): __tablename__ = "slides" id = db.Column(db.Integer, primary_key=True) title = db.Column(db.String(300)) hsize = db.Column(db.String(100)) hcolor = db.Column(db.String(100)) hfont = db.Column(db.String(100)) hbg = db.Column(db.String(100)) subti = db.Column(db.String(300)) tsize = db.Column(db.String(100)) tcolor = db.Column(db.String(100)) tfont = db.Column(db.String(100)) tbg = db.Column(db.String(100)) bname = db.Column(db.String(300)) ikey = db.Column(db.Integer, db.ForeignKey("media.id", ondelete='cascade'), nullable=True)
class Roles(db.Model): __tablename__ = "roles" id = db.Column(db.Integer, primary_key=True) name = db.Column(db.String(10), unique=True)
class Serial(db.Model): __tablename__ = "serials" id = db.Column(db.Integer, primary_key=True) number = db.Column(db.Integer) timestamp = db.Column(db.DateTime(), index=True, default=datetime.utcnow) date = db.Column(db.Date(), default=datetime.utcnow().date) name = db.Column(db.String(300), nullable=True) n = db.Column(db.Boolean) p = db.Column(db.Boolean) # stands for proccessed , which be modified after been processed pdt = db.Column(db.DateTime()) # Fix: adding pulled by feature to tickets pulledBy = db.Column(db.Integer) office_id = db.Column(db.Integer, db.ForeignKey('offices.id')) task_id = db.Column(db.Integer, db.ForeignKey('tasks.id')) def __init__(self, number=100, office_id=1, task_id=1, name=None, n=False, p=False, pulledBy=0): self.number = number self.office_id = office_id self.task_id = task_id self.name = name self.n = n # fixing mass use tickets multi operators conflict self.p = p self.pulledBy = pulledBy @property def task(self): return Task.query.filter_by(id=self.task_id).first() @property def office(self): return Office.query.filter_by(id=self.office_id).first()
class Display_store(db.Model, Mixin): __tablename__ = 'displays' id = db.Column(db.Integer, primary_key=True) title = db.Column(db.String(300)) hsize = db.Column(db.String(100)) hcolor = db.Column(db.String(100)) h2size = db.Column(db.String(100)) h2color = db.Column(db.String(100)) h2font = db.Column(db.String(100)) hbg = db.Column(db.String(100)) tsize = db.Column(db.String(100)) tcolor = db.Column(db.String(100)) ssize = db.Column(db.String(100)) scolor = db.Column(db.String(100)) audio = db.Column(db.String(10)) hfont = db.Column(db.String(100)) tfont = db.Column(db.String(100)) sfont = db.Column(db.String(100)) mduration = db.Column(db.String(100)) rrate = db.Column(db.String(100)) announce = db.Column(db.String(100)) anr = db.Column(db.Integer) anrt = db.Column(db.String(100)) effect = db.Column(db.String(100)) repeats = db.Column(db.String(100)) bgcolor = db.Column(db.String(100)) tmp = db.Column(db.Integer) prefix = db.Column(db.Boolean, default=False) always_show_ticket_number = db.Column(db.Boolean, default=False) # adding repeat announcement value r_announcement = db.Column(db.Boolean) akey = db.Column(db.Integer, db.ForeignKey("media.id", ondelete='cascade'), nullable=True) ikey = db.Column(db.Integer, db.ForeignKey("media.id", ondelete='cascade'), nullable=True) vkey = db.Column(db.Integer, db.ForeignKey("media.id", ondelete='cascade'), nullable=True) def __init__(self, id=0, title="FQM Queue Management", hsize="500%", hcolor="rgb(129, 200, 139)", h2size="600%", h2color="rgb(184, 193, 255)", h2font="Mada", hbg="rgba(0, 0, 0, 0.5)", tsize="600%", tcolor="rgb(184, 193, 255)", ssize="500%", scolor="rgb(224, 224, 224)", audio="false", hfont="El Messiri", tfont="Mada", repeats="3", effect="fade", sfont="Amiri", mduration="3000", rrate="2000", announce="en-us", ikey=4, vkey=None, akey=None, anr=2, anrt="each", bgcolor="rgb(0,0,0)", tmp=1): self.id = 0 self.tfont = tfont self.hfont = hfont self.h2font = h2font self.sfont = sfont self.mduration = mduration self.rrate = rrate self.title = title self.hsize = hsize self.hcolor = hcolor self.hsize = hsize self.hcolor = hcolor self.h2color = h2color self.h2size = h2size self.hcolor = hcolor self.hbg = hbg self.tsize = tsize self.tcolor = tcolor self.ssize = ssize self.scolor = scolor self.audio = audio self.announce = announce self.bgcolor = bgcolor self.tmp = tmp self.anr = anr self.anrt = anrt self.effect = effect self.repeats = repeats self.ikey = ikey self.vkey = vkey self.akey = akey @classmethod def get(cls): return cls.query.first()
class Touch_store(db.Model, Mixin): __tablename__ = 'touchs' id = db.Column(db.Integer, primary_key=True) title = db.Column(db.String(300)) message = db.Column(db.Text()) hsize = db.Column(db.String(100)) hcolor = db.Column(db.String(100)) hbg = db.Column(db.String(100)) tsize = db.Column(db.String(100)) tcolor = db.Column(db.String(100)) msize = db.Column(db.String(100)) mcolor = db.Column(db.String(100)) mbg = db.Column(db.String(100)) audio = db.Column(db.String(100)) hfont = db.Column(db.String(100)) mfont = db.Column(db.String(100)) tfont = db.Column(db.String(100)) mduration = db.Column(db.String(100)) bgcolor = db.Column(db.String(100)) p = db.Column(db.Boolean) n = db.Column(db.Boolean) tmp = db.Column(db.Integer) akey = db.Column(db.Integer, db.ForeignKey("media.id", ondelete='cascade'), nullable=True) ikey = db.Column(db.Integer, db.ForeignKey("media.id", ondelete='cascade'), nullable=True) def __init__(self, id=0, title="Please select a task to pull a tick for", hsize="500%", hcolor="rgb(129, 200, 139)", hbg="rgba(0, 0, 0, 0.50)", tsize="400%", mbg="rgba(0, 0, 0, 0.50)", msize="400%", mcolor="rgb(255, 255, 0)", tcolor="btn-danger", message="Ticket has been issued, pleas wait your turn", audio="false", hfont="El Messiri", mfont="Mada", tfont="Amiri", ikey=None, tmp=2, akey=None, mduration="3000", bgcolor="rgb(0, 0, 0)", p=False, n=True): self.id = 0 self.hfont = hfont self.mfont = mfont self.tfont = tfont self.mduration = mduration self.title = title self.message = message self.hsize = hsize self.hcolor = hcolor self.hbg = hbg self.tsize = tsize self.tcolor = tcolor self.msize = msize self.mcolor = mcolor self.mbg = mbg self.audio = audio self.bgcolor = bgcolor self.ikey = ikey self.akey = akey self.p = p self.n = n self.tmp = tmp
# -*- coding: utf-8 -*- # This Source Code Form is subject to the terms of the Mozilla Public # License, v. 2.0. If a copy of the MPL was not distributed with this # file, You can obtain one at http://mozilla.org/MPL/2.0/. from flask_login import UserMixin, current_user from sqlalchemy.sql import and_, or_ from werkzeug.security import generate_password_hash, check_password_hash from datetime import datetime from app.middleware import db from app.constants import USER_ROLES, DEFAULT_PASSWORD mtasks = db.Table( 'mtasks', db.Column('office_id', db.Integer, db.ForeignKey('offices.id'), primary_key=True), db.Column('task_id', db.Integer, db.ForeignKey('tasks.id'), primary_key=True)) class Mixin: @classmethod def get(cls, id=False): if id is False: return cls.query.first() return cls.query.filter_by(id=id).first()
class Serial(db.Model, TicketsMixin): __tablename__ = "serials" id = db.Column(db.Integer, primary_key=True) number = db.Column(db.Integer) timestamp = db.Column(db.DateTime(), index=True, default=datetime.utcnow) date = db.Column(db.Date(), default=datetime.utcnow().date) name = db.Column(db.String(300), nullable=True) n = db.Column(db.Boolean) p = db.Column(db.Boolean) # stands for proccessed , which be modified after been processed pdt = db.Column(db.DateTime()) # Fix: adding pulled by feature to tickets pulledBy = db.Column(db.Integer) on_hold = db.Column(db.Boolean, default=False) office_id = db.Column(db.Integer, db.ForeignKey('offices.id')) task_id = db.Column(db.Integer, db.ForeignKey('tasks.id')) def __init__(self, number=100, office_id=1, task_id=1, name=None, n=False, p=False, pulledBy=0): self.number = number self.office_id = office_id self.task_id = task_id self.name = name self.n = n # fixing mass use tickets multi operators conflict self.p = p self.pulledBy = pulledBy @property def task(self): return Task.query.filter_by(id=self.task_id).first() @property def office(self): return Office.query.filter_by(id=self.office_id).first() @property def puller_name(self): return User.get(self.pulledBy).name @classmethod def all_office_tickets(cls, office_id): ''' get tickets of the common task from other offices. Parameters ---------- office_id: int id of the office to retreive tickets for. Returns ------- Query of office tickets unionned with other offices tickets. ''' strict_pulling = Settings.get().strict_pulling office = Office.get(office_id) all_tickets = cls.query.filter(cls.office_id == office_id, cls.number != 100) if not strict_pulling: for task in office.tasks: other_office_tickets = cls.query.filter( and_(cls.task_id == task.id, cls.office_id != office_id)) if other_office_tickets.count(): all_tickets = all_tickets.union(other_office_tickets) return all_tickets.filter(Serial.number != 100)\ .order_by(Serial.p, Serial.timestamp.desc()) @classmethod def all_task_tickets(cls, office_id, task_id): ''' get tickets related to a given task and office. Parameters ---------- office_id: int id of the office that we're querying from. task_id: int id of the task we want to retrieve its tickets. Returns ------- Query of task tickets filterred based on `strict_pulling`. ''' strict_pulling = Settings.get().strict_pulling filter_parameters = {'office_id': office_id, 'task_id': task_id} if not strict_pulling: filter_parameters.pop('office_id') return cls.query.filter_by(**filter_parameters)\ .filter(cls.number != 100)\ .order_by(cls.p, cls.timestamp.desc()) @classmethod def get_last_pulled_ticket(cls, office_id=None): ''' get the last pulled ticket. Parameters ---------- office_id: int office's id to filter last tickets by. Returns ------- Last ticket pulled record. ''' last_ticket = cls.query.filter_by(p=True)\ .filter(cls.number != 100) if office_id: last_ticket = last_ticket.filter_by(office_id=office_id) return last_ticket.order_by(cls.pdt.desc())\ .first() @classmethod def get_waiting_list_tickets(cls, office_id=None, limit=9): ''' get list of waiting tickets to be processed next. Parameters ---------- office_id: int office's id to filter tickets by. limit: int number of ticket to limit the query to. Returns ------- List of waiting list tickets. ''' waiting_tickets = cls.query.filter_by(p=False)\ .filter(cls.number != 100) if office_id: waiting_tickets = waiting_tickets.filter( cls.office_id == office_id) return waiting_tickets.order_by(cls.pdt.desc())\ .limit(limit)\ .all() def pull(self, office_id): ''' Mark a ticket as pulled and do the dues. Parameters ---------- office_id: int id of the office from which the ticket is pulled. ''' self.p = True self.pdt = datetime.utcnow() self.pulledBy = getattr(current_user, 'id', None) self.office_id = office_id db.session.add(self) db.session.commit() def toggle_on_hold(self): ''' Toggle the ticket `on_hold` status. ''' self.on_hold = not self.on_hold db.session.add(self) db.session.commit()