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
51
52
53
54
55
56
57
58
59
60
61
62
63
|
from hashlib import md5
import datetime
from flask_peewee.auth import BaseUser
from peewee import *
from app import db
class User(db.Model, BaseUser):
username = CharField()
password = CharField()
email = CharField()
join_date = DateTimeField(default=datetime.datetime.now)
active = BooleanField(default=True)
admin = BooleanField(default=False)
def __str__(self):
return self.username
def following(self):
return User.select().join(
Relationship, on=Relationship.to_user
).where(Relationship.from_user==self).order_by(User.username)
def followers(self):
return User.select().join(
Relationship, on=Relationship.from_user
).where(Relationship.to_user==self).order_by(User.username)
def is_following(self, user):
return Relationship.select().where(
Relationship.from_user==self,
Relationship.to_user==user
).exists()
def gravatar_url(self, size=80):
return 'http://www.gravatar.com/avatar/%s?d=identicon&s=%d' % \
(md5(self.email.strip().lower().encode('utf-8')).hexdigest(), size)
class Relationship(db.Model):
from_user = ForeignKeyField(User, related_name='relationships')
to_user = ForeignKeyField(User, related_name='related_to')
def __str__(self):
return 'Relationship from %s to %s' % (self.from_user, self.to_user)
class Message(db.Model):
user = ForeignKeyField(User)
content = TextField()
pub_date = DateTimeField(default=datetime.datetime.now)
def __str__(self):
return '%s: %s' % (self.user, self.content)
class Note(db.Model):
user = ForeignKeyField(User)
message = TextField()
status = IntegerField(choices=((1, 'live'), (2, 'deleted')), null=True)
created_date = DateTimeField(default=datetime.datetime.now)
|
