62 lines
1.5 KiB
Python
62 lines
1.5 KiB
Python
|
from typing import Tuple
|
||
|
|
import os
|
||
|
|
|
||
|
|
from ig import ctxPictuterProma, IG
|
||
|
|
from db import DBHelper
|
||
|
|
from utils import fileProcessor
|
||
|
|
from utils.const import DEBUG
|
||
|
|
|
||
|
|
# returns (errmsg | code, errcode)
|
||
|
|
def upload(aid:int) -> Tuple[str, int]:
|
||
|
|
# 抓取文章本體
|
||
|
|
article = DBHelper.solo_article_fetcher(key = aid)
|
||
|
|
if article is None:
|
||
|
|
return "Post not found", 1
|
||
|
|
|
||
|
|
# 抓取檔案
|
||
|
|
files = [
|
||
|
|
DBHelper.solo_file_fetcher(id = k)
|
||
|
|
for k in article["files"]
|
||
|
|
]
|
||
|
|
if None in files:
|
||
|
|
return "File not found", 1
|
||
|
|
|
||
|
|
# 轉出暫存檔案
|
||
|
|
tmp_path:list = []
|
||
|
|
for t in files:
|
||
|
|
filename, err = fileProcessor.file_saver(t.get("type"), t.get("binary"))
|
||
|
|
if err: # 如果錯誤
|
||
|
|
return filename, 1
|
||
|
|
tmp_path.append(filename)
|
||
|
|
|
||
|
|
# 合成文字圖
|
||
|
|
proma_file = ctxPictuterProma.new_proma(article["ctx"])
|
||
|
|
tmp_path = [proma_file] + tmp_path
|
||
|
|
|
||
|
|
# 送交 IG 上傳
|
||
|
|
if not DEBUG:
|
||
|
|
media = IG.upload_media(article["ctx"], tmp_path)
|
||
|
|
if media is None:
|
||
|
|
return "Upload failed", 1
|
||
|
|
else:
|
||
|
|
media = {"code":"fake_data"}
|
||
|
|
|
||
|
|
# 刪除檔案
|
||
|
|
for t in tmp_path:
|
||
|
|
os.remove(t)
|
||
|
|
|
||
|
|
return media["code"], 0
|
||
|
|
|
||
|
|
|
||
|
|
# return (errmsg, code)
|
||
|
|
def remove(aid:int) -> Tuple[str, int]:
|
||
|
|
# 抓取文章本體
|
||
|
|
article = DBHelper.solo_article_fetcher(key = aid)
|
||
|
|
if article is None:
|
||
|
|
return "Post not found", 1
|
||
|
|
|
||
|
|
err = IG.delete_media(article["igid"])
|
||
|
|
if err:
|
||
|
|
return "Remove failed", 1
|
||
|
|
|
||
|
|
return "OK", 0
|