108 lines
3.6 KiB
Python
108 lines
3.6 KiB
Python
# %%
|
|
# ZED Python REPL
|
|
|
|
import requests
|
|
import os
|
|
import logging
|
|
import asyncio
|
|
import telegram
|
|
import re
|
|
from models import Notifikasi, Session
|
|
from sqlalchemy import and_, or_, func
|
|
from sqlalchemy.exc import IntegrityError
|
|
from dotenv import load_dotenv
|
|
from datetime import datetime
|
|
|
|
# Request ke kauaku
|
|
load_dotenv()
|
|
gts_api = os.getenv('GTS_API')
|
|
gts_token = os.getenv('GTS_TOKEN')
|
|
|
|
headers = {"Authorization": f"Bearer {gts_token}"}
|
|
|
|
session = Session()
|
|
|
|
TELE_BOT = os.getenv('TELE_BOT')
|
|
TELE_CHATID = os.getenv('TELE_CHATID')
|
|
bot = telegram.Bot(token=f'{TELE_BOT}')
|
|
|
|
async def getNotif():
|
|
req = requests.get(f'{gts_api}?limit=5', headers=headers)
|
|
|
|
if req.status_code == 200:
|
|
data = req.json()
|
|
try:
|
|
for notif in data:
|
|
type = notif['type']
|
|
inreply = notif['status']['in_reply_to_id']
|
|
postid = notif['id']
|
|
created = datetime.strptime(notif['created_at'], "%Y-%m-%dT%H:%M:%S.%fZ")
|
|
handler = notif['account']['acct']
|
|
username = notif['account']['display_name']
|
|
status = notif['status']['content']
|
|
url = notif['status']['url']
|
|
kirim = 'Belum'
|
|
|
|
if type == 'follow':
|
|
insert_follow = Notifikasi(post_id=postid, created_at=created, handler=handler, display_name=username, type=type, remark=kirim)
|
|
session.add(insert_follow)
|
|
else:
|
|
insert_mentions = Notifikasi(post_id=postid, created_at=created, handler=handler, display_name=username, type=type, status=status,url=url, remark=kirim)
|
|
|
|
session.add(insert_mentions)
|
|
session.commit()
|
|
|
|
except IntegrityError as err:
|
|
session.rollback()
|
|
if 'post_id' in str(err.orig):
|
|
print("Post Id sudah ada")
|
|
else:
|
|
print(f"Integritas data bermasalah", str(err))
|
|
finally:
|
|
session.close()
|
|
|
|
|
|
async def kirim_ke_tele():
|
|
cek_data = session.query(Notifikasi).filter(Notifikasi.remark.like('%elu%')).limit(10).all()
|
|
|
|
|
|
for notif in cek_data:
|
|
if str(notif.type) == 'follow':
|
|
try:
|
|
await bot.send_message(parse_mode='html',chat_id=str(TELE_CHATID), text=f"""
|
|
<b>{notif.display_name}</b>\n<i>{notif.handler}</i>\n{flag}
|
|
"""
|
|
)
|
|
except:
|
|
print('Ada masalah saat kirim ke Telebot')
|
|
else:
|
|
html_strip = re.compile('<.*?>')
|
|
status = str(notif.status)
|
|
status_text = re.sub(html_strip,'', status)
|
|
|
|
def icon_(type_):
|
|
if type_ == 'mention':
|
|
return '💬 mention your note'
|
|
elif type_ == 'reblog':
|
|
return '🚀 boost your note'
|
|
elif type_ == 'favourite':
|
|
return '💖 falling in love with your note'
|
|
elif type_ == 'follow':
|
|
return '✋ started to follow you'
|
|
else:
|
|
return None
|
|
|
|
flag = icon_(notif.type)
|
|
|
|
try:
|
|
await bot.send_message(parse_mode='html',chat_id=str(TELE_CHATID), text=f"""
|
|
<b>{notif.display_name}</b>\n<i>{notif.handler}</i>\n{flag}\n\n{status_text}\n\n{notif.url}
|
|
"""
|
|
)
|
|
except Exception as e:
|
|
print(f'Ada masalah saat kirim ke Telebot1', e)
|
|
|
|
|
|
|
|
asyncio.run(getNotif())
|
|
asyncio.run(kirim_ke_tele())
|