-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmodels.py
More file actions
50 lines (38 loc) · 1.41 KB
/
models.py
File metadata and controls
50 lines (38 loc) · 1.41 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
from flask_sqlalchemy import SQLAlchemy
from marshmallow import Schema, fields, ValidationError,pre_load
import os
db = SQLAlchemy()
class Expense(db.Model):
expenseid = db.Column(db.Integer, autoincrement=True, primary_key=True)
email=db.Column(db.String, db.ForeignKey('user_detail.email'))
date = db.Column(db.String)
amount = db.Column(db.Float)
category = db.Column(db.String(80))
description = db.Column(db.String(80))
def __init__(self):
expenseid=self.expenseid,
date=self.date,
amount=self.amount,
category=self.category
class user_detail(db.Model):
uid = db.Column(db.Integer, autoincrement=True,primary_key=True)
uname = db.Column(db.String(30))
email = db.Column(db.String(30),primary_key=True)
username = db.Column(db.String(80))
password = db.Column(db.String(80))
class category(db.Model):
cid=db.Column(db.Integer, autoincrement=True,primary_key=True)
cname=db.Column(db.String(30),primary_key=True)
email=db.Column(db.String(50));
class ExpenseSchema(Schema):
expenseid = fields.Int(dump_only=True)
date = fields.Date()
amount = fields.Int()
category=fields.Str()
def format_name(self, Expense):
return "{}, {}".format(Expense.date, Expense.amount)
# Custom validator
def must_not_be_blank(data):
if not data:
raise ValidationError('Data not provided.')
expense_schema = ExpenseSchema()