forked from mrlan/EnglishPal
refactor: refactor the model
parent
82896de336
commit
3e35679a91
|
@ -1,7 +1,7 @@
|
||||||
from pony.orm import *
|
from pony.orm import *
|
||||||
|
|
||||||
db = Database()
|
db = Database()
|
||||||
db.bind("sqlite", "./static/wordfreqapp.db", create_db=True) # bind sqlite file
|
db.bind("sqlite", "../static/wordfreqapp.db", create_db=True) # bind sqlite file
|
||||||
|
|
||||||
|
|
||||||
class User(db.Entity):
|
class User(db.Entity):
|
|
@ -0,0 +1,34 @@
|
||||||
|
from model import *
|
||||||
|
from datetime import datetime
|
||||||
|
|
||||||
|
def add_article(content, source="manual_input", level="5", question="No question"):
|
||||||
|
with db_session:
|
||||||
|
# add one article to sqlite
|
||||||
|
Article(
|
||||||
|
text=content,
|
||||||
|
source=source,
|
||||||
|
date=datetime.now().strftime("%-d %b %Y"), # format style of `5 Oct 2022`
|
||||||
|
level=level,
|
||||||
|
question=question,
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def delete_article_by_id(article_id):
|
||||||
|
article_id &= 0xFFFFFFFF # max 32 bits
|
||||||
|
with db_session:
|
||||||
|
article = Article.select(article_id=article_id)
|
||||||
|
if article:
|
||||||
|
article.first().delete()
|
||||||
|
|
||||||
|
|
||||||
|
def get_number_of_articles():
|
||||||
|
with db_session:
|
||||||
|
return len(Article.select()[:])
|
||||||
|
|
||||||
|
|
||||||
|
def get_page_articles(num, size):
|
||||||
|
with db_session:
|
||||||
|
return [
|
||||||
|
x
|
||||||
|
for x in Article.select().order_by(desc(Article.article_id)).page(num, size)
|
||||||
|
]
|
|
@ -0,0 +1,19 @@
|
||||||
|
from model import *
|
||||||
|
from Login import md5
|
||||||
|
|
||||||
|
def get_users():
|
||||||
|
with db_session:
|
||||||
|
return User.select().order_by(User.name)[:]
|
||||||
|
|
||||||
|
|
||||||
|
def update_password_by_username(username, password="123456"):
|
||||||
|
with db_session:
|
||||||
|
user = User.select(name=username)
|
||||||
|
if user:
|
||||||
|
user.first().password = md5(username + password)
|
||||||
|
|
||||||
|
def update_expiry_time_by_username(username, expiry_time="20230323"):
|
||||||
|
with db_session:
|
||||||
|
user = User.select(name=username)
|
||||||
|
if user:
|
||||||
|
user.first().expiry_date = expiry_time
|
Loading…
Reference in New Issue