Coverage for demo_server/database/models.py: 89%

1# Copyright (c) Microsoft Corporation. 

2# Licensed under the MIT License. 

3 

4# The examples in this file come from the Flask-SQLAlchemy documentation 

5# For more information take a look at: 

6# http://flask-sqlalchemy.pocoo.org/2.1/quickstart/#simple-relationships 

7import os,binascii 

8 

9from datetime import datetime 

10from flask_sqlalchemy import SQLAlchemy 

11 

12db = SQLAlchemy() 

13 

14 

15class Post(db.Model): 

16 id = db.Column(db.Integer, primary_key=True) 

17 body = db.Column(db.Text) 

18 checksum = db.Column(db.Text) 

19 

20 def __init__(self, body): 

21 self.body = body 

22 self.checksum = binascii.b2a_hex(os.urandom(100))[:5] 

23 

24class Category(db.Model): 

25 categoryId = db.Column(db.Integer, primary_key=True) 

26 name = db.Column(db.String(50)) 

27 

28 def __init__(self, name): 

29 self.name = name 

30 

31 def __repr__(self): 

32 return '<Category %r>' % self.name