103 lines
3.2 KiB
Python
103 lines
3.2 KiB
Python
|
from flask import Blueprint, current_app, request, jsonify
|
||
|
|
import hashlib
|
||
|
|
import time
|
||
|
|
from sqlalchemy.orm import sessionmaker
|
||
|
|
from sqlalchemy import desc
|
||
|
|
|
||
|
|
article = Blueprint('article', __name__)
|
||
|
|
|
||
|
|
# 匿名文列表
|
||
|
|
@article.route('/list', methods = ["GET"])
|
||
|
|
def listing():
|
||
|
|
# variables
|
||
|
|
rst = int(request.args.get("start"))
|
||
|
|
count = int(request.args.get("count"))
|
||
|
|
|
||
|
|
# db
|
||
|
|
db = current_app.shared_resource.engine
|
||
|
|
Session = sessionmaker(bind=db)
|
||
|
|
session = Session()
|
||
|
|
|
||
|
|
# get ctx
|
||
|
|
table = current_app.shared_resource.SQLarticle
|
||
|
|
res = session.query(table.id, table.ctx, table.igid, table.created_at, table.mark).order_by(desc(table.id)).filter(table.mark == 'visible').offset(rst).limit(count).all()
|
||
|
|
|
||
|
|
# mapping
|
||
|
|
res = [ {"id":r[0], "ctx":r[1], "igid":r[2], "created_at":r[3], "mark":r[4]} for r in res ]
|
||
|
|
|
||
|
|
return jsonify(res)
|
||
|
|
|
||
|
|
# 獲取指定文章
|
||
|
|
@article.route("/get/<int:id>", methods = ["GET"])
|
||
|
|
def getarticle(id:int):
|
||
|
|
db = current_app.shared_resource.engine
|
||
|
|
Session = sessionmaker(bind=db)
|
||
|
|
session = Session()
|
||
|
|
|
||
|
|
# get ctx
|
||
|
|
table = current_app.shared_resource.SQLarticle
|
||
|
|
res = session.query(table.id, table.ctx, table.igid, table.created_at, table.mark).filter(table.id == id).filter(table.mark == 'visible').all()
|
||
|
|
|
||
|
|
# mapping
|
||
|
|
res = [ {"id":r[0], "ctx":r[1], "igid":r[2], "created_at":r[3], "mark":r[4]} for r in res ]
|
||
|
|
|
||
|
|
return jsonify(res)
|
||
|
|
|
||
|
|
# 上傳文章
|
||
|
|
@article.route("/post", methods = ["POST"])
|
||
|
|
def posting():
|
||
|
|
db = current_app.shared_resource.engine
|
||
|
|
Session = sessionmaker(bind=db)
|
||
|
|
session = Session()
|
||
|
|
|
||
|
|
# content
|
||
|
|
ctx = str(request.json["ctx"])
|
||
|
|
if ctx is None: return "No brain no content"
|
||
|
|
|
||
|
|
# hash
|
||
|
|
seed = ctx + str(time.time())
|
||
|
|
hash = hashlib.sha256(seed.encode()).hexdigest()
|
||
|
|
|
||
|
|
# file processing
|
||
|
|
|
||
|
|
# ig posting
|
||
|
|
igid = None
|
||
|
|
|
||
|
|
# mark
|
||
|
|
mark = "visible"
|
||
|
|
|
||
|
|
# pg commit
|
||
|
|
table = current_app.shared_resource.SQLarticle
|
||
|
|
data = table(hash = hash, ctx = ctx, igid = igid, mark = mark)
|
||
|
|
session.add(data)
|
||
|
|
session.commit()
|
||
|
|
|
||
|
|
# pg getdata
|
||
|
|
res = session.query(table.id, table.ctx, table.igid, table.created_at, table.mark).filter(table.hash == hash).all()
|
||
|
|
res = [ {"id":r[0], "ctx":r[1], "igid":r[2], "created_at":r[3], "mark":r[4]} for r in res ]
|
||
|
|
|
||
|
|
session.close()
|
||
|
|
return jsonify(res)
|
||
|
|
|
||
|
|
# 只有發文者可以看到的獲取指定文章
|
||
|
|
# 只有發文者可以做到的刪除文章
|
||
|
|
@article.route("/own/<sha256>", methods = ["GET", "DELETE"])
|
||
|
|
def owner_getarticle(sha256:str):
|
||
|
|
db = current_app.shared_resource.engine
|
||
|
|
Session = sessionmaker(bind=db)
|
||
|
|
session = Session()
|
||
|
|
table = current_app.shared_resource.SQLarticle
|
||
|
|
|
||
|
|
# 獲取指定文章
|
||
|
|
if request.method == "GET":
|
||
|
|
res = session.query(table.id, table.ctx, table.igid, table.created_at, table.mark, table.hash).filter(table.hash == sha256).all()
|
||
|
|
res = [ {"id":r[0], "ctx":r[1], "igid":r[2], "created_at":r[3], "mark":r[4], "hash":r[5]} for r in res ]
|
||
|
|
return jsonify(res)
|
||
|
|
# 刪除指定文章
|
||
|
|
elif request.method == "DELETE":
|
||
|
|
res = session.query(table).filter(table.hash == sha256).first()
|
||
|
|
session.delete(res)
|
||
|
|
session.commit()
|
||
|
|
return "OK", 200
|
||
|
|
|
||
|
|
session.close()
|