bx-gitea-integration/app/main.py

68 lines
2.0 KiB
Python

import requests
import json
import re
from urllib.parse import parse_qs
from fastapi import FastAPI, Request
from app.constants import *
app = FastAPI()
tg = {
38: '@aarizona'
}
TOKEN = "7102273461:AAGi8RXCEvns2ONu3RcG9Up973vcLzTdCJg"
@app.get("/")
def read_root():
return {"service": "BX integrations"}
@app.post("/integration")
async def intgr(request: Request):
body = await request.json()
ref = body["ref"]
branch = ref.split("/")[-1]
[tag, number, *args] = branch.split("-")
logger.info(f"input tag {tag} number {number} args {args}")
bx_comment = (
f"Сообщение от GITEA для задачи {number} (ветка {body['ref']} в репозитории {body['repository']['full_name']})"
"\n\n"
f"{'Создана ветка' if not 'head_commit' in body else body['head_commit']['message']}"
)
bx_res = requests.get(
f"{WEBHOOK}/task.commentitem.add.json?"
f"taskId={number}"
f"&fields[POST_MESSAGE]={bx_comment}"
)
logger.info(f"result {json.loads(bx_res.text)}")
return {"status": "success"}
@app.post("/integration_tg")
async def tg_intgr_get(request: Request):
try:
body = await request.body()
query = parse_qs(body)
task_id = query[b"data[FIELDS_AFTER][TASK_ID]"][0].decode()
item_id = query[b"data[FIELDS_AFTER][ID]"][0].decode()
get_comment_hook = f"https://crm.svs-tech.pro/rest/38/7ufrqnfpncmt279p/task.commentitem.get.json?taskId={task_id}&ITEMID={item_id}"
data = requests.get(get_comment_hook)
data_json = data.json()
comment = data_json["result"]["POST_MESSAGE"]
mentions = re.findall(r"\[USER=(\d*)\]", comment)
for mention in mentions:
if tg[int(mention)]:
requests.get(f'https://api.telegram.org/bot{TOKEN}/sendMessage?chat_id={tg[int(mention)]}&text={comment}')
return {"status": "success"}
except:
logger.info(request.__dict__)
return {"status": "error"}