Model
import json
import time
from utils import log
def save(data, path):
"""
data 是 dict 或者 list
path 是保存文件的路徑
"""
s = json.dumps(data, indent=2, ensure_ascii=False)
with open(path, 'w+', encoding='utf-8') as f:
# log('save', path, s, data)
f.write(s)
def load(path):
with open(path, 'r', encoding='utf-8') as f:
s = f.read()
# log('load', s)
return json.loads(s)
class Model(object):
"""
Model 是所有 model 的基類
@classmethod 是一個(gè)套路用法
例如
user = User()
user.db_path() 返回 User.txt
"""
@classmethod
def db_path(cls):
"""
cls 是類名, 誰調(diào)用的類名就是誰的
classmethod 有一個(gè)參數(shù)是 class(這里我們用 cls 這個(gè)名字)
所以我們可以得到 class 的名字
"""
classname = cls.__name__
path = '{}.txt'.format(classname)
return path
@classmethod
def load(cls, d):
m = cls({})
for k,v in d.items() :
setattr(m, k,v)
return m
@classmethod
def all(cls):
"""
all 方法(類里面的函數(shù)叫方法)使用 load 函數(shù)得到所有的 models
"""
path = cls.db_path()
models = load(path)
# 這里用了列表推導(dǎo)生成一個(gè)包含所有 實(shí)例 的 list
# m 是 dict, 用 cls(m) 可以初始化一個(gè) cls 的實(shí)例
# 不明白就 log 大法看看這些都是啥
ms = [cls.load(m) for m in models]
return ms
@classmethod
def find_by(cls, **kwargs):
"""
用法如下，kwargs 是只有一個(gè)元素的 dict
u = User.find_by(username='gua')
"""
log('kwargs, ', kwargs)
k, v = '', ''
for key, value in kwargs.items():
k, v = key, value
all = cls.all()
for m in all:
if v == m.__dict__[k]:
return m
return None
@classmethod
def find(cls, id):
return cls.find_by(id=id)
def __repr__(self):
"""
__repr__ 是一個(gè)魔法方法
簡單來說, 它的作用是得到類的 字符串表達(dá) 形式
比如 print(u) 實(shí)際上是 print(u.__repr__())
"""
classname = self.__class__.__name__
properties = ['{}: ({})'.format(k, v) for k, v in self.__dict__.items()]
s = '\n'.join(properties)
return '< {}\n{} \n>\n'.format(classname, s)
def save(self):
"""
用 all 方法讀取文件中的所有 model 并生成一個(gè) list
把 self 添加進(jìn)去并且保存進(jìn)文件
"""
# log('debug save')
models = self.all()
# log('models', models)
# 如果沒有 id，說明是新添加的元素
if self.id is None:
# 設(shè)置 self.id
# 先看看是否是空 list
if len(models) == 0:
# 我們讓第一個(gè)元素的 id 為 1（當(dāng)然也可以為 0）
self.id = 1
else:
m = models[-1]
# log('m', m)
self.id = m.id + 1
models.append(self)
else:
# index = self.find(self.id)
index = -1
for i, m in enumerate(models):
if m.id == self.id:
index = i
break
log('debug', index)
models[index] = self
l = [m.__dict__ for m in models]
path = self.db_path()
save(l, path)
class User(Model):
"""
User 是一個(gè)保存用戶數(shù)據(jù)的 model
現(xiàn)在只有兩個(gè)屬性 username 和 password
"""
def __init__(self, form):
self.id = form.get('id', None)
self.username = form.get('username', '')
self.password = form.get('password', '')
self.note = form.get('note', '')
def validate_login(self):
# return self.username == 'gua' and self.password == '123'
u = User.find_by(username=self.username)
return u is not None and u.password == self.password
def validate_register(self):
return len(self.username) > 2 and len(self.password) > 2
class Message(Model):
"""
Message 是用來保存留言的 model
"""
def __init__(self, form):
self.id = None
self.author = form.get('author', '')
self.message = form.get('message', '')
def test():
# users = User.all()
# u = User.find_by(username='gua')
# log('users', u)
form = dict(
username='gua',
password='gua',
)
u = User(form)
u.save()
log('u.id', u.id)
u3 = User.find(3)
u3.password = '123 789'
u3.save()
log('u3', u3)
log(User.all())
class Weibo(Model):
def __init__(self, form):
self.id = form.get('id',None)
self.content =form.get('content','')
self.created_time = int(time.time())
self.user_id = None
if __name__ == '__main__':
test()
路由 route
import json
import time
from utils import log
def save(data, path):
"""
data 是 dict 或者 list
path 是保存文件的路徑
"""
s = json.dumps(data, indent=2, ensure_ascii=False)
with open(path, 'w+', encoding='utf-8') as f:
# log('save', path, s, data)
f.write(s)
def load(path):
with open(path, 'r', encoding='utf-8') as f:
s = f.read()
# log('load', s)
return json.loads(s)
class Model(object):
"""
Model 是所有 model 的基類
@classmethod 是一個(gè)套路用法
例如
user = User()
user.db_path() 返回 User.txt
"""
@classmethod
def db_path(cls):
"""
cls 是類名, 誰調(diào)用的類名就是誰的
classmethod 有一個(gè)參數(shù)是 class(這里我們用 cls 這個(gè)名字)
所以我們可以得到 class 的名字
"""
classname = cls.__name__
path = '{}.txt'.format(classname)
return path
@classmethod
def load(cls, d):
m = cls({})
for k,v in d.items() :
setattr(m, k,v)
return m
@classmethod
def all(cls):
"""
all 方法(類里面的函數(shù)叫方法)使用 load 函數(shù)得到所有的 models
"""
path = cls.db_path()
models = load(path)
# 這里用了列表推導(dǎo)生成一個(gè)包含所有 實(shí)例 的 list
# m 是 dict, 用 cls(m) 可以初始化一個(gè) cls 的實(shí)例
# 不明白就 log 大法看看這些都是啥
ms = [cls.load(m) for m in models]
return ms
@classmethod
def find_by(cls, **kwargs):
"""
用法如下，kwargs 是只有一個(gè)元素的 dict
u = User.find_by(username='gua')
"""
log('kwargs, ', kwargs)
k, v = '', ''
for key, value in kwargs.items():
k, v = key, value
all = cls.all()
for m in all:
if v == m.__dict__[k]:
return m
return None
@classmethod
def find(cls, id):
return cls.find_by(id=id)
def __repr__(self):
"""
__repr__ 是一個(gè)魔法方法
簡單來說, 它的作用是得到類的 字符串表達(dá) 形式
比如 print(u) 實(shí)際上是 print(u.__repr__())
"""
classname = self.__class__.__name__
properties = ['{}: ({})'.format(k, v) for k, v in self.__dict__.items()]
s = '\n'.join(properties)
return '< {}\n{} \n>\n'.format(classname, s)
def save(self):
"""
用 all 方法讀取文件中的所有 model 并生成一個(gè) list
把 self 添加進(jìn)去并且保存進(jìn)文件
"""
# log('debug save')
models = self.all()
# log('models', models)
# 如果沒有 id，說明是新添加的元素
if self.id is None:
# 設(shè)置 self.id
# 先看看是否是空 list
if len(models) == 0:
# 我們讓第一個(gè)元素的 id 為 1（當(dāng)然也可以為 0）
self.id = 1
else:
m = models[-1]
# log('m', m)
self.id = m.id + 1
models.append(self)
else:
# index = self.find(self.id)
index = -1
for i, m in enumerate(models):
if m.id == self.id:
index = i
break
log('debug', index)
models[index] = self
l = [m.__dict__ for m in models]
path = self.db_path()
save(l, path)
class User(Model):
"""
User 是一個(gè)保存用戶數(shù)據(jù)的 model
現(xiàn)在只有兩個(gè)屬性 username 和 password
"""
def __init__(self, form):
self.id = form.get('id', None)
self.username = form.get('username', '')
self.password = form.get('password', '')
self.note = form.get('note', '')
def validate_login(self):
# return self.username == 'gua' and self.password == '123'
u = User.find_by(username=self.username)
return u is not None and u.password == self.password
def validate_register(self):
return len(self.username) > 2 and len(self.password) > 2
class Message(Model):
"""
Message 是用來保存留言的 model
"""
def __init__(self, form):
self.id = None
self.author = form.get('author', '')
self.message = form.get('message', '')
def test():
# users = User.all()
# u = User.find_by(username='gua')
# log('users', u)
form = dict(
username='gua',
password='gua',
)
u = User(form)
u.save()
log('u.id', u.id)
u3 = User.find(3)
u3.password = '123 789'
u3.save()
log('u3', u3)
log(User.all())
class Weibo(Model):
def __init__(self, form):
self.id = form.get('id',None)
self.content =form.get('content','')
self.created_time = int(time.time())
self.user_id = None
if __name__ == '__main__':
test()
