252 lines
10 KiB
Python
252 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
||
#https://docs-python.ru/packages/telegram-klient-telethon-python/ <-info
|
||
import asyncio
|
||
|
||
from datetime import datetime, timedelta
|
||
#from telethon.sync import TelegramClient
|
||
from telethon import TelegramClient, events, utils
|
||
|
||
import os
|
||
import re
|
||
import random
|
||
#import pytz
|
||
import time
|
||
|
||
import pymysql
|
||
import pymysql.cursors
|
||
|
||
import sqlite3
|
||
|
||
#Название сессии
|
||
sessdb = 'tl-ub'
|
||
#Api ID и Api Hash полученные на my.telegram.org
|
||
|
||
api_id = 00000000
|
||
api_hash = 'blahblahblahblahblahblahblahblah'
|
||
timezone = "Europe/Kiev"
|
||
|
||
db_pymysql = True#set True or False
|
||
db_sqlite3 = True#set True or False
|
||
|
||
async def main():
|
||
async with TelegramClient(sessdb,api_id,api_hash) as client:
|
||
client.parse_mode="HTML"
|
||
print('User-Bot started')
|
||
me= await client.get_me()
|
||
my_id = int(me.id)
|
||
my_fn = me.first_name
|
||
print(my_id)
|
||
|
||
if db_pymysql:
|
||
con = pymysql.connect(host='localhost',
|
||
user='root',
|
||
password='V3rY$tR0NgPaS$Sw0Rd',
|
||
db='db',
|
||
charset='utf8mb4',
|
||
cursorclass=pymysql.cursors.DictCursor)
|
||
d = con.cursor()
|
||
d.execute('''CREATE TABLE IF NOT EXISTS `tg_iris_zarazy` (
|
||
`when_int` int(11) unsigned NOT NULL DEFAULT '0',
|
||
`who_id` bigint(20) unsigned NOT NULL DEFAULT '0',
|
||
`user_id` bigint(20) unsigned NOT NULL DEFAULT '0',
|
||
`u_link` varchar(500) NOT NULL DEFAULT '',
|
||
`bio_str` varchar(11) NOT NULL DEFAULT '1',
|
||
`bio_int` int(11) unsigned NOT NULL DEFAULT '1',
|
||
`expr_int` int(11) unsigned NOT NULL DEFAULT '0',
|
||
`expr_str` varchar(11) NOT NULL DEFAULT '0',
|
||
UNIQUE KEY `UNIQUE` (`who_id`,`user_id`)
|
||
);''');
|
||
con.commit()
|
||
d.execute('''CREATE TABLE IF NOT EXISTS `tg_users_url` (
|
||
`user_id` bigint(20) unsigned NOT NULL DEFAULT '0',
|
||
`when_int` int(11) unsigned NOT NULL DEFAULT '0',
|
||
`u_link` varchar(64) NOT NULL DEFAULT '',
|
||
`f_name` text NOT NULL,
|
||
PRIMARY KEY (`user_id`),
|
||
UNIQUE KEY (`u_link`)
|
||
);''');
|
||
con.commit()
|
||
|
||
if db_sqlite3:
|
||
conn = sqlite3.connect(f"{my_id}.sqlite")#покласти базу рядом?
|
||
#conn = sqlite3.connect(f"D:\\Misc\\projects\\Python\\ub4tg_db\\{my_id}.sqlite")#Або повністю
|
||
c = conn.cursor()
|
||
c.execute('''CREATE TABLE IF NOT EXISTS zarazy (
|
||
user_id INTEGER NOT NULL DEFAULT 0 UNIQUE,
|
||
when_int INTEGER NOT NULL DEFAULT 0,
|
||
bio_str VARCHAR NOT NULL DEFAULT 1,
|
||
bio_int INTEGER NOT NULL DEFAULT 1,
|
||
expr_int INTEGER NOT NULL DEFAULT 0,
|
||
expr_str VARCHAR NOT NULL DEFAULT 0
|
||
)''');
|
||
conn.commit()
|
||
####################################################################
|
||
async def get_id(url):
|
||
user_id = 0
|
||
if "tg://openmessage?user_id=" in url:
|
||
user_id = int(re.findall(r'user_id=([0-9]+)',url)[0])
|
||
print(user_id)
|
||
return user_id
|
||
if "t.me/" in url:
|
||
if db_pymysql:
|
||
try:
|
||
d.execute("SELECT * FROM `tg_users_url` WHERE `u_link` = '%s' ORDER BY `when_int` DESC" % str(url));
|
||
user = d.fetchone();
|
||
if user is None:
|
||
#print(f'не знайшли {url} у `tg_users_url`')
|
||
pass
|
||
else:
|
||
user_id = int(user['user_id'])
|
||
print(f'{url} in db: @{user_id}')
|
||
except Exception as Err:
|
||
print(f'E:{Err}/S {url} у `tg_users_url`')
|
||
if user_id==0:
|
||
try:
|
||
d.execute("SELECT * FROM `tg_iris_zarazy` WHERE `u_link` = '%s' ORDER BY `when_int` DESC" % str(url));
|
||
user = d.fetchone();
|
||
if user is None:
|
||
#print(f'не знайшли {url} у `tg_iris_zarazy`')
|
||
pass
|
||
else:
|
||
user_id = int(user['user_id'])
|
||
print(f'{url} in db: @{user_id}')
|
||
except Exception as Err:
|
||
print(f'E:{Err}/S {url} у `tg_iris_zarazy`')
|
||
if user_id==0:
|
||
try:
|
||
user_entity = await client.get_entity(url)
|
||
if user_entity.id:
|
||
user_id = int(user_entity.id)
|
||
user_fn = user_entity.first_name or ''
|
||
print(f'ok:{url}/@{user_id}')
|
||
if db_pymysql:
|
||
try:
|
||
d.execute("INSERT INTO `tg_users_url` (`when_int`,`user_id`,`u_link`,`f_name`) VALUES (%s,%s,%s,%s) ON DUPLICATE KEY UPDATE user_id = VALUES (user_id),u_link = VALUES (u_link),f_name = VALUES (f_name),when_int = VALUES (when_int);", (int(time.time()),int(user_id),str(url),str(user_fn))); con.commit()
|
||
except Exception as Err:
|
||
print(f'E:{Err}')
|
||
except Exception as Err:
|
||
print(f'E:{Err}')
|
||
#pass
|
||
return user_id
|
||
|
||
async def message_q( # спизжено
|
||
text: str,
|
||
user_id: int,
|
||
mark_read: bool = False,
|
||
delete: bool = False,
|
||
):
|
||
"""Отправляет сообщение и возращает ответ"""
|
||
async with client.conversation(user_id, exclusive=False) as conv:
|
||
msg = await conv.send_message(text)
|
||
response = await conv.get_response()
|
||
if mark_read:
|
||
await conv.mark_read()
|
||
return response
|
||
|
||
####################################################################
|
||
|
||
|
||
@client.on(events.NewMessage(pattern='.*подверг(ла)? заражению.*'))
|
||
async def podverg(event):
|
||
#хто там кого подверг(ла)
|
||
m = event.message
|
||
t = m.raw_text
|
||
irises = [707693258,5137994780,5226378684,5443619563,5434504334]
|
||
if m.sender_id not in irises:
|
||
#print(f"@{m.sender_id} не Iris!?");#Або це або pass. що краще?
|
||
pass
|
||
elif len(m.entities) > 1:
|
||
h= utils.sanitize_parse_mode('html').unparse(t,m.entities)#HTML
|
||
r= re.findall(r'🦠 <a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">.*</a> подверг.+<a href="(tg://openmessage\?user_id=\d+|https://t\.me/\w+)">',h)
|
||
if r:
|
||
u1url=r[0][0]
|
||
u2url=r[0][1]
|
||
u1id = await get_id(u1url)
|
||
u2id = await get_id(u2url)
|
||
#print(f'{u1url} [@{u1id}] подверг(ла) {u2url} [@{u2id}]')#показать
|
||
when=int(datetime.timestamp(m.date))
|
||
days=int(re.sub(r' ','',re.findall(r' на ([0-9\ ]+) д.*', t)[0]))
|
||
experience=re.findall(r"\+([0-9\.\,k]+) био-опыта", t)[0]
|
||
if ',' in experience:
|
||
experience=re.sub(r',', r'.',experience)
|
||
if 'k' in experience:
|
||
exp_int=int(float(re.sub('k', '',experience)) * 1000)
|
||
else:
|
||
exp_int=int(experience)
|
||
if 'Объект ещё не подвергался заражению вашим патогеном' in event.raw_text:
|
||
exp_int=int(re.sub(r' ','',re.findall(r'по ([0-9\ ]+) ед.*',event.raw_text)[0]))
|
||
a=datetime.utcfromtimestamp(when)+timedelta(days=int(days), hours=3)
|
||
do_int=datetime.timestamp(a)
|
||
do_txt=str(a.strftime("%d.%m.%y"))
|
||
if u1id > 0 and u2id > 0 and u1id != u2id:
|
||
if db_sqlite3 and u1id==my_id:
|
||
try:
|
||
c.execute("INSERT INTO zarazy(user_id,when_int,bio_str,bio_int,expr_int,expr_str) VALUES (?, ?, ?, ?, ?, ?)", (int(u2id),int(when),str(experience),int(exp_int),int(datetime.timestamp(a)),str(a.strftime("%d.%m.%y")))); conn.commit()
|
||
print('add/db.sqlite')
|
||
except:
|
||
try:
|
||
c.execute("UPDATE zarazy SET when_int = :wh, bio_str = :xp, bio_int = :xpi, expr_int = :end, expr_str = :do WHERE user_id = :z AND when_int <= :wh;", {"wh":int(when),"xp":str(experience),"xpi":int(exp_int),"end":int(datetime.timestamp(a)),"do":str(a.strftime("%d.%m.%y")),"z":int(u2id)}); conn.commit()
|
||
except Exception as Err:
|
||
print(f'err: {Err} zarazy')
|
||
if db_pymysql:
|
||
try:
|
||
d.execute("INSERT INTO `tg_iris_zarazy` (`who_id`, `user_id`, `when_int`, `bio_str`, `bio_int`, `expr_int`, `expr_str`, `u_link`) VALUES (%s,%s,%s,%s,%s,%s,%s,%s) ON DUPLICATE KEY UPDATE when_int=VALUES (when_int),bio_str=VALUES (bio_str),bio_int=VALUES (bio_int),expr_int=VALUES (expr_int),expr_str=VALUES (expr_str),u_link = VALUES (u_link);", (int(u1id),int(u2id),int(when),str(experience), int(exp_int), int(datetime.timestamp(a)),str(a.strftime("%d.%m.%y")),str(u2url))); con.commit()
|
||
print(f"\nINSERT INTO .... ON DUPLICATE KEY UPDATE # [@{u1id}] => [@{u2id}]\n")
|
||
except Exception as Err:
|
||
print(f'err: {Err} /localhost')
|
||
#pass
|
||
print(f'''{u1url} [@{u1id}] подверг(ла) {u2url} [@{u2id}] +{experience}''')#показать
|
||
|
||
|
||
@client.on(events.NewMessage(outgoing=True, pattern='.l2f'))
|
||
async def cmd_l2f(event): #Local->file/{id}.sqlite
|
||
msg='для успішного виконання повинно бути обидві бази True'
|
||
if db_pymysql:
|
||
try:
|
||
d.execute("SELECT * FROM `tg_iris_zarazy` WHERE who_id = %d ORDER BY when_int;" % int(my_id));
|
||
bz_info = d.fetchall()#получить
|
||
count=len(bz_info)
|
||
if count==0:
|
||
msg='🤷 інфа нема.'
|
||
print(msg)
|
||
else:
|
||
saved=0
|
||
for row in bz_info:
|
||
print(row)
|
||
id_user=int(row["user_id"])
|
||
bio_int=int(row["bio_int"])
|
||
bio_str=str(row["bio_str"])
|
||
when_int=int(row["when_int"])
|
||
expr_int=int(row["expr_int"])
|
||
expr_str=str(re.sub(r'.20', r'.',row["expr_str"])) #.2024->.24
|
||
user_url=str(f'tg://openmessage?user_id={id_user}') #fix для любителів мінять його
|
||
if db_sqlite3:
|
||
try:
|
||
c.execute("INSERT INTO zarazy (user_id,when_int,bio_str,bio_int,expr_int,expr_str) VALUES (?, ?, ?, ?, ?, ?)", (int(id_user),int(when_int),str(bio_str),int(bio_int),int(expr_int),str(expr_str))); conn.commit()
|
||
saved+=1
|
||
except:
|
||
try:
|
||
c.execute("UPDATE zarazy SET when_int = :wh, bio_str = :xp, bio_int = :xpi, expr_int = :end, expr_str = :do WHERE user_id = :z AND when_int <= :wh;", {"wh":int(when_int),"xp":str(bio_str),"xpi":int(bio_int),"end":int(expr_int),"do":str(expr_str),"z":int(id_user)}); conn.commit()
|
||
except Exception as Err:
|
||
print(f'err: {Err} zarazy')
|
||
msg=f"{saved} із {count}"
|
||
else:
|
||
msg='для успішного виконання повинно бути обидві бази True'
|
||
except Exception as Err:
|
||
print(f'err: {Err} zarazy localhost')
|
||
msg=Err
|
||
m=await event.reply(msg)
|
||
await asyncio.sleep(5)
|
||
await client.delete_messages(event.chat_id, [event.id, m.id])
|
||
|
||
|
||
@client.on(events.NewMessage(outgoing=True, pattern='.ping'))
|
||
async def cmd_ping(event):
|
||
# Say "pong!" whenever you send "!ping", then delete both messages
|
||
m = await event.reply('pong!')
|
||
await asyncio.sleep(5)
|
||
await client.delete_messages(event.chat_id, [event.id, m.id])
|
||
|
||
await client.run_until_disconnected()
|
||
|
||
asyncio.run(main())
|