261 lines
7.7 KiB
Python
261 lines
7.7 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
from datetime import datetime, timedelta
|
||
|
||
from aiogram import Bot, types
|
||
from aiogram.dispatcher import Dispatcher
|
||
from aiogram.utils import executor
|
||
|
||
from config import TOKEN
|
||
|
||
from const import REG_OK
|
||
|
||
import os
|
||
import re
|
||
import random
|
||
import time
|
||
|
||
import pymysql
|
||
import pymysql.cursors
|
||
|
||
import sqlite3
|
||
|
||
db_pymysql = True#set True or False
|
||
db_sqlite3 = True#set True or False
|
||
|
||
if db_sqlite3:
|
||
con=sqlite3.connect("db4tg.sqlite")
|
||
cur=con.cursor()
|
||
cur.execute('''CREATE TABLE IF NOT EXISTS users (
|
||
user_id INTEGER NOT NULL DEFAULT 0 UNIQUE,
|
||
reg_int INTEGER NOT NULL DEFAULT 0,
|
||
f_name VARCHAR NOT NULL DEFAULT 'хз',
|
||
mcoins INTEGER NOT NULL DEFAULT 1024,
|
||
rnd_kd INTEGER NOT NULL DEFAULT 0,
|
||
lng_code VARCHAR NOT NULL DEFAULT ''
|
||
)''');
|
||
|
||
if db_pymysql:
|
||
ldb=pymysql.connect(
|
||
host='localhost',
|
||
user='root',
|
||
password='V3rY$tR0Ng',
|
||
db='db',
|
||
charset='utf8mb4',
|
||
cursorclass=pymysql.cursors.DictCursor)
|
||
dbc = ldb.cursor()
|
||
|
||
async def reg_user(message: types.Message):
|
||
print(message)
|
||
user_id = int(message.from_user.id)
|
||
user_fn = message.from_user.first_name or ''
|
||
lng_code = message.from_user.language_code or ''
|
||
when_int = int(datetime.timestamp(message.date))
|
||
reg_date = 0
|
||
mcoins_c = 0
|
||
if db_sqlite3:
|
||
try:
|
||
cur.execute("INSERT OR IGNORE INTO users(user_id,reg_int,f_name,lng_code) VALUES (?,?,?,?)", (int(user_id),int(when_int),str(user_fn),str(lng_code))); con.commit()
|
||
except Exception as Err:
|
||
print(f"sqlite INSERT:{Err}")
|
||
|
||
sqlite_rd=when_int#for min (rd,rd)
|
||
sqlite_co=1024
|
||
try:
|
||
cur.execute("SELECT reg_int,mcoins FROM users WHERE user_id = %d" % int(user_id));
|
||
rd = cur.fetchone();
|
||
if rd is None:
|
||
print('не знайшли юзера у базі sqlite')
|
||
else:
|
||
sqlite_rd=int(rd[0])
|
||
sqlite_co=int(rd[1])
|
||
if sqlite_rd > 0:
|
||
reg_date=sqlite_rd
|
||
else:
|
||
reg_date=when_int
|
||
if sqlite_co > 0:
|
||
mcoins_c=sqlite_co
|
||
except Exception as Err:
|
||
print(f"sqlite SELECT:{Err}")
|
||
|
||
if db_pymysql:
|
||
try:
|
||
dbc.execute("INSERT INTO `tg_bot_users` (user_id,reg_int,f_name,lng_code) VALUES (%s,%s,%s,%s) ON DUPLICATE KEY UPDATE f_name=VALUES(f_name);",(int(user_id),int(when_int),str(user_fn),str(lng_code))); ldb.commit();
|
||
except Exception as Err:
|
||
print(f"localhost INSERT:{Err}")
|
||
pymysql_rd=when_int#for min (rd,rd)
|
||
pymysql_co=1024
|
||
try:
|
||
dbc.execute("SELECT reg_int,mcoins FROM `tg_bot_users` WHERE user_id = %d" % int(user_id));
|
||
rd = dbc.fetchone();
|
||
if rd is None:
|
||
print('не знайшли юзера у базі localhost')
|
||
else:
|
||
pymysql_rd=int(rd['reg_int'])
|
||
pymysql_co=int(rd['mcoins'])
|
||
if pymysql_rd > 0:
|
||
reg_date=pymysql_rd
|
||
else:
|
||
reg_date=when_int
|
||
if pymysql_co > 0:
|
||
mcoins_c=pymysql_co
|
||
except Exception as Err:
|
||
print(f"localhost SELECT:{Err}")
|
||
|
||
if db_sqlite3 and db_pymysql:
|
||
#якщо юзаємо обидві бази,то
|
||
reg_date=int(min(sqlite_rd,pymysql_rd))
|
||
mcoins_c=int(max(sqlite_co,pymysql_co))
|
||
|
||
if sqlite_rd < pymysql_rd or sqlite_co > pymysql_co:
|
||
#UPDATE `tg_bot_users` SET `reg_int`=?,`mcoins`=? WHERE `user_id`=?;
|
||
try:
|
||
dbc.execute(f"UPDATE `tg_bot_users` SET `reg_int` ='{reg_date}',`mcoins` ='{mcoins_c}' WHERE user_id = %d" % int(user_id)); ldb.commit()#як я хотів воно нехотіло, тому буде пока так.
|
||
except Exception as Err:
|
||
print(f"localhost UPDATE:{Err}")
|
||
|
||
if sqlite_rd > pymysql_rd or sqlite_co < pymysql_co:
|
||
try:
|
||
cur.execute("UPDATE users SET reg_int =?,mcoins=? WHERE user_id=?", (int(reg_date),int(mcoins_c),int(user_id))); con.commit()
|
||
except Exception as Err:
|
||
print(f"sqlite UPDATE:{Err}")
|
||
|
||
if reg_date==when_int:
|
||
print(REG_OK.get(lng_code, REG_OK['default']).format(user_fn=user_fn))
|
||
|
||
return reg_date
|
||
|
||
bot = Bot(token=TOKEN)
|
||
dp = Dispatcher(bot)
|
||
|
||
print('Bot started')
|
||
|
||
@dp.message_handler(commands=['start'])
|
||
async def process_start_command(message: types.Message):
|
||
#print(message)
|
||
snd_msg = "🖖"
|
||
user_id = int(message.from_user.id)
|
||
user_fn = message.from_user.first_name or ''
|
||
lng_code = message.from_user.language_code or ''
|
||
when_int = int(datetime.timestamp(message.date))
|
||
rd=int(await reg_user(message))#create or date
|
||
if rd == when_int:
|
||
snd_msg = REG_OK.get(lng_code, REG_OK['default']).format(user_fn=user_fn)
|
||
await message.answer(snd_msg)
|
||
|
||
@dp.message_handler(commands=['reg'])
|
||
async def cmd_reg(message: types.Message):
|
||
#print(message)
|
||
snd_msg = "🤷"
|
||
user_id = int(message.from_user.id)
|
||
user_fn = message.from_user.first_name or ''
|
||
lng_code = message.from_user.language_code or ''
|
||
when_int = int(datetime.timestamp(message.date))
|
||
rd=int(await reg_user(message))#create or date
|
||
if rd == when_int:
|
||
snd_msg = REG_OK.get(lng_code, REG_OK['default']).format(user_fn=user_fn)
|
||
elif rd > 0:
|
||
snd_msg = time.strftime('%d.%m.%Y', time.localtime(rd))
|
||
await message.answer(snd_msg)
|
||
|
||
@dp.message_handler(commands=['farm','ферма','random','rand','rnd'])
|
||
async def cmd_farm (message: types.Message):
|
||
user_id = int(message.from_user.id)
|
||
when_int = int(datetime.timestamp(message.date))
|
||
rd=await reg_user(message)#register_user#+sync
|
||
bal = int(42)#min bal
|
||
rkd = int(rd)#reg int
|
||
if db_sqlite3:
|
||
try:
|
||
cur.execute("SELECT mcoins,rnd_kd FROM users WHERE user_id = %d" % int(user_id));
|
||
rd = cur.fetchone();
|
||
if rd is None:
|
||
msg = "ERROR: user not registred in sqlite"
|
||
print(msg)
|
||
else:
|
||
bal = int(max(int(rd[0]),bal))
|
||
rkd = int(max(int(rd[1]),rkd))
|
||
except Exception as Err:
|
||
msg = Err
|
||
print(Err)
|
||
if db_pymysql:
|
||
try:
|
||
dbc.execute("SELECT mcoins,rnd_kd FROM `tg_bot_users` WHERE user_id = %d" % int(user_id));
|
||
rd = dbc.fetchone();
|
||
if rd is None:
|
||
print('не знайшли юзера у базі localhost')
|
||
print(msg)
|
||
else:
|
||
bal = int(max(int(rd['mcoins']),bal))
|
||
rkd = int(max(int(rd['rnd_kd']),rkd))
|
||
except Exception as Err:
|
||
msg = Err
|
||
print(Err)
|
||
|
||
if db_sqlite3 or db_pymysql:
|
||
# якщо вобще юзаєм базу
|
||
if when_int > rkd:
|
||
rnd = random.randint(-32,64)
|
||
if rnd > 0:
|
||
msg = f"✅ ok! +{rnd}"
|
||
rkd = rnd * 60
|
||
if rnd < 0:
|
||
msg = f"❎ ой! {rnd}"
|
||
rkd = (64-rnd) *32
|
||
if rnd == 0:
|
||
rnd = 100
|
||
rkd = rnd * 64
|
||
msg = f"✅ оу! +{rnd}"
|
||
bal+=rnd
|
||
if bal<10:
|
||
bal =10 #я сьодня добрьій.
|
||
msg=f"{msg}\n🤑 бл: {bal} \n⏱ кд: {rkd} сек"
|
||
rkd+=when_int
|
||
if db_sqlite3:
|
||
try:
|
||
cur.execute("UPDATE users SET mcoins = :bal, rnd_kd = :rkd WHERE user_id = :uid;",
|
||
{"rkd":int(rkd),"bal":int(bal),"uid":int(user_id)}); con.commit()
|
||
except Exception as Err:
|
||
msg = Err
|
||
print(Err)
|
||
if db_pymysql:
|
||
try:
|
||
dbc.execute(f"UPDATE `tg_bot_users` SET `rnd_kd` ='{rkd}',`mcoins` ='{bal}' WHERE user_id = %d" % int(user_id)); ldb.commit()#як я хотів воно нехотіло, тому буде пока так.
|
||
except Exception as Err:
|
||
msg = Err
|
||
print(Err)
|
||
else:
|
||
rkd = rkd-when_int
|
||
msg=f"\n ⏱ кд: {rkd} сек.\n🤑 бл: {bal}"
|
||
else:
|
||
rnd = random.randint(-32,100)
|
||
msg = rnd
|
||
await message.answer(msg)
|
||
|
||
@dp.message_handler(commands=['help'])
|
||
async def process_help_command(message: types.Message):
|
||
#await message.answer(emoji="🤷")
|
||
await message.answer('''
|
||
• 💬 /chats
|
||
• 🎲 /dice
|
||
• 🤑 /rnd
|
||
''')
|
||
|
||
@dp.message_handler(commands=['ping'])
|
||
async def process_ping_command(message: types.Message):
|
||
await message.reply("PONG!")
|
||
|
||
@dp.message_handler(commands=['dice','кубик'])
|
||
async def cmd_dice(message: types.Message):
|
||
await message.answer_dice(emoji="🎲")
|
||
|
||
@dp.message_handler(commands=['chats','чати','чаты','чаті'])
|
||
async def cmd_chats(message: types.Message):
|
||
await message.answer('''
|
||
• ☕ @misc_chat
|
||
• 🦠 @misc_games
|
||
• 🗃 @misc_files_v2
|
||
''')
|
||
|
||
if __name__ == '__main__':
|
||
executor.start_polling(dp)
|