class Note(db.Model):
# A note on a request.
__tablename__ = 'note'
id = db.Column(db.Integer, primary_key = True)
date_created = db.Column(db.DateTime)
text = db.Column(db.String())
request_id = db.Column(db.Integer, db.ForeignKey('request.id')) # The request it belongs to.
user_id = db.Column(db.Integer, db.ForeignKey('user.id')) # The user who wrote the note. Right now only city staff can.
def __init__(self, request_id, text, user_id):
self.text = text
self.request_id = request_id
self.user_id = user_id
self.date_created = datetime.now().isoformat()
def __repr__(self):
return '<Note %r>' % self.text
class Subscriber(db.Model):
# A person subscribed to a request, who may or may not have created the request, and may or may not own a part of the request.
__tablename__ = 'subscriber'
id = db.Column(db.Integer, primary_key = True)
should_notify = db.Column(db.Boolean, default = True) # Allows a subscriber to unsubscribe
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
request_id = db.Column(db.Integer, db.ForeignKey('request.id'))
date_created = db.Column(db.DateTime)
owner_id = db.Column(db.Integer, db.ForeignKey('owner.id')) # Not null if responsible for fulfilling a part of the request
def __init__(self, request_id, user_id, creator = False):
self.user_id = user_id
self.request_id = request_id
self.date_created = datetime.now().isoformat()
def __repr__(self):
return '<Subscriber %r>' %self.user_id
class Owner(db.Model):
# A member of city staff assigned to a particular request, that may or may not upload records towards that request.
__tablename__ = 'owner'
id = db.Column(db.Integer, primary_key=True)
user_id = Column(Integer, ForeignKey('user.id'))
user = relationship("User", uselist=False)
request_id = db.Column(db.Integer, db.ForeignKey('request.id'))
request = relationship("Request", foreign_keys=[request_id])
active = db.Column(
db.Boolean, default=True
) # Indicate whether they're still involved in the request or not.
reason = db.Column(db.String()) # Reason they were assigned
reason_unassigned = db.Column(db.String()) # Reason they were unassigned
date_created = db.Column(db.DateTime)
date_updated = db.Column(db.DateTime)
is_point_person = db.Column(db.Boolean)
def __init__(self,
request_id,
user_id,
reason=None,
is_point_person=False):
self.reason = reason
self.user_id = user_id
self.request_id = request_id
self.date_created = datetime.now().isoformat()
self.date_updated = self.date_created
self.is_point_person = is_point_person
def __repr__(self):
return '<Owner %r>' % self.id
class Record(db.Model):
# A record that is attached to a particular request. A record can be online (uploaded document, link) or offline.
__tablename__ = 'record'
id = db.Column(db.Integer, primary_key=True)
date_created = db.Column(db.DateTime)
user_id = db.Column(
db.Integer, db.ForeignKey('user.id')
) # The user who uploaded the record, right now only city staff can
doc_id = db.Column(
db.Integer
) # The document ID. Currently using Scribd API to upload documents.
request_id = db.Column(db.Integer, db.ForeignKey(
'request.id')) # The request this record was uploaded for
description = db.Column(
db.String(400)) # A short description of what the record is.
filename = db.Column(
db.String(400)) # The original name of the file being uploaded.
url = db.Column(db.String()) # Where it exists on the internet.
download_url = db.Column(
db.String()) # Where it can be downloaded on the internet.
access = db.Column(db.String(
)) # How to access it. Probably only defined on offline docs for now.
def __init__(self,
request_id,
user_id,
url=None,
filename=None,
doc_id=None,
description=None,
access=None):
self.doc_id = doc_id
self.request_id = request_id
self.user_id = user_id
self.date_created = datetime.now().isoformat()
self.description = description
self.url = url
self.filename = filename
self.access = access
def __repr__(self):
return '<Record %r>' % self.description
class Subscriber(db.Model):
# A person subscribed to a request, who may or may not have created the request, and may or may not own a part of the request.
__tablename__ = 'subscriber'
id = db.Column(db.Integer, primary_key=True)
should_notify = db.Column(
db.Boolean, default=True) # Allows a subscriber to unsubscribe
user_id = db.Column(db.Integer, db.ForeignKey('user.id'))
user = relationship("User", uselist=False, lazy='joined')
request_id = db.Column(db.Integer, db.ForeignKey('request.id'))
date_created = db.Column(db.DateTime)
owner_id = db.Column(
db.Integer, db.ForeignKey('owner.id')
) # Not null if responsible for fulfilling a part of the request. UPDATE 6-11-2014: This isn't used. we should get rid of it.
def __init__(self, request_id, user_id, creator=False):
self.user_id = user_id
self.request_id = request_id
self.date_created = datetime.now().isoformat()
def __repr__(self):
return '<Subscriber %r>' % self.user_id
def __str__(self):
return str(self.user)
class Request(db.Model):
# The public records request
__tablename__ = 'request'
id = db.Column(db.Integer, primary_key =True)
date_created = db.Column(db.DateTime)
extended = db.Column(db.Boolean, default = False) # Has the due date been extended?
qas = relationship("QA", cascade="all,delete", order_by = "QA.date_created.desc()") # The list of QA units for this request
status_updated = db.Column(db.DateTime)
text = db.Column(db.String(), unique=True) # The actual request text.
subscribers = relationship("Subscriber", cascade ="all, delete") # The list of subscribers following this request.
owners = relationship("Owner", cascade="all,delete") # The list of city staff ever assigned to the request.
current_owner = db.Column(db.Integer) # The Owner ID for the city staff that currently 'owns' the request.
records = relationship("Record", cascade="all,delete", order_by = "Record.date_created.desc()") # The list of records that have been uploaded for this request.
notes = relationship("Note", cascade="all,delete", order_by = "Note.date_created.desc()") # The list of notes appended to this request.
status = db.Column(db.String(400)) # The status of the request (open, closed, etc.)
creator_id = db.Column(db.Integer, db.ForeignKey('user.id')) # If city staff created it on behalf of the public, otherwise the creator is the subscriber with creator = true
department = db.Column(db.String())
def __init__(self, text, creator_id = None, department = None):
self.text = text
self.date_created = datetime.now().isoformat()
self.creator_id = creator_id
self.department = department
def __repr__(self):
return '<Request %r>' % self.text
class Request(db.Model):
# The public records request
__tablename__ = 'request'
id = db.Column(db.Integer, primary_key=True)
date_created = db.Column(db.DateTime)
due_date = db.Column(db.DateTime)
extended = db.Column(db.Boolean,
default=False) # Has the due date been extended?
qas = relationship("QA",
cascade="all,delete",
order_by="QA.date_created.desc()"
) # The list of QA units for this request
status_updated = db.Column(db.DateTime)
text = db.Column(db.String(), unique=True) # The actual request text.
owners = relationship("Owner",
cascade="all, delete",
order_by="Owner.date_created.asc()")
subscribers = relationship(
"Subscriber", cascade="all, delete"
) # The list of subscribers following this request.
records = relationship(
"Record", cascade="all,delete", order_by="Record.date_created.desc()"
) # The list of records that have been uploaded for this request.
notes = relationship("Note",
cascade="all,delete",
order_by="Note.date_created.desc()"
) # The list of notes appended to this request.
status = db.Column(
db.String(400)) # The status of the request (open, closed, etc.)
creator_id = db.Column(
db.Integer, db.ForeignKey('user.id')
) # If city staff created it on behalf of the public, otherwise the creator is the subscriber with creator = true
department_id = db.Column(db.Integer, db.ForeignKey("department.id"))
department = relationship("Department", uselist=False)
date_received = db.Column(db.DateTime)
offline_submission_type = db.Column(db.String())
def __init__(self,
text,
creator_id=None,
offline_submission_type=None,
date_received=None):
self.text = text
self.date_created = datetime.now().isoformat()
self.creator_id = creator_id
self.offline_submission_type = offline_submission_type
if date_received and type(date_received) is datetime:
self.date_received = date_received
def __repr__(self):
return '<Request %r>' % self.text
def set_due_date(self):
if not self.date_received:
self.date_received = self.date_created
if self.extended == True:
self.due_date = self.date_received + timedelta(
days=int(app.config['DAYS_AFTER_EXTENSION']))
else:
self.due_date = self.date_received + timedelta(
days=int(app.config['DAYS_TO_FULFILL']))
def extension(self):
self.extended = True
self.due_date = self.due_date + timedelta(
days=int(app.config['DAYS_AFTER_EXTENSION']))
def point_person(self):
for o in self.owners:
if o.is_point_person:
return o
return None
def all_owners(self):
all_owners = []
for o in self.owners:
all_owners.append(o.user.get_alias())
return all_owners
def requester(self):
if self.subscribers:
return self.subscribers[
0] or None # The first subscriber is always the requester
return None
def requester_name(self):
requester = self.requester()
if requester and requester.user:
return requester.user.get_alias()
return "N/A"
def requester_phone(self):
requester = self.requester()
if requester and requester.user:
return requester.user.get_phone()
return "N/A"
def point_person_name(self):
point_person = self.point_person()
if point_person and point_person.user:
return point_person.user.get_alias()
return "N/A"
def department_name(self):
if self.department:
return self.department.get_name()
return "N/A"
def is_closed(self):
if self.status:
return re.match('.*(closed).*', self.status,
re.IGNORECASE) is not None
else:
app.logger.info("\n\n Request with this ID has no status: %s" %
self.id)
return False
def solid_status(self, cron_job=False):
if self.is_closed():
return "closed"
else:
if cron_job or (not current_user.is_anonymous()):
if self.due_date:
if datetime.now() >= self.due_date:
return "overdue"
elif (datetime.now() +
timedelta(days=int(app.config['DAYS_UNTIL_OVERDUE']))
) >= self.due_date:
return "due soon"
return "open"
@hybrid_property
def open(self):
two_days = datetime.now() + timedelta(days=2)
return and_(~self.closed, self.due_date > two_days)
@hybrid_property
def due_soon(self):
two_days = datetime.now() + timedelta(days=2)
return and_(self.due_date < two_days, self.due_date > datetime.now(),
~self.closed)
@hybrid_property
def overdue(self):
return and_(self.due_date < datetime.now(), ~self.closed)
@hybrid_property
def closed(self):
return Request.status.ilike("%closed%")