summaryrefslogtreecommitdiff
path: root/maconomy/models.py
diff options
context:
space:
mode:
Diffstat (limited to 'maconomy/models.py')
-rw-r--r--maconomy/models.py25
1 files changed, 25 insertions, 0 deletions
diff --git a/maconomy/models.py b/maconomy/models.py
new file mode 100644
index 0000000..15fb46b
--- /dev/null
+++ b/maconomy/models.py
@@ -0,0 +1,25 @@
+
+class Employee:
+ def __init__(self, result):
+ self.id, self.name, self.email = result[:3]
+
+ def __unicode__(self):
+ return u"{} ({})".format(self.name, self.id)
+
+
+class Timesheet:
+ def __init__(self, result):
+ self.week, self.submitted, self.approved = result[:3]
+
+ def status(self):
+ return "submitted" if self.is_submitted() else "unsubmitted"
+ def is_submitted(self):
+ return self.submitted == 1
+ def is_missing(self):
+ return self.submitted is None
+ def is_approved(self):
+ return self.approved == 1
+ def __str__(self):
+ return self.__unicode__()
+ def __unicode__(self):
+ u"{} ({})".format(self.week, self.status)