Compare commits
103 Commits
Author | SHA1 | Date | |
---|---|---|---|
9f7ae4b9a7 | |||
|
d41eeb3232 | ||
|
55e5303ab2 | ||
|
31235ef9ac | ||
|
6709eebb88 | ||
|
1533c5eb91 | ||
|
4c520b39fe | ||
|
db842a1ceb | ||
|
a571709afa | ||
|
192687a5ea | ||
|
f20c90ea34 | ||
|
1b2410afc4 | ||
|
56450bc566 | ||
|
aa3ea439fe | ||
|
dae3f05aa7 | ||
|
66c2ef80c3 | ||
|
4a3f799284 | ||
|
7090ab218e | ||
|
53bac66692 | ||
|
ee27dbfd7b | ||
|
14c238ec2f | ||
|
9d6d8791fd | ||
|
d2b8c76817 | ||
|
48d53fd1b9 | ||
|
505d90aea8 | ||
|
eb59091615 | ||
|
c8a9936a00 | ||
|
ecbe419abb | ||
|
a58f6c404f | ||
|
e677591b97 | ||
|
c734044b71 | ||
|
e55b1c54d8 | ||
|
21bc3724c8 | ||
|
52f9edf9f4 | ||
|
ef4c806071 | ||
|
b188edd98d | ||
|
bb91a90e9e | ||
|
d6da239718 | ||
|
deac88534a | ||
|
50504b14ce | ||
|
3965b11643 | ||
|
4e128198b7 | ||
|
0cf7423806 | ||
|
28d0e6fdbf | ||
|
49ac3b7a48 | ||
|
2b7cbb0664 | ||
|
bba000877e | ||
|
6c22fa72d1 | ||
|
1a6d0e7e91 | ||
|
9315a123f1 | ||
|
1f31c5155b | ||
|
fc437530e4 | ||
|
cbd8e13d79 | ||
|
8fc70c267d | ||
|
f49233b811 | ||
|
3831d4e8d4 | ||
|
0278189ded | ||
|
a842b3ae01 | ||
|
d427d189b1 | ||
|
08cc6b40c1 | ||
|
ba0867f662 | ||
|
c764b71af0 | ||
|
b9cd608d9b | ||
|
77b6293ea6 | ||
|
8c269fd768 | ||
|
256bdfb89b | ||
|
b5903bf79f | ||
|
11ddb1536c | ||
|
19999b28ad | ||
|
e8190a500d | ||
|
76d14126a1 | ||
|
345168aa80 | ||
|
e2467803c5 | ||
|
489a3304a4 | ||
|
456d0619df | ||
|
360f8d4c55 | ||
|
c39f1cde04 | ||
|
3f04de5cb1 | ||
|
53c48dbb44 | ||
|
361648f1c7 | ||
|
2de30513b2 | ||
|
f46ba32473 | ||
|
8d599c239b | ||
|
9403350e15 | ||
|
9f27070585 | ||
|
40927b95cc | ||
|
460fccc6ab | ||
|
1e4e6fa4d6 | ||
|
c4ca3a98a1 | ||
|
25e82e28d5 | ||
|
e93d2c0ef7 | ||
|
b673b43c93 | ||
|
c02215574d | ||
|
dea7805dff | ||
|
40abc7065d | ||
|
c02458b209 | ||
fe64756bc4 | |||
6923cd2aa8 | |||
b73468c73f | |||
671c39be27 | |||
b9813528b1 | |||
293d90fb29 | |||
9432f67ada |
16
.drone.yml
16
.drone.yml
@ -15,11 +15,23 @@ steps:
|
|||||||
- name: test it first
|
- name: test it first
|
||||||
commands:
|
commands:
|
||||||
- /usr/bin/python3 -m py_compile bot.py
|
- /usr/bin/python3 -m py_compile bot.py
|
||||||
|
- /usr/bin/python3 pipisa_functions/pipisa_time.py
|
||||||
- name: pull into the folduh
|
- name: pull into the folduh
|
||||||
commands:
|
commands:
|
||||||
- cd /usr/share/$DRONE_REPO_NAME
|
- cd /usr/share/$DRONE_REPO_NAME
|
||||||
- git pull
|
- su -c "git pull" jack
|
||||||
- chown -R jack:jack .
|
#- chown -R jack:jack .
|
||||||
|
when:
|
||||||
|
branch:
|
||||||
|
- main
|
||||||
|
environment:
|
||||||
|
ARANGO_PASSWORD:
|
||||||
|
from_secret: arango-pwd
|
||||||
|
ARANGO_USERNAME:
|
||||||
|
from_secret: arango-usr
|
||||||
- name: restart systemd unit
|
- name: restart systemd unit
|
||||||
commands:
|
commands:
|
||||||
- systemctl restart jack_bot
|
- systemctl restart jack_bot
|
||||||
|
when:
|
||||||
|
branch:
|
||||||
|
- main
|
||||||
|
13
README.md
Normal file
13
README.md
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
Пробуем проверять что-либо.
|
||||||
|
|
||||||
|
```mermaid
|
||||||
|
journey
|
||||||
|
title Пиписа
|
||||||
|
section Наш Чат
|
||||||
|
Оче больше пипис: 5: ГЕГЕМОН
|
||||||
|
Обычно большепенис: 3: Догоняющие
|
||||||
|
Скромный дик: 1: Отставшие
|
||||||
|
section Не наш чат
|
||||||
|
Не запустил: -1: Отставшие
|
||||||
|
Не крутил: 1: Отставшие
|
||||||
|
```
|
0
__init__.py
Normal file
0
__init__.py
Normal file
46
bot.py
46
bot.py
@ -1,29 +1,53 @@
|
|||||||
import logging
|
import logging
|
||||||
import graypy
|
from aiogram import executor, Bot, Dispatcher
|
||||||
|
|
||||||
from aiogram import executor
|
|
||||||
|
|
||||||
from create_bot import dp
|
from create_bot import dp
|
||||||
|
|
||||||
import handlers
|
import handlers
|
||||||
|
import json, logging
|
||||||
|
from global_conf import CONFIG
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def remake_field_generator(d, field='text'):
|
||||||
|
if isinstance(d, list):
|
||||||
|
for k2 in d:
|
||||||
|
for id_val in remake_field_generator(k2):
|
||||||
|
yield id_val
|
||||||
|
elif isinstance(d, dict):
|
||||||
|
for k, v in d.items():
|
||||||
|
if k == field:
|
||||||
|
d[k] = '<blank!>'
|
||||||
|
yield v
|
||||||
|
elif isinstance(v, list):
|
||||||
|
for k2 in v:
|
||||||
|
for id_val in remake_field_generator(k2):
|
||||||
|
yield id_val
|
||||||
|
elif isinstance(v, dict):
|
||||||
|
for id_val in remake_field_generator(v):
|
||||||
|
yield id_val
|
||||||
|
|
||||||
|
def filter_grammar_messages(record):
|
||||||
|
if record.args and (not None in record.args):
|
||||||
|
j = json.loads(record.args[2])
|
||||||
|
for _ in remake_field_generator(j):
|
||||||
|
pass
|
||||||
|
record.args = (record.args[0], record.args[1], json.dumps(j))
|
||||||
|
return True
|
||||||
|
|
||||||
logging.basicConfig(level=logging.DEBUG)
|
logging.basicConfig(level=logging.DEBUG)
|
||||||
|
logging.getLogger("aiogram").addFilter(filter_grammar_messages)
|
||||||
|
|
||||||
|
|
||||||
logging.basicConfig(
|
|
||||||
level = logging.DEBUG,
|
|
||||||
handlers = [ graypy.GELFUDPHandler('localhost', 12201) ]
|
|
||||||
)
|
|
||||||
|
|
||||||
# pipisa.register_handlers_pipisa(dp)
|
# pipisa.register_handlers_pipisa(dp)
|
||||||
# time_new_year.register_handlers_time(dp)
|
# time_new_year.register_handlers_time(dp)
|
||||||
# sendalarm.register_handlers_test(dp)
|
# sendalarm.register_handlers_test(dp)
|
||||||
# start_help.register_handlers_StartHelp(dp)
|
# start_help.register_handlers_StartHelp(dp)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
|
#print(CONFIG)
|
||||||
|
|
||||||
|
logging.info('Бот в строю.')
|
||||||
executor.start_polling(dp, skip_updates=True)
|
executor.start_polling(dp, skip_updates=True)
|
||||||
|
logging.info('Всем пока.')
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
19
config.yaml
Normal file
19
config.yaml
Normal file
@ -0,0 +1,19 @@
|
|||||||
|
---
|
||||||
|
databaso:
|
||||||
|
host: 'https://arango.guaranteedstruggle.host'
|
||||||
|
#user: root
|
||||||
|
#pass: stolendick527
|
||||||
|
base: pipisa
|
||||||
|
collection: dicks
|
||||||
|
posts_removal_collection: posts_removal
|
||||||
|
|
||||||
|
telegram_token: 6991523312:AAFpxPlWHfnNq2dVXM_v1GuHcrbNgWdSWBM
|
||||||
|
telegram_admins_ids:
|
||||||
|
- 317051301 # hogweed1
|
||||||
|
- 226765676 # blackjack
|
||||||
|
- 94830892 # viktorRMI
|
||||||
|
- 132002803
|
||||||
|
|
||||||
|
openai:
|
||||||
|
token: 5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc
|
||||||
|
api_key: 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 ## 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
|
@ -1,4 +1,8 @@
|
|||||||
from aiogram import Bot, Dispatcher
|
from aiogram import Bot, Dispatcher
|
||||||
API_TOKEN = '5724991559:AAEuLvpLsgP6LHRGMSyFtQLlR5qPQUO4b_w'
|
from global_conf import CONFIG
|
||||||
bot = Bot(token=API_TOKEN, parse_mode="HTML")
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
bot = Bot(token=CONFIG['telegram_token'], parse_mode="HTML")
|
||||||
dp = Dispatcher(bot)
|
dp = Dispatcher(bot)
|
0
db_logic/__init__.py
Normal file
0
db_logic/__init__.py
Normal file
44
db_logic/collections.py
Normal file
44
db_logic/collections.py
Normal file
@ -0,0 +1,44 @@
|
|||||||
|
|
||||||
|
from global_conf import CONFIG
|
||||||
|
|
||||||
|
#### https://docs.python-arango.com/en/main/
|
||||||
|
from arango import ArangoClient
|
||||||
|
|
||||||
|
import logging
|
||||||
|
import os
|
||||||
|
|
||||||
|
|
||||||
|
def get_dicks_collection():
|
||||||
|
try:
|
||||||
|
arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] )
|
||||||
|
pipisa_db = arango_client.db(
|
||||||
|
CONFIG['databaso']['base'],
|
||||||
|
username=os.environ['ARANGO_USR'],
|
||||||
|
password=os.environ['ARANGO_PWD'],
|
||||||
|
#username=CONFIG['databaso']['user'],
|
||||||
|
#password=CONFIG['databaso']['pass']
|
||||||
|
)
|
||||||
|
dicks_collection = pipisa_db.collection(CONFIG['databaso']['collection'])
|
||||||
|
|
||||||
|
return dicks_collection
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB при взятии коллекции пипис')
|
||||||
|
logging.error(e)
|
||||||
|
|
||||||
|
|
||||||
|
def get_posts_removal_collection():
|
||||||
|
try:
|
||||||
|
arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] )
|
||||||
|
pipisa_db = arango_client.db(
|
||||||
|
CONFIG['databaso']['base'],
|
||||||
|
username=os.environ['ARANGO_USR'],
|
||||||
|
password=os.environ['ARANGO_PWD'],
|
||||||
|
#username=CONFIG['databaso']['user'],
|
||||||
|
#password=CONFIG['databaso']['pass']
|
||||||
|
)
|
||||||
|
posts_removal_collection = pipisa_db.collection(CONFIG['databaso']['posts_removal_collection'])
|
||||||
|
|
||||||
|
return posts_removal_collection
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB при взятии коллекции постов-на-удаление')
|
||||||
|
logging.error(e)
|
75
db_logic/postcleaner.py
Normal file
75
db_logic/postcleaner.py
Normal file
@ -0,0 +1,75 @@
|
|||||||
|
from db_logic import collections
|
||||||
|
import datetime, logging
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def append_post_to_cleaning_sequence(message, type=None):
|
||||||
|
try:
|
||||||
|
novenkiy = {
|
||||||
|
'msg_id': message.message_id,
|
||||||
|
'chat_id': message.chat.id,
|
||||||
|
'type': type,
|
||||||
|
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
|
}
|
||||||
|
metadata = collections.get_posts_removal_collection().insert(novenkiy, overwrite_mode='update')
|
||||||
|
|
||||||
|
logging.debug('Успешно добавлен пост на удаление')
|
||||||
|
except Exception as e2:
|
||||||
|
logging.error('ошибка DB :: добавление нового поста на удаление')
|
||||||
|
logging.error(e2)
|
||||||
|
|
||||||
|
|
||||||
|
def get_posts_to_be_removed(chat_id, type=None, max_id=None):
|
||||||
|
# собираем
|
||||||
|
try:
|
||||||
|
#posts = [p for p in collections.get_posts_removal_collection().all( ) ]
|
||||||
|
|
||||||
|
if type:
|
||||||
|
posts = [p for p in collections.get_posts_removal_collection().find({'chat_id': chat_id, 'type': type}, skip=0, limit=1100) ]
|
||||||
|
else:
|
||||||
|
posts = [p for p in collections.get_posts_removal_collection().find({'chat_id': chat_id}, skip=0, limit=1100) ]
|
||||||
|
|
||||||
|
posts_ret = [ p for p in posts ]
|
||||||
|
# # for p in posts:
|
||||||
|
# # last_time = datetime.datetime.strptime(p['datetimes'], '%Y-%m-%d %H:%M:%S')
|
||||||
|
# # timediff = (datetime.datetime.now() - last_time).total_seconds()
|
||||||
|
|
||||||
|
# # if timediff > 60:
|
||||||
|
# # posts_ret.append(p)
|
||||||
|
|
||||||
|
if max_id:
|
||||||
|
posts_ret = [ p for p in posts_ret if p['msg_id'] != max_id]
|
||||||
|
else:
|
||||||
|
posts_ret = [ p for p in posts_ret if p['msg_id'] != max([pp['msg_id'] for pp in posts_ret])]
|
||||||
|
|
||||||
|
|
||||||
|
#### TODO удалять все предыдущие без учёта времени
|
||||||
|
|
||||||
|
return posts_ret
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB :: получение постов на удаление')
|
||||||
|
logging.error(e)
|
||||||
|
|
||||||
|
|
||||||
|
def del_post(msg_id, chat_id):
|
||||||
|
# удаляем из базы
|
||||||
|
try:
|
||||||
|
|
||||||
|
candidate_cursor = collections.get_posts_removal_collection().find(
|
||||||
|
{
|
||||||
|
'msg_id': msg_id,
|
||||||
|
'chat_id': chat_id,
|
||||||
|
},
|
||||||
|
skip = 0,
|
||||||
|
limit = 1488
|
||||||
|
)
|
||||||
|
if candidate_cursor.count() > 0:
|
||||||
|
pp = candidate_cursor.pop()
|
||||||
|
else:
|
||||||
|
pp = None
|
||||||
|
|
||||||
|
collections.get_posts_removal_collection().delete(pp)
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB :: удаление поста на удаление')
|
||||||
|
logging.error(e)
|
65
db_logic/tops.py
Normal file
65
db_logic/tops.py
Normal file
@ -0,0 +1,65 @@
|
|||||||
|
from global_conf import CONFIG
|
||||||
|
from db_logic import collections
|
||||||
|
import logging
|
||||||
|
|
||||||
|
|
||||||
|
def get_tops(top_ = False, glob_ = False, chat_id = None):
|
||||||
|
|
||||||
|
if top_:
|
||||||
|
try:
|
||||||
|
dicks = [d for d in collections.get_dicks_collection().find({'chat_id': chat_id}, skip=0, limit=1100)]
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB в /topdick')
|
||||||
|
logging.error(e)
|
||||||
|
elif glob_:
|
||||||
|
try:
|
||||||
|
dicks = [d for d in collections.get_dicks_collection().all( ) if d['user_id'] != d['chat_id']]
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB в /globaldick')
|
||||||
|
logging.error(e)
|
||||||
|
else:
|
||||||
|
logging.error('вызывают хз что!')
|
||||||
|
|
||||||
|
top_dicks = sorted(dicks, key=lambda dick: dick['dick_size'], reverse=True)
|
||||||
|
top_dicks = top_dicks[:( min(len(top_dicks), 10) )]
|
||||||
|
|
||||||
|
dickos = ''
|
||||||
|
emo = ['🏆','🚀','🍆','🍌','🐍','🐎','🌭','🌶','👌','💩']
|
||||||
|
|
||||||
|
if len(top_dicks) > 0:
|
||||||
|
for i in range(len(top_dicks)):
|
||||||
|
dickos += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n'
|
||||||
|
i += 1
|
||||||
|
|
||||||
|
return dickos
|
||||||
|
|
||||||
|
|
||||||
|
def get_antitops(top_ = False, glob_ = False, chat_id = None):
|
||||||
|
dicks = []
|
||||||
|
if top_:
|
||||||
|
try:
|
||||||
|
dicks = [d for d in collections.get_dicks_collection().find({'chat_id': chat_id}, skip=0, limit=1100)]
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB в /topdick')
|
||||||
|
logging.error(e)
|
||||||
|
elif glob_:
|
||||||
|
try:
|
||||||
|
dicks = [d for d in collections.get_dicks_collection().all() if d['user_id'] != d['chat_id']]
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB в /globaldick')
|
||||||
|
logging.error(e)
|
||||||
|
else:
|
||||||
|
logging.error('вызывают хз что!')
|
||||||
|
|
||||||
|
antitop_dicks = sorted(dicks, key=lambda dick: dick['dick_size'], reverse=False)
|
||||||
|
antitop_dicks = antitop_dicks[:( min(len(antitop_dicks), 10) )]
|
||||||
|
|
||||||
|
dickos = ''
|
||||||
|
emo = ['💩', '😞', '🤕', '😣', '🙁', '😐', '😑', '😶', '😒', '😕']
|
||||||
|
|
||||||
|
if len(antitop_dicks) > 0:
|
||||||
|
for i in range(len(antitop_dicks)):
|
||||||
|
dickos += f' {emo[i]} {i+1}. {antitop_dicks[i]["user_fullname"]}: {antitop_dicks[i]["dick_size"]}см\n'
|
||||||
|
i += 1
|
||||||
|
|
||||||
|
return dickos
|
86
db_logic/user_stuff.py
Normal file
86
db_logic/user_stuff.py
Normal file
@ -0,0 +1,86 @@
|
|||||||
|
from db_logic import collections
|
||||||
|
import datetime, logging
|
||||||
|
|
||||||
|
|
||||||
|
def store_new_user(message, result ):
|
||||||
|
|
||||||
|
try:
|
||||||
|
novenkiy = {
|
||||||
|
'user_id': message.from_user.id,
|
||||||
|
'user_fullname': message.from_user.full_name,
|
||||||
|
'dick_size': abs(result),
|
||||||
|
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||||
|
'attempts': 1,
|
||||||
|
'chat_id': message.chat.id
|
||||||
|
}
|
||||||
|
metadata = collections.get_dicks_collection().insert(novenkiy, overwrite_mode='update')
|
||||||
|
|
||||||
|
logging.debug(f'Успешно добавлен нового пользователь @{message.from_user.username}')
|
||||||
|
except Exception as e2:
|
||||||
|
logging.error(f'ошибка DB в /dick :: добавление нового пользователя @{message.from_user.username}')
|
||||||
|
logging.error(e2)
|
||||||
|
|
||||||
|
|
||||||
|
def update_attempts(message, user ):
|
||||||
|
try:
|
||||||
|
metadata = collections.get_dicks_collection().insert(
|
||||||
|
{
|
||||||
|
'_key': user['_key'], ### этот нужен чтобы апдейт прилетал тому же юзеру
|
||||||
|
'user_id': user['user_id'] ,
|
||||||
|
'user_fullname': message.from_user.full_name,
|
||||||
|
'dick_size': user['dick_size'] ,
|
||||||
|
'datetimes': user['datetimes'] ,
|
||||||
|
'attempts': user['attempts'] + 1,
|
||||||
|
'chat_id': message.chat.id
|
||||||
|
},
|
||||||
|
overwrite_mode='update'
|
||||||
|
)
|
||||||
|
|
||||||
|
logging.debug(f'Успешно апдейтнули попытку крутить пипису @{message.from_user.username}')
|
||||||
|
except Exception as e2:
|
||||||
|
logging.error(f'ошибка DB в /dick :: обновление попытки крутить пипису @{message.from_user.username}')
|
||||||
|
logging.error(e2)
|
||||||
|
|
||||||
|
|
||||||
|
def update_dick_size(message, user, updatedDick ):
|
||||||
|
|
||||||
|
try:
|
||||||
|
metadata = collections.get_dicks_collection().insert(
|
||||||
|
{
|
||||||
|
'_key': user['_key'], ### этот нужен чтобы апдейт прилетал тому же юзеру
|
||||||
|
'user_id': user['user_id'],
|
||||||
|
'user_fullname': message.from_user.full_name,
|
||||||
|
'dick_size': updatedDick,
|
||||||
|
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||||
|
'attempts': user['attempts'] + 1,
|
||||||
|
'chat_id': message.chat.id
|
||||||
|
},
|
||||||
|
overwrite_mode='update'
|
||||||
|
)
|
||||||
|
|
||||||
|
logging.info(f'Успешно апдейтнули пипису @{message.from_user.username}')
|
||||||
|
except Exception as e2:
|
||||||
|
logging.error(f'ошибка DB в /dick :: обновление пиписы @{message.from_user.username}')
|
||||||
|
logging.error(e2)
|
||||||
|
|
||||||
|
|
||||||
|
def get_user(message):
|
||||||
|
try:
|
||||||
|
#### Чекнуть есть ли юзер в базе
|
||||||
|
candidate_cursor = collections.get_dicks_collection().find(
|
||||||
|
{
|
||||||
|
'user_id': message.from_user.id,
|
||||||
|
'chat_id': message.chat.id
|
||||||
|
},
|
||||||
|
skip = 0,
|
||||||
|
limit = 1488
|
||||||
|
)
|
||||||
|
if candidate_cursor.count() > 0:
|
||||||
|
user = candidate_cursor.pop()
|
||||||
|
else:
|
||||||
|
user = None
|
||||||
|
|
||||||
|
return user
|
||||||
|
except Exception as e:
|
||||||
|
logging.error('ошибка DB в /dick :: поиск юзера')
|
||||||
|
logging.error(e)
|
@ -8,9 +8,5 @@ def RandomDick():
|
|||||||
with open('dicktxt/dick_sinonims.txt', 'r', encoding='utf-8') as file:
|
with open('dicktxt/dick_sinonims.txt', 'r', encoding='utf-8') as file:
|
||||||
words = file.readlines()
|
words = file.readlines()
|
||||||
|
|
||||||
random_line = random.choice(words).strip()
|
return random.choice(words).strip()
|
||||||
return random_line
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
21
dicktxt/WordsChange.py
Normal file
21
dicktxt/WordsChange.py
Normal file
@ -0,0 +1,21 @@
|
|||||||
|
import yaml
|
||||||
|
import random
|
||||||
|
|
||||||
|
|
||||||
|
def ChangeWord(result): #выбирает рандомное слово для изменения размера пиписы
|
||||||
|
|
||||||
|
with open("dicktxt/dick_changes.yaml", 'r', encoding='utf-8') as f:
|
||||||
|
words = yaml.load(f, Loader=yaml.Loader)
|
||||||
|
|
||||||
|
dick_inc = [ f'<b>{w}</b>' for w in words['increment'] ]
|
||||||
|
dick_static = [ f'<b>{w}</b>' for w in words['nothing'] ]
|
||||||
|
dick_decr = [ f'<b>{w}</b>' for w in words['decrement'] ]
|
||||||
|
|
||||||
|
if result > 0:
|
||||||
|
size_change = random.choice(dick_inc) + '🚀'
|
||||||
|
elif result == 0:
|
||||||
|
size_change = random.choice(dick_static) + '🤨'
|
||||||
|
else:
|
||||||
|
size_change = random.choice(dick_decr) + '✂'
|
||||||
|
|
||||||
|
return size_change
|
30
dicktxt/dick_changes.yaml
Normal file
30
dicktxt/dick_changes.yaml
Normal file
@ -0,0 +1,30 @@
|
|||||||
|
---
|
||||||
|
increment:
|
||||||
|
- вырос
|
||||||
|
- подрос
|
||||||
|
- привстал
|
||||||
|
- увеличился
|
||||||
|
- преисполнился
|
||||||
|
- приподнялся
|
||||||
|
- изменил размер
|
||||||
|
- облагородился
|
||||||
|
- чрезвычайно преуспел
|
||||||
|
decrement:
|
||||||
|
- неудачно потусил на сходке
|
||||||
|
- уменьшился
|
||||||
|
- откусила злая гадюка
|
||||||
|
- продали евреи, он стал дешевле
|
||||||
|
- схуднул
|
||||||
|
- отрицательно вырос
|
||||||
|
- словил низкие вибрации
|
||||||
|
- повредился после простуды
|
||||||
|
- стал жертвой репрессий
|
||||||
|
- был подло использован
|
||||||
|
nothing:
|
||||||
|
- продвинулся
|
||||||
|
- наноуменьшился
|
||||||
|
- познал стабильность
|
||||||
|
- зафиксировал прибыль
|
||||||
|
- подумал о судьбах Китая
|
||||||
|
- прогулялся по лесу
|
||||||
|
- самоутвердился
|
@ -1,4 +1,4 @@
|
|||||||
Кожаный бабах
|
кожаный бабах
|
||||||
пистон
|
пистон
|
||||||
дуршлаг
|
дуршлаг
|
||||||
елдак
|
елдак
|
||||||
@ -93,3 +93,4 @@
|
|||||||
венерический суккуб
|
венерический суккуб
|
||||||
боец невидимого фронта
|
боец невидимого фронта
|
||||||
сморчок
|
сморчок
|
||||||
|
джечник
|
6
global_conf.py
Normal file
6
global_conf.py
Normal file
@ -0,0 +1,6 @@
|
|||||||
|
import yaml
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
with open("config.yaml", 'r') as f:
|
||||||
|
CONFIG = yaml.load(f, Loader=yaml.Loader)
|
@ -1,7 +1,6 @@
|
|||||||
|
|
||||||
from handlers import pipisa
|
from handlers import pipisa
|
||||||
from handlers import davinci
|
|
||||||
from handlers import time_new_year
|
|
||||||
from handlers import sendalarm
|
|
||||||
from handlers import start_help
|
from handlers import start_help
|
||||||
|
|
||||||
|
from handlers import time_new_year
|
||||||
|
|
||||||
|
@ -1,25 +1,70 @@
|
|||||||
import openai
|
from openai import AsyncOpenAI
|
||||||
from aiogram import types, Dispatcher
|
from aiogram import types
|
||||||
from create_bot import dp, bot
|
from create_bot import dp, bot
|
||||||
|
import logging
|
||||||
|
from global_conf import CONFIG
|
||||||
|
|
||||||
token = '5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc'
|
admins = CONFIG['telegram_admins_ids']
|
||||||
openai.api_key = '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'
|
|
||||||
|
|
||||||
|
|
||||||
|
ban_comands = ['/dick','/topdick']
|
||||||
|
|
||||||
|
# Image_promt =['тупица сгенерируй','тупица сгенерируй картинку',\
|
||||||
|
# 'тупица отправь картинку', 'тупица отправь картинку']
|
||||||
|
client = AsyncOpenAI(api_key="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")
|
||||||
|
|
||||||
|
# @dp.message_handler(commands=['image'])
|
||||||
|
# async def send_image(message: types.Message):
|
||||||
|
# if message.from_user.id in admins and message.chat.type == 'private':
|
||||||
|
# try:
|
||||||
|
# description = message.text.replace('/image', '').strip()
|
||||||
|
|
||||||
|
# if not description:
|
||||||
|
# await message.reply('Ты даун, описание запроса напиши после пробела')
|
||||||
|
# return
|
||||||
|
# except Exception as e:
|
||||||
|
# logging.error(f'Error in send_image: {e}')
|
||||||
|
# try:
|
||||||
|
|
||||||
|
# response = openai.Image.create(
|
||||||
|
# prompt=message.text,
|
||||||
|
# n=1,
|
||||||
|
# size="1024x1024",
|
||||||
|
# response_format="url",
|
||||||
|
|
||||||
|
# )
|
||||||
|
# image_url = response['data'][0]['url']
|
||||||
|
# await message.answer('Генерирую изображение...')
|
||||||
|
# await message.reply_photo(image_url)
|
||||||
|
# except Exception as e:
|
||||||
|
# await message.reply('у меня не получилось')
|
||||||
|
|
||||||
|
|
||||||
|
async def gpt_text(req):
|
||||||
|
completion = await client.chat.completions.create(
|
||||||
|
messages=[
|
||||||
|
{
|
||||||
|
"role": "user",
|
||||||
|
"content": req,
|
||||||
|
}
|
||||||
|
],
|
||||||
|
model="gpt-4",
|
||||||
|
)
|
||||||
|
return completion
|
||||||
|
|
||||||
@dp.message_handler()
|
@dp.message_handler()
|
||||||
async def send(message: types.Message):
|
async def send(message: types.Message):
|
||||||
if 'тупица' in message.text.lower():
|
if message.from_user.id in admins and message.chat.type == 'private':
|
||||||
|
response = await gpt_text(message.text)
|
||||||
response = openai.Completion.create(
|
await message.answer(response.choices[0].message.content)
|
||||||
model="text-davinci-003",
|
|
||||||
prompt=message.text[7:],
|
|
||||||
temperature=0.7,
|
|
||||||
max_tokens=1000,
|
|
||||||
top_p=1.0,
|
|
||||||
frequency_penalty=0.0,
|
|
||||||
presence_penalty=0.6,
|
|
||||||
stop=["сброс"]
|
|
||||||
)
|
|
||||||
await message.reply(response['choices'][0]['text'])
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -1,179 +1,146 @@
|
|||||||
from aiogram import types, Dispatcher
|
from aiogram import types
|
||||||
from sqlalchemy.orm.session import close_all_sessions
|
|
||||||
from create_bot import bot, dp
|
from create_bot import bot, dp
|
||||||
from sqdb import user, session
|
|
||||||
import random
|
import random
|
||||||
from random import randint
|
from random import randint
|
||||||
import datetime
|
import datetime, logging
|
||||||
from dicktxt import ForReadDict
|
from dicktxt import ForReadDict, WordsChange
|
||||||
|
from pipisa_functions import pipisa_time
|
||||||
|
|
||||||
|
from global_conf import CONFIG
|
||||||
|
|
||||||
|
from db_logic import tops, user_stuff, postcleaner
|
||||||
|
|
||||||
|
|
||||||
|
admins = CONFIG['telegram_admins_ids']
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=["dick"])
|
@dp.message_handler(commands=["dick"])
|
||||||
async def up_dick(message: types.Message):
|
async def up_dick(message: types.Message):
|
||||||
|
|
||||||
|
if message.from_user.id in admins or message.chat.type != 'private':
|
||||||
|
|
||||||
|
|
||||||
|
postcleaner.append_post_to_cleaning_sequence(message=message, type=f'COMMAND_CALL__DICK')
|
||||||
|
await clean_posts(chat_id=message.chat.id, type='COMMAND_CALL__DICK', max_id=message.message_id)
|
||||||
|
|
||||||
|
#### TODO уплатить нолог с 100 см для самого нищего
|
||||||
|
|
||||||
# рандомайзер
|
# рандомайзер
|
||||||
numb = randint(-6,10)
|
result = randint(-9,10)
|
||||||
global size_change, dick_plus, dick_minus
|
|
||||||
dick_plus = ('<b>вырос</b>', '<b>подрос</b>', '<b>привстал</b>')
|
|
||||||
dick_minus = ('<b>уменьшился</b>', '<b>откусила злая гадюка</b>', '<b>продали евреи, он стал дешевле</b>')
|
|
||||||
if numb > 0:
|
|
||||||
size_change = random.choice(dick_plus) + '🚀'
|
|
||||||
#if numb = 0 size_change = random.choice(const) TODO
|
|
||||||
else: size_change = random.choice(dick_minus) + '✂'
|
|
||||||
|
|
||||||
|
#### Чекнуть есть ли юзер в базе
|
||||||
|
user = user_stuff.get_user(message)
|
||||||
|
|
||||||
# работа с таблицей
|
if user:
|
||||||
try:
|
## Если есть то нарастить и отправить сообщение
|
||||||
b = session.query(user).filter(user.user_id == message.from_user.id).first()
|
|
||||||
if b:
|
|
||||||
last_time = b.datetimes
|
|
||||||
now_time = datetime.datetime.now()
|
|
||||||
diff = now_time.day - last_time.day
|
|
||||||
|
|
||||||
if abs(diff) <= 0:
|
if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()):
|
||||||
try:
|
# пришло время крутить!!
|
||||||
current_dick = b.dick_size
|
|
||||||
|
|
||||||
|
updatedDick = user["dick_size"] + result
|
||||||
|
user_stuff.update_dick_size(message, user, updatedDick)
|
||||||
|
|
||||||
await bot.send_message(message.chat.id, f'@{message.from_user.username},🚫 вы уже крутили пипису, ее размер <b>{current_dick}</b> см! ')
|
randomdickword = ForReadDict.RandomDick()
|
||||||
except:
|
change_phrase = WordsChange.ChangeWord(result)
|
||||||
print('ошибка в if diff <=0')
|
|
||||||
session.rollback()
|
await bot.send_message( message.chat.id,
|
||||||
finally:
|
f'@{message.from_user.username}, ваш <em>{randomdickword}</em> {change_phrase} на <b>{abs(result)}</b> см!\nТеперь он равен <b>{updatedDick}</b> см! '
|
||||||
session.close_all()
|
)
|
||||||
print('пользователь уже крутил')
|
else:
|
||||||
|
mmm = await bot.send_message( message.chat.id,
|
||||||
|
f'@{message.from_user.username}, 🚫 вы уже крутили пипису, её размер <b>{user["dick_size"]}</b> см! '
|
||||||
|
)
|
||||||
|
user_stuff.update_attempts( message, user)
|
||||||
|
|
||||||
|
postcleaner.append_post_to_cleaning_sequence(message=mmm, type='ALREADY_ROLLED')
|
||||||
|
await clean_posts(chat_id=mmm.chat.id, type='ALREADY_ROLLED', max_id=mmm.message_id)
|
||||||
|
else:
|
||||||
|
## если нету, то создать
|
||||||
|
user_stuff.store_new_user(message, result)
|
||||||
|
|
||||||
|
await bot.send_message( message.chat.id,
|
||||||
|
f'@{message.from_user.username}, Добро пожаловать в игру! Ваш писюн показал головку 🚀\nна <b>{abs(result)}</b> см!\nТеперь он равен <b>{abs(result)}</b> см!'
|
||||||
|
)
|
||||||
|
|
||||||
else:
|
else:
|
||||||
try:
|
await message.reply('Растить елду можно только в общих чатах!')
|
||||||
a = session.query(user).filter(user.user_id == message.from_user.id).first() # запрос на поиск в таблице
|
|
||||||
updetedDick = a.dick_size = (a.dick_size + numb)
|
|
||||||
a.datetimes = datetime.datetime.now()
|
|
||||||
session.flush()
|
|
||||||
session.commit() #обновляет запись
|
|
||||||
randomword = ForReadDict.RandomDick()
|
|
||||||
|
|
||||||
await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш <em>{randomword}</em> {size_change}\
|
|
||||||
на <b>{abs(numb)}</b> см!\n\
|
|
||||||
Теперь он равен <b>{updetedDick}</b> см! ')
|
|
||||||
except:
|
|
||||||
session.rollback()
|
|
||||||
print('ошибка в обновлении пиписы')
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
print('пиписа обновлена')
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
else:
|
@dp.message_handler(commands = ['topdick', 'globaldick'])
|
||||||
session.close()
|
|
||||||
try:
|
|
||||||
int_table = user(user_id = message.from_user.id, user_fullname = message.from_user.full_name, dick_size = numb, datetimes = datetime.datetime.now(),\
|
|
||||||
chat_id = message.chat.id)
|
|
||||||
|
|
||||||
session.add(int_table)
|
|
||||||
session.flush()
|
|
||||||
session.commit()
|
|
||||||
session.close()
|
|
||||||
|
|
||||||
await bot.send_message(message.chat.id, f'@{message.from_user.username}, ваш писюн {size_change}\n\
|
|
||||||
на <b>{abs(numb)}</b> см!\n\
|
|
||||||
Теперь он равен <b>{numb}</b> см!')
|
|
||||||
except:
|
|
||||||
print('ошибка в добавлении нового пользователя')
|
|
||||||
session.rollback()
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
print('Успешно добавлен пользователь')
|
|
||||||
|
|
||||||
except:
|
|
||||||
session.rollback()
|
|
||||||
print('ошибка в основном коде')
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=["topdick"])
|
|
||||||
async def send_topchat(message: types.Message):
|
async def send_topchat(message: types.Message):
|
||||||
try:
|
if message.from_user.id in admins or message.chat.type != 'private':
|
||||||
changechat_id = session.query(user).filter(user.user_id == message.from_user.id).first()
|
|
||||||
chats = message.chat.id
|
top_ = message['text'].startswith('/topdick')
|
||||||
if changechat_id.chat_id != chats:
|
glob_ = message['text'].startswith('/globaldick')
|
||||||
changechat_id.chat_id = chats
|
|
||||||
session.commit()
|
if top_:
|
||||||
session.close()
|
dickos = tops.get_tops( top_ = True, chat_id=message.chat.id )
|
||||||
|
postcleaner.append_post_to_cleaning_sequence(message=message, type=f'COMMAND_CALL__TOPDICK')
|
||||||
|
await clean_posts(chat_id=message.chat.id, type='COMMAND_CALL__TOPDICK', max_id=message.message_id)
|
||||||
|
elif glob_:
|
||||||
|
dickos = tops.get_tops( glob_ = True )
|
||||||
|
postcleaner.append_post_to_cleaning_sequence(message=message, type=f'COMMAND_CALL__GLOBALDICK')
|
||||||
|
await clean_posts(chat_id=message.chat.id, type='COMMAND_CALL__GLOBALDICK', max_id=message.message_id)
|
||||||
else:
|
else:
|
||||||
session.close()
|
logging.error('вызывают хз что!')
|
||||||
top = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).filter(user.chat_id == chats).limit(10).all()
|
|
||||||
delimiter = ': '
|
|
||||||
|
|
||||||
# проверка на длину списка, если меньше limit, то:
|
|
||||||
|
|
||||||
sizelen = len(top)
|
|
||||||
if sizelen < 10:
|
|
||||||
lenminus = 10 - sizelen
|
|
||||||
top.extend(('-'*lenminus))
|
|
||||||
await bot.send_message(message.chat.id, f'🏆Топ 10 бубылд чата🏆\n\n\
|
|
||||||
🚀 {delimiter.join(map(str, top[0]))}\n\
|
|
||||||
💫 {delimiter.join(map(str, top[1]))}\n\
|
|
||||||
🍆 {delimiter.join(map(str, top[2]))}\n\
|
|
||||||
🍌 {delimiter.join(map(str, top[3]))}\n\
|
|
||||||
🐍 {delimiter.join(map(str, top[4]))}\n\
|
|
||||||
🐎 {delimiter.join(map(str, top[5]))}\n\
|
|
||||||
🌭 {delimiter.join(map(str, top[6]))}\n\
|
|
||||||
🌶 {delimiter.join(map(str, top[7]))}\n\
|
|
||||||
👌 {delimiter.join(map(str, top[8]))}\n\
|
|
||||||
💩 {delimiter.join(map(str, top[9]))}\n')
|
|
||||||
session.close()
|
|
||||||
# если все нормально и участников минимум 10
|
|
||||||
|
|
||||||
|
if not dickos:
|
||||||
|
await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌')
|
||||||
else:
|
else:
|
||||||
|
if top_:
|
||||||
|
await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos)
|
||||||
|
elif glob_:
|
||||||
|
await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + dickos)
|
||||||
|
else:
|
||||||
|
await message.reply('Работает только в общих чатах!\n'\
|
||||||
|
'Вы можете посмотреть топ по миру /globaldick')
|
||||||
|
|
||||||
await bot.send_message(message.chat.id, f'🏆Топ 10 бубылд чата🏆\n\n\
|
|
||||||
🚀 {delimiter.join(map(str, top[0]))}\n\
|
@dp.message_handler(commands = ['antitopdick', 'antiglobaldick'])
|
||||||
💫 {delimiter.join(map(str, top[1]))}\n\
|
async def send_antitopchat(message: types.Message):
|
||||||
🍆 {delimiter.join(map(str, top[2]))}\n\
|
if message.from_user.id in admins or message.chat.type != 'private':
|
||||||
🍌 {delimiter.join(map(str, top[3]))}\n\
|
antitop_ = message['text'].startswith('/antitopdick')
|
||||||
🐍 {delimiter.join(map(str, top[4]))}\n\
|
antiglob_ = message['text'].startswith('/antiglobaldick')
|
||||||
🐎 {delimiter.join(map(str, top[5]))}\n\
|
|
||||||
🌭 {delimiter.join(map(str, top[6]))}\n\
|
if antitop_:
|
||||||
🌶 {delimiter.join(map(str, top[7]))}\n\
|
dickos = tops.get_antitops(top_=True, chat_id=message.chat.id)
|
||||||
👌 {delimiter.join(map(str, top[8]))}\n\
|
postcleaner.append_post_to_cleaning_sequence(message=message, type=f'COMMAND_CALL__ANTITOPDICK')
|
||||||
💩 {delimiter.join(map(str, top[9]))}\n')
|
await clean_posts(chat_id=message.chat.id, type='COMMAND_CALL__ANTITOPDICK', max_id=message.message_id)
|
||||||
except:
|
elif antiglob_:
|
||||||
session.rollback()
|
dickos = tops.get_antitops(glob_=True)
|
||||||
print('ошибка в /topdick')
|
postcleaner.append_post_to_cleaning_sequence(message=message, type=f'COMMAND_CALL__ANTIGLOBALDICK')
|
||||||
finally:
|
await clean_posts(chat_id=message.chat.id, type='COMMAND_CALL__ANTIGLOBALDICK', max_id=message.message_id)
|
||||||
session.close_all()
|
else:
|
||||||
|
logging.error('вызывают хз что!')
|
||||||
|
|
||||||
|
if not dickos:
|
||||||
|
await bot.send_message(message.chat.id, '🍌 У всех пользователей бубылда растёт! 🍆')
|
||||||
|
else:
|
||||||
|
if antitop_:
|
||||||
|
await bot.send_message(message.chat.id, '🚫Антитоп 10 вагись чата🚫\n\n' + dickos)
|
||||||
|
elif antiglob_:
|
||||||
|
await bot.send_message(message.chat.id, '🚫Антитоп 10 вагись в мире🚫\n\n' + dickos)
|
||||||
|
else:
|
||||||
|
await message.reply('Работает только в общих чатах!\n'\
|
||||||
|
'Вы можете посмотреть антитоп по миру /antiglobaldick')
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=["globaldick"])
|
async def clean_posts(chat_id, type=None, max_id=None):
|
||||||
async def send_global_top(message: types.Message):
|
psts = postcleaner.get_posts_to_be_removed(chat_id, type, max_id)
|
||||||
|
|
||||||
|
for p in psts:
|
||||||
|
postcleaner.del_post(chat_id=p['chat_id'], msg_id=p['msg_id'])
|
||||||
|
|
||||||
|
#### TODO проверить админит ли бот
|
||||||
|
|
||||||
try:
|
try:
|
||||||
top_chat = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).limit(10).all()
|
await bot.delete_message( chat_id=p['chat_id'], message_id=p['msg_id'], )
|
||||||
|
except Exception as e:
|
||||||
delimiter = ': '
|
logging.error('ошибка удаления поста-на-удаление')
|
||||||
|
logging.error(e)
|
||||||
|
|
||||||
|
|
||||||
await bot.send_message(message.chat.id, f'🏆Топ 10 пипис в мире🏆\n\n\
|
|
||||||
1. {delimiter.join(map(str, top_chat[0]))}\n\
|
|
||||||
2. {delimiter.join(map(str, top_chat[1]))}\n\
|
|
||||||
3. {delimiter.join(map(str, top_chat[2]))}\n\
|
|
||||||
4. {delimiter.join(map(str, top_chat[3]))}\n\
|
|
||||||
5. {delimiter.join(map(str, top_chat[4]))}\n\
|
|
||||||
6. {delimiter.join(map(str, top_chat[5]))}\n\
|
|
||||||
7. {delimiter.join(map(str, top_chat[6]))}\n\
|
|
||||||
8. {delimiter.join(map(str, top_chat[7]))}\n\
|
|
||||||
9. {delimiter.join(map(str, top_chat[8]))}\n\
|
|
||||||
10. {delimiter.join(map(str, top_chat[9]))}\n')
|
|
||||||
except:
|
|
||||||
session.rollback()
|
|
||||||
print('ошибка в /glovaldick')
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
|
|
||||||
|
|
||||||
# def register_handlers_pipisa(dp: Dispatcher):
|
|
||||||
# dp.register_message_handler(up_dick)
|
|
||||||
# dp.register_message_handler(send_topchat)
|
|
||||||
# dp.register_message_handler(send_global_top)
|
|
||||||
|
|
||||||
|
@ -1,29 +0,0 @@
|
|||||||
|
|
||||||
from create_bot import dp, bot
|
|
||||||
from aiogram import types, Dispatcher
|
|
||||||
from sqdb import user, session
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=['sendall'])
|
|
||||||
async def sendall(message: types.Message):
|
|
||||||
print('команда прошла')
|
|
||||||
if message.chat.type == 'private':
|
|
||||||
if message.from_user.id == 226765676:
|
|
||||||
text = message.text[9:]
|
|
||||||
try:
|
|
||||||
chats_id = session.query(user.chat_id).distinct()
|
|
||||||
for row in chats_id:
|
|
||||||
try:
|
|
||||||
await bot.send_message(row[0], text)
|
|
||||||
except:
|
|
||||||
print('вероятно бота нет в чате')
|
|
||||||
await bot.send_message(message.from_user.id, 'Сообщение успешно отправлено во все чаты')
|
|
||||||
except:
|
|
||||||
session.rollback()
|
|
||||||
print('eror sendall')
|
|
||||||
finally: session.close_all()
|
|
||||||
|
|
||||||
|
|
||||||
# def register_handlers_test(dp: Dispatcher):
|
|
||||||
# dp.register_message_handler(send_welcome)
|
|
@ -1,22 +1,27 @@
|
|||||||
from aiogram import types, Dispatcher
|
from aiogram import types
|
||||||
|
from aiogram.types import InlineKeyboardMarkup, InlineKeyboardButton
|
||||||
from create_bot import dp, bot
|
from create_bot import dp, bot
|
||||||
|
|
||||||
@dp.message_handler(commands=['start'])
|
@dp.message_handler(commands=['start'])
|
||||||
async def start_func(message: types.Message):
|
async def start_func(message: types.Message):
|
||||||
await message.reply('похуй')
|
if message.chat.type == 'private':
|
||||||
|
|
||||||
|
lkb = InlineKeyboardMarkup(row_width=1).add(InlineKeyboardButton(text = 'Добавить в группу', callback_data='add_2_group'))
|
||||||
|
|
||||||
|
await message.answer(f'<b>Добро пожаловать, {message.from_user.full_name}!</b>\n'
|
||||||
|
'Это бот, который растит члены. Чтобы начать, добавь бота в чат'\
|
||||||
|
, reply_markup=lkb
|
||||||
|
|
||||||
|
)
|
||||||
|
# TODO добавить кнопку, по которой смогут добавлять юзера в чат
|
||||||
|
|
||||||
|
@dp.callback_query_handler(text = 'add_2_group')
|
||||||
|
async def addgroup(callback: types.CallbackQuery):
|
||||||
|
await callback.answer('жмав')
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=['photo'])
|
|
||||||
async def send_image(message: types.Message):
|
|
||||||
await bot.send_photo(message.chat.id, photo='https://memepedia.ru/wp-content/uploads/2018/08/ya-pidoras.jpg', reply_to_message_id=message.message_id)
|
|
||||||
|
|
||||||
@dp.message_handler(content_types=types.ContentTypes.TEXT)
|
|
||||||
async def send_faggot(message: types.Message):
|
|
||||||
print(message.text)
|
|
||||||
if message.text == 'хто я':
|
|
||||||
await bot.send_photo(message.chat.id, photo='https://www.meme-arsenal.com/memes/ecebd2339c7eab40e09874bd86a38a09.jpg')
|
|
||||||
|
|
||||||
|
|
||||||
def register_handlers_StartHelp(dp: Dispatcher):
|
|
||||||
dp.register_message_handler(start_func)
|
|
||||||
dp.register_message_handler(send_faggot)
|
|
||||||
dp.register_message_handler(send_image)
|
|
@ -4,10 +4,10 @@ import datetime
|
|||||||
|
|
||||||
@dp.message_handler(commands=["time"])
|
@dp.message_handler(commands=["time"])
|
||||||
async def send_time(message: types.Message):
|
async def send_time(message: types.Message):
|
||||||
new_year = datetime.datetime(2022, 12, 31) #дата нового года
|
new_year = datetime.datetime(datetime.datetime.now().year, 12, 31) #дата нового года
|
||||||
ct = datetime.datetime.now() #датавремя
|
ct = datetime.datetime.now() #датавремя
|
||||||
cd = datetime.datetime.now().strftime("%d/%m/%Y") #дата (д,м,г)
|
cd = datetime.datetime.now().strftime("%d/%m/%Y") #дата (д,м,г)
|
||||||
ct1 = ct+datetime.timedelta(hours=15) # +14 часов от сервера
|
ct1 = ct # zoichem??? +datetime.timedelta(hours=15) # +14 часов от сервера
|
||||||
ct2 = ct1.strftime('%H:%M') # форматирует датувремя просто во время(ч,м)
|
ct2 = ct1.strftime('%H:%M') # форматирует датувремя просто во время(ч,м)
|
||||||
raznitsa = (new_year - ct).days #отнимает от нг текущее время и получаем разницу в днях
|
raznitsa = (new_year - ct).days #отнимает от нг текущее время и получаем разницу в днях
|
||||||
|
|
||||||
|
15
jack_bot.service
Normal file
15
jack_bot.service
Normal file
@ -0,0 +1,15 @@
|
|||||||
|
# пример того что лежит на хосте
|
||||||
|
[Unit]
|
||||||
|
Description=pipisa-bot
|
||||||
|
After=network.target
|
||||||
|
|
||||||
|
[Service]
|
||||||
|
User=jack
|
||||||
|
Type=simple
|
||||||
|
Restart=always
|
||||||
|
WorkingDirectory=/usr/share/python_bot
|
||||||
|
ExecStart=/usr/bin/python3 /usr/share/python_bot/bot.py
|
||||||
|
ExecStart=/usr/bin/python3 /usr/share/python_bot/bot.py
|
||||||
|
|
||||||
|
[Install]
|
||||||
|
WantedBy=multi-user.target#
|
1
pipisa_functions/__init__.py
Normal file
1
pipisa_functions/__init__.py
Normal file
@ -0,0 +1 @@
|
|||||||
|
from . import pipisa_time
|
88
pipisa_functions/pipisa_time.py
Normal file
88
pipisa_functions/pipisa_time.py
Normal file
@ -0,0 +1,88 @@
|
|||||||
|
|
||||||
|
import unittest
|
||||||
|
import datetime
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def rolltime(check_datetime , curr_time=datetime.datetime.now()):
|
||||||
|
|
||||||
|
last_time = datetime.datetime.strptime(check_datetime, '%Y-%m-%d %H:%M:%S')
|
||||||
|
time_to_grow = curr_time.replace(hour=14, minute=8, second=8, microsecond=0) #+ datetime.timedelta(days=0)
|
||||||
|
|
||||||
|
timediff = (time_to_grow - last_time).total_seconds()
|
||||||
|
|
||||||
|
# print()
|
||||||
|
# print(f'\tlast_time : {last_time}')
|
||||||
|
# print(f'\ttime_to_grow : {time_to_grow}')
|
||||||
|
# print(f'\tcurr_time : {curr_time.strftime("%Y-%m-%d %H:%M:%S")}')
|
||||||
|
# print(f'\t\tsecs diff : {timediff}')
|
||||||
|
|
||||||
|
|
||||||
|
if timediff > 86400 or (curr_time > time_to_grow and last_time < time_to_grow):
|
||||||
|
return True
|
||||||
|
else:
|
||||||
|
return False
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class TestPipisa(unittest.TestCase):
|
||||||
|
def test_very_long_ago(self):
|
||||||
|
self.assertEqual(rolltime('2021-10-11 23:44:00'), True, "Пиписа прокручена очень давно. Очевидный прокрут")
|
||||||
|
|
||||||
|
def test_just_after(self):
|
||||||
|
self.assertEqual(rolltime(
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0).strftime('%Y-%m-%d %H:%M:%S')
|
||||||
|
), False, "Попытка крутить когда старый прокрут был через 1 минуту после сегодняшнего обновления роллов. Фейл.")
|
||||||
|
|
||||||
|
def test_yesterdayroll(self):
|
||||||
|
self.assertEqual(
|
||||||
|
rolltime(
|
||||||
|
(
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0) - datetime.timedelta(days=1)
|
||||||
|
).strftime('%Y-%m-%d %H:%M:%S'),
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0)
|
||||||
|
), True, "Крутилась вчера через минуту после обновления роллов. Должно пропускать ")
|
||||||
|
|
||||||
|
|
||||||
|
def test_hour_before_roll(self):
|
||||||
|
self.assertEqual(
|
||||||
|
rolltime(
|
||||||
|
(
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0) - datetime.timedelta(hours=1)
|
||||||
|
).strftime('%Y-%m-%d %H:%M:%S'),
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0)
|
||||||
|
), True, "Крутилась последний раз за час до прокрута. Надо разрешать заново!")
|
||||||
|
|
||||||
|
def test_after_midnight(self):
|
||||||
|
self.assertEqual(
|
||||||
|
rolltime(
|
||||||
|
(
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0) - datetime.timedelta(hours=14)
|
||||||
|
).strftime('%Y-%m-%d %H:%M:%S'),
|
||||||
|
datetime.datetime.now().replace(hour=0, minute=11, second=8, microsecond=0)
|
||||||
|
), False, "Где-то ночью прокручивается повторно. Фейл.")
|
||||||
|
|
||||||
|
def test_before_midnight(self):
|
||||||
|
self.assertEqual(
|
||||||
|
rolltime(
|
||||||
|
(
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=9, second=8, microsecond=0) - datetime.timedelta(hours=15)
|
||||||
|
).strftime('%Y-%m-%d %H:%M:%S'),
|
||||||
|
datetime.datetime.now().replace(hour=0, minute=11, second=8, microsecond=0) - datetime.timedelta(hours=15)
|
||||||
|
), False, "Где-то ночью прокручивается повторно. Фейл.")
|
||||||
|
|
||||||
|
def test_somewhere_after(self):
|
||||||
|
self.assertEqual(
|
||||||
|
rolltime(
|
||||||
|
(
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=26, second=8, microsecond=0)
|
||||||
|
).strftime('%Y-%m-%d %H:%M:%S'),
|
||||||
|
datetime.datetime.now().replace(hour=14, minute=27, second=8, microsecond=0)
|
||||||
|
), False, "Сегодня уже открутили а хотят ещё. Фейл.")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main(verbosity=2)
|
@ -1,4 +1,3 @@
|
|||||||
aiogram
|
aiogram==2.15
|
||||||
sqdb
|
python-arango
|
||||||
sqlalchemy
|
pyyaml
|
||||||
graypy
|
|
56
sqdb.py
56
sqdb.py
@ -1,56 +0,0 @@
|
|||||||
from sqlalchemy import create_engine, MetaData, Table, Integer, String, BIGINT, \
|
|
||||||
Column, DateTime, ForeignKey, Numeric
|
|
||||||
from sqlalchemy.ext.declarative import declarative_base
|
|
||||||
from sqlalchemy.orm import sessionmaker
|
|
||||||
from sqlalchemy import select, update
|
|
||||||
|
|
||||||
engine = create_engine('postgresql+psycopg2://postgres:postgres@postgres.vdk2ch.ru:5432/jack_bot', echo=True)
|
|
||||||
Session = sessionmaker(bind=engine)
|
|
||||||
session = Session()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
Base = declarative_base()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class user(Base):
|
|
||||||
__tablename__ = 'users'
|
|
||||||
|
|
||||||
id = Column(Integer, primary_key=True)
|
|
||||||
user_id = Column(Integer)
|
|
||||||
user_fullname = Column(String)
|
|
||||||
dick_size = (Column(Integer))
|
|
||||||
datetimes = (Column(DateTime))
|
|
||||||
chat_id = Column(BIGINT)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
session.close()
|
|
||||||
engine.dispose()
|
|
||||||
# top = session.query(user.user_fullname, user.dick_size).order_by(user.dick_size.desc()).filter(user.chat_id == user.chat_id).limit(10).all()
|
|
||||||
# print(top)
|
|
||||||
|
|
||||||
|
|
||||||
#Base.metadata.create_all(engine)
|
|
||||||
|
|
||||||
# int_table = user(user_id = 12345678, user_fullname = 'test', dick_size = 10) #импортирую юзер из sqdb
|
|
||||||
# session.add(int_table)
|
|
||||||
# session.commit()
|
|
||||||
# select user_id, name from employees limit 10
|
|
||||||
|
|
||||||
# def adding():
|
|
||||||
|
|
||||||
# session.add()
|
|
||||||
# session.commit()
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user