forked from Pavel_Durov/python_bot
Compare commits
68 Commits
Author | SHA1 | Date | |
---|---|---|---|
|
c3b6e56010 | ||
|
150472a2de | ||
|
10df69242d | ||
|
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 |
@ -15,11 +15,12 @@ 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:
|
when:
|
||||||
branch:
|
branch:
|
||||||
- main
|
- main
|
||||||
|
18
DBChat/user_operations.py
Normal file
18
DBChat/user_operations.py
Normal file
@ -0,0 +1,18 @@
|
|||||||
|
import datetime
|
||||||
|
|
||||||
|
def UserCreate(DBName, message, result, replyId): #create new user
|
||||||
|
|
||||||
|
new_user = { #описание класса пользователя
|
||||||
|
'user_id': message.from_user.id, #уникальный идентификатор пользователя
|
||||||
|
'user_fullname': message.from_user.full_name, #никнэйм
|
||||||
|
'dick_size': result, #размер пиписы
|
||||||
|
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"), #время прокрута
|
||||||
|
'attempts': 1, #общее количество попыток прокрутки
|
||||||
|
'chat_id': message.chat.id, #уникальный идентификатор конфы, в которой этот пользователь крутит
|
||||||
|
'reply_id': replyId.id #идентификатор последнего ответа для этого пользователя
|
||||||
|
}
|
||||||
|
try:
|
||||||
|
metadata = DBName.insert(new_user, overwrite_mode='update')
|
||||||
|
except Exception as add_err:
|
||||||
|
print('ошибка в добавлении нового пользователя')
|
||||||
|
print(add_err)
|
13
README.md
Normal file
13
README.md
Normal file
@ -0,0 +1,13 @@
|
|||||||
|
Пробуем проверять что-либо.
|
||||||
|
|
||||||
|
```mermaid
|
||||||
|
journey
|
||||||
|
title Пиписа
|
||||||
|
section Наш Чат
|
||||||
|
Оче больше пипис: 5: ГЕГЕМОН
|
||||||
|
Обычно большепенис: 3: Догоняющие
|
||||||
|
Скромный дик: 1: Отставшие
|
||||||
|
section Не наш чат
|
||||||
|
Не запустил: -1: Отставшие
|
||||||
|
Не крутил: 1: Отставшие
|
||||||
|
```
|
64
TopMsg/topdicks.py
Normal file
64
TopMsg/topdicks.py
Normal file
@ -0,0 +1,64 @@
|
|||||||
|
async def topdicks_conf(chatId): # тело метода лучших результатов отдельной конфы
|
||||||
|
arango_client = ArangoClient(hosts='https://arango.guaranteedstruggle.host')
|
||||||
|
pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527')
|
||||||
|
dicks_collection = pipisa_db.collection('dicks')
|
||||||
|
|
||||||
|
try:
|
||||||
|
dicks = dicks_collection.find({'chat_id': chatId}, skip=0, limit=1100)
|
||||||
|
dicks = [d for d in dicks]
|
||||||
|
except Exception as e:
|
||||||
|
print('ошибка DB в /topdick')
|
||||||
|
print(e)
|
||||||
|
|
||||||
|
top_dicks = sorted(dicks, key=lambda dick: dick['dick_size'], reverse=True)
|
||||||
|
|
||||||
|
dick_limit = min(len(top_dicks), 10)
|
||||||
|
top_dicks = top_dicks[:dick_limit]
|
||||||
|
|
||||||
|
conf_top = ''
|
||||||
|
emo = ['🏆','🚀','🍆','🍌','🐍','🐎','🌭','🌶','👌','💩']
|
||||||
|
|
||||||
|
if len(top_dicks) == 0:
|
||||||
|
await bot.send_message(message.chat.id, '🍆 Никто ничего не нарастил! 🍌')
|
||||||
|
else:
|
||||||
|
for i in range(len(top_dicks)):
|
||||||
|
conf_top += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n'
|
||||||
|
i += 1
|
||||||
|
|
||||||
|
|
||||||
|
return conf_top
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
async def topdicks_global(chatId): #тело метода глобальных лучших результатов
|
||||||
|
arango_client = ArangoClient(hosts='https://arango.guaranteedstruggle.host')
|
||||||
|
pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527')
|
||||||
|
dicks_collection = pipisa_db.collection('dicks')
|
||||||
|
|
||||||
|
try:
|
||||||
|
dicks = dicks_collection.all( )
|
||||||
|
dicks = [d for d in dicks if d['user_id'] != d['chat_id']]
|
||||||
|
except Exception as e:
|
||||||
|
print('ошибка DB в /glovaldick')
|
||||||
|
print(e)
|
||||||
|
|
||||||
|
top_dicks = sorted(dicks, key=lambda dick: dick['dick_size'], reverse=True)
|
||||||
|
|
||||||
|
|
||||||
|
dick_limit = min(len(top_dicks), 10)
|
||||||
|
top_dicks = top_dicks[:dick_limit]
|
||||||
|
|
||||||
|
global_top = ''
|
||||||
|
emo = ['🏆','🚀','💫','🍆','🍌','🐍','🐎','🌭','🌶','👌']
|
||||||
|
|
||||||
|
if len(top_dicks) == 0:
|
||||||
|
await bot.send_message(chatId, '🍆 Никто ничего не нарастил! 🍌')
|
||||||
|
else:
|
||||||
|
for i in range(len(top_dicks)):
|
||||||
|
global_top += f' {emo[i]} {i+1}. {top_dicks[i]["user_fullname"]}: {top_dicks[i]["dick_size"]}см\n'
|
||||||
|
i += 1
|
||||||
|
|
||||||
|
return global_top
|
||||||
|
|
||||||
|
|
||||||
|
|
0
__init__.py
Normal file
0
__init__.py
Normal file
110
alembic.ini
110
alembic.ini
@ -1,110 +0,0 @@
|
|||||||
# A generic, single database configuration.
|
|
||||||
|
|
||||||
[alembic]
|
|
||||||
# path to migration scripts
|
|
||||||
script_location = alembic
|
|
||||||
|
|
||||||
# template used to generate migration file names; The default value is %%(rev)s_%%(slug)s
|
|
||||||
# Uncomment the line below if you want the files to be prepended with date and time
|
|
||||||
# see https://alembic.sqlalchemy.org/en/latest/tutorial.html#editing-the-ini-file
|
|
||||||
# for all available tokens
|
|
||||||
# file_template = %%(year)d_%%(month).2d_%%(day).2d_%%(hour).2d%%(minute).2d-%%(rev)s_%%(slug)s
|
|
||||||
|
|
||||||
# sys.path path, will be prepended to sys.path if present.
|
|
||||||
# defaults to the current working directory.
|
|
||||||
prepend_sys_path = .
|
|
||||||
|
|
||||||
# timezone to use when rendering the date within the migration file
|
|
||||||
# as well as the filename.
|
|
||||||
# If specified, requires the python-dateutil library that can be
|
|
||||||
# installed by adding `alembic[tz]` to the pip requirements
|
|
||||||
# string value is passed to dateutil.tz.gettz()
|
|
||||||
# leave blank for localtime
|
|
||||||
# timezone =
|
|
||||||
|
|
||||||
# max length of characters to apply to the
|
|
||||||
# "slug" field
|
|
||||||
# truncate_slug_length = 40
|
|
||||||
|
|
||||||
# set to 'true' to run the environment during
|
|
||||||
# the 'revision' command, regardless of autogenerate
|
|
||||||
# revision_environment = false
|
|
||||||
|
|
||||||
# set to 'true' to allow .pyc and .pyo files without
|
|
||||||
# a source .py file to be detected as revisions in the
|
|
||||||
# versions/ directory
|
|
||||||
# sourceless = false
|
|
||||||
|
|
||||||
# version location specification; This defaults
|
|
||||||
# to alembic/versions. When using multiple version
|
|
||||||
# directories, initial revisions must be specified with --version-path.
|
|
||||||
# The path separator used here should be the separator specified by "version_path_separator" below.
|
|
||||||
# version_locations = %(here)s/bar:%(here)s/bat:alembic/versions
|
|
||||||
|
|
||||||
# version path separator; As mentioned above, this is the character used to split
|
|
||||||
# version_locations. The default within new alembic.ini files is "os", which uses os.pathsep.
|
|
||||||
# If this key is omitted entirely, it falls back to the legacy behavior of splitting on spaces and/or commas.
|
|
||||||
# Valid values for version_path_separator are:
|
|
||||||
#
|
|
||||||
# version_path_separator = :
|
|
||||||
# version_path_separator = ;
|
|
||||||
# version_path_separator = space
|
|
||||||
version_path_separator = os # Use os.pathsep. Default configuration used for new projects.
|
|
||||||
|
|
||||||
# set to 'true' to search source files recursively
|
|
||||||
# in each "version_locations" directory
|
|
||||||
# new in Alembic version 1.10
|
|
||||||
# recursive_version_locations = false
|
|
||||||
|
|
||||||
# the output encoding used when revision files
|
|
||||||
# are written from script.py.mako
|
|
||||||
# output_encoding = utf-8
|
|
||||||
|
|
||||||
sqlalchemy.url = postgresql+psycopg2://postgres:postgres@postgres.vdk2ch.ru:5432/jack_bot
|
|
||||||
|
|
||||||
|
|
||||||
[post_write_hooks]
|
|
||||||
# post_write_hooks defines scripts or Python functions that are run
|
|
||||||
# on newly generated revision scripts. See the documentation for further
|
|
||||||
# detail and examples
|
|
||||||
|
|
||||||
# format using "black" - use the console_scripts runner, against the "black" entrypoint
|
|
||||||
# hooks = black
|
|
||||||
# black.type = console_scripts
|
|
||||||
# black.entrypoint = black
|
|
||||||
# black.options = -l 79 REVISION_SCRIPT_FILENAME
|
|
||||||
|
|
||||||
# Logging configuration
|
|
||||||
[loggers]
|
|
||||||
keys = root,sqlalchemy,alembic
|
|
||||||
|
|
||||||
[handlers]
|
|
||||||
keys = console
|
|
||||||
|
|
||||||
[formatters]
|
|
||||||
keys = generic
|
|
||||||
|
|
||||||
[logger_root]
|
|
||||||
level = WARN
|
|
||||||
handlers = console
|
|
||||||
qualname =
|
|
||||||
|
|
||||||
[logger_sqlalchemy]
|
|
||||||
level = WARN
|
|
||||||
handlers =
|
|
||||||
qualname = sqlalchemy.engine
|
|
||||||
|
|
||||||
[logger_alembic]
|
|
||||||
level = INFO
|
|
||||||
handlers =
|
|
||||||
qualname = alembic
|
|
||||||
|
|
||||||
[handler_console]
|
|
||||||
class = StreamHandler
|
|
||||||
args = (sys.stderr,)
|
|
||||||
level = NOTSET
|
|
||||||
formatter = generic
|
|
||||||
|
|
||||||
[formatter_generic]
|
|
||||||
format = %(levelname)-5.5s [%(name)s] %(message)s
|
|
||||||
datefmt = %H:%M:%S
|
|
@ -1 +0,0 @@
|
|||||||
Generic single-database configuration.
|
|
@ -1,78 +0,0 @@
|
|||||||
from logging.config import fileConfig
|
|
||||||
|
|
||||||
from sqlalchemy import engine_from_config
|
|
||||||
from sqlalchemy import pool
|
|
||||||
from sqdb import Base
|
|
||||||
from alembic import context
|
|
||||||
|
|
||||||
# this is the Alembic Config object, which provides
|
|
||||||
# access to the values within the .ini file in use.
|
|
||||||
config = context.config
|
|
||||||
|
|
||||||
# Interpret the config file for Python logging.
|
|
||||||
# This line sets up loggers basically.
|
|
||||||
if config.config_file_name is not None:
|
|
||||||
fileConfig(config.config_file_name)
|
|
||||||
|
|
||||||
# add your model's MetaData object here
|
|
||||||
# for 'autogenerate' support
|
|
||||||
# from myapp import mymodel
|
|
||||||
# target_metadata = mymodel.Base.metadata
|
|
||||||
target_metadata = Base.metadata
|
|
||||||
|
|
||||||
# other values from the config, defined by the needs of env.py,
|
|
||||||
# can be acquired:
|
|
||||||
# my_important_option = config.get_main_option("my_important_option")
|
|
||||||
# ... etc.
|
|
||||||
|
|
||||||
|
|
||||||
def run_migrations_offline() -> None:
|
|
||||||
"""Run migrations in 'offline' mode.
|
|
||||||
|
|
||||||
This configures the context with just a URL
|
|
||||||
and not an Engine, though an Engine is acceptable
|
|
||||||
here as well. By skipping the Engine creation
|
|
||||||
we don't even need a DBAPI to be available.
|
|
||||||
|
|
||||||
Calls to context.execute() here emit the given string to the
|
|
||||||
script output.
|
|
||||||
|
|
||||||
"""
|
|
||||||
url = config.get_main_option("sqlalchemy.url")
|
|
||||||
context.configure(
|
|
||||||
url=url,
|
|
||||||
target_metadata=target_metadata,
|
|
||||||
literal_binds=True,
|
|
||||||
dialect_opts={"paramstyle": "named"},
|
|
||||||
)
|
|
||||||
|
|
||||||
with context.begin_transaction():
|
|
||||||
context.run_migrations()
|
|
||||||
|
|
||||||
|
|
||||||
def run_migrations_online() -> None:
|
|
||||||
"""Run migrations in 'online' mode.
|
|
||||||
|
|
||||||
In this scenario we need to create an Engine
|
|
||||||
and associate a connection with the context.
|
|
||||||
|
|
||||||
"""
|
|
||||||
connectable = engine_from_config(
|
|
||||||
config.get_section(config.config_ini_section, {}),
|
|
||||||
prefix="sqlalchemy.",
|
|
||||||
poolclass=pool.NullPool,
|
|
||||||
)
|
|
||||||
|
|
||||||
with connectable.connect() as connection:
|
|
||||||
context.configure(
|
|
||||||
connection=connection, target_metadata=target_metadata
|
|
||||||
)
|
|
||||||
|
|
||||||
with context.begin_transaction():
|
|
||||||
context.run_migrations()
|
|
||||||
|
|
||||||
|
|
||||||
if context.is_offline_mode():
|
|
||||||
run_migrations_offline()
|
|
||||||
else:
|
|
||||||
run_migrations_online()
|
|
@ -1,24 +0,0 @@
|
|||||||
"""${message}
|
|
||||||
|
|
||||||
Revision ID: ${up_revision}
|
|
||||||
Revises: ${down_revision | comma,n}
|
|
||||||
Create Date: ${create_date}
|
|
||||||
|
|
||||||
"""
|
|
||||||
from alembic import op
|
|
||||||
import sqlalchemy as sa
|
|
||||||
${imports if imports else ""}
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
|
||||||
revision = ${repr(up_revision)}
|
|
||||||
down_revision = ${repr(down_revision)}
|
|
||||||
branch_labels = ${repr(branch_labels)}
|
|
||||||
depends_on = ${repr(depends_on)}
|
|
||||||
|
|
||||||
|
|
||||||
def upgrade() -> None:
|
|
||||||
${upgrades if upgrades else "pass"}
|
|
||||||
|
|
||||||
|
|
||||||
def downgrade() -> None:
|
|
||||||
${downgrades if downgrades else "pass"}
|
|
@ -1,47 +0,0 @@
|
|||||||
"""empty message
|
|
||||||
|
|
||||||
Revision ID: 7521d874503f
|
|
||||||
Revises:
|
|
||||||
Create Date: 2023-06-22 15:21:26.835612
|
|
||||||
|
|
||||||
"""
|
|
||||||
from alembic import op
|
|
||||||
import sqlalchemy as sa
|
|
||||||
from sqlalchemy.dialects import postgresql
|
|
||||||
|
|
||||||
# revision identifiers, used by Alembic.
|
|
||||||
revision = '7521d874503f'
|
|
||||||
down_revision = None
|
|
||||||
branch_labels = None
|
|
||||||
depends_on = None
|
|
||||||
|
|
||||||
|
|
||||||
def upgrade() -> None:
|
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
|
||||||
op.create_table('dick_requests',
|
|
||||||
sa.Column('id', sa.Integer(), nullable=False),
|
|
||||||
sa.Column('size_change', sa.Integer(), nullable=True),
|
|
||||||
sa.Column('user_id', sa.Integer(), nullable=True),
|
|
||||||
sa.Column('date_changed', sa.DateTime(), nullable=True),
|
|
||||||
sa.ForeignKeyConstraint(['user_id'], ['users.user_id'], ),
|
|
||||||
sa.PrimaryKeyConstraint('id')
|
|
||||||
)
|
|
||||||
op.create_foreign_key(None, 'user_dicks', 'users', ['user_id'], ['user_id'])
|
|
||||||
op.add_column('users', sa.Column('date_created', sa.DateTime(), nullable=True))
|
|
||||||
op.add_column('users', sa.Column('date_updated', sa.DateTime(), nullable=True))
|
|
||||||
op.drop_column('users', 'chat_id')
|
|
||||||
op.drop_column('users', 'datetimes')
|
|
||||||
op.drop_column('users', 'dick_size')
|
|
||||||
# ### end Alembic commands ###
|
|
||||||
|
|
||||||
|
|
||||||
def downgrade() -> None:
|
|
||||||
# ### commands auto generated by Alembic - please adjust! ###
|
|
||||||
op.add_column('users', sa.Column('dick_size', sa.INTEGER(), autoincrement=False, nullable=True))
|
|
||||||
op.add_column('users', sa.Column('datetimes', postgresql.TIMESTAMP(), autoincrement=False, nullable=True))
|
|
||||||
op.add_column('users', sa.Column('chat_id', sa.BIGINT(), autoincrement=False, nullable=True))
|
|
||||||
op.drop_column('users', 'date_updated')
|
|
||||||
op.drop_column('users', 'date_created')
|
|
||||||
op.drop_constraint(None, 'user_dicks', type_='foreignkey')
|
|
||||||
op.drop_table('dick_requests')
|
|
||||||
# ### end Alembic commands ###
|
|
12
bot.py
12
bot.py
@ -1,20 +1,12 @@
|
|||||||
import logging
|
import logging
|
||||||
import graypy
|
|
||||||
|
|
||||||
from aiogram import executor
|
from aiogram import executor
|
||||||
|
|
||||||
from create_bot import dp
|
from create_bot import dp
|
||||||
|
|
||||||
import handlers
|
import handlers
|
||||||
|
|
||||||
|
|
||||||
logging.basicConfig(level=logging.DEBUG)
|
logging.basicConfig(level=logging.DEBUG)
|
||||||
|
|
||||||
|
|
||||||
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)
|
||||||
@ -25,3 +17,7 @@ logging.basicConfig(
|
|||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
executor.start_polling(dp, skip_updates=True)
|
executor.start_polling(dp, skip_updates=True)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -1,4 +1,7 @@
|
|||||||
from aiogram import Bot, Dispatcher
|
from aiogram import Bot, Dispatcher
|
||||||
|
|
||||||
|
|
||||||
API_TOKEN = '5724991559:AAEuLvpLsgP6LHRGMSyFtQLlR5qPQUO4b_w'
|
API_TOKEN = '5724991559:AAEuLvpLsgP6LHRGMSyFtQLlR5qPQUO4b_w'
|
||||||
|
|
||||||
bot = Bot(token=API_TOKEN, parse_mode="HTML")
|
bot = Bot(token=API_TOKEN, parse_mode="HTML")
|
||||||
dp = Dispatcher(bot)
|
dp = Dispatcher(bot)
|
34
dicktxt/change_words.py
Normal file
34
dicktxt/change_words.py
Normal file
@ -0,0 +1,34 @@
|
|||||||
|
import random
|
||||||
|
|
||||||
|
def ChangeWord(result): #выбирает рандомное слово для изменения размера пиписы
|
||||||
|
|
||||||
|
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) + '✂'
|
||||||
|
|
||||||
|
|
||||||
|
dick_inc = (
|
||||||
|
'<b>вырос</b>',
|
||||||
|
'<b>подрос</b>',
|
||||||
|
'<b>привстал</b>',
|
||||||
|
'<b>увеличился</b>',
|
||||||
|
'<b>преисполнился</b>'
|
||||||
|
)
|
||||||
|
dick_static = (
|
||||||
|
'<b>продвинулся</b>',
|
||||||
|
'<b>наноуменьшился</b>',
|
||||||
|
'<b>познал стабильность</b>',
|
||||||
|
'<b>зафиксировал прибыль</b>'
|
||||||
|
)
|
||||||
|
dick_decr = (
|
||||||
|
'<b>неудачно потусил на сходке</b>',
|
||||||
|
'<b>уменьшился</b>',
|
||||||
|
'<b>откусила злая гадюка</b>',
|
||||||
|
'<b>продали евреи, он стал дешевле</b>',
|
||||||
|
'<b>схуднул</b>'
|
||||||
|
'<b>отрицательно вырос</b>'
|
||||||
|
)
|
||||||
|
return size_change
|
@ -1,4 +1,4 @@
|
|||||||
Кожаный бабах
|
кожаный бабах
|
||||||
пистон
|
пистон
|
||||||
дуршлаг
|
дуршлаг
|
||||||
елдак
|
елдак
|
||||||
@ -93,3 +93,4 @@
|
|||||||
венерический суккуб
|
венерический суккуб
|
||||||
боец невидимого фронта
|
боец невидимого фронта
|
||||||
сморчок
|
сморчок
|
||||||
|
джечник
|
@ -1,7 +1,8 @@
|
|||||||
|
|
||||||
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
|
||||||
|
from handlers import sendalarm
|
||||||
|
from handlers import davinci
|
||||||
|
|
||||||
|
@ -1,15 +1,49 @@
|
|||||||
import openai
|
import openai
|
||||||
from aiogram import types
|
from aiogram import types, Dispatcher
|
||||||
from create_bot import dp
|
from create_bot import dp, bot
|
||||||
|
import logging
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
token = '5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc'
|
token = '5947963644:AAF_GKgMmU5ovqMpc1KXIpcf4aN0JMyKPqc'
|
||||||
openai.api_key = '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'
|
#openai.api_key = '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'
|
||||||
|
openai.api_key = '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'
|
||||||
|
ban_comands = ['/dick','/topdick']
|
||||||
|
|
||||||
|
# Image_promt =['тупица сгенерируй','тупица сгенерируй картинку',\
|
||||||
|
# 'тупица отправь картинку', 'тупица отправь картинку']
|
||||||
|
|
||||||
|
|
||||||
|
@dp.message_handler(commands=['image'])
|
||||||
|
async def send_image(message: types.Message):
|
||||||
|
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('у меня не получилось')
|
||||||
|
|
||||||
@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.text.lower().startswith('тупица'):
|
||||||
|
|
||||||
|
## TODO уточнить остатки токенов и денег
|
||||||
response = openai.Completion.create(
|
response = openai.Completion.create(
|
||||||
model="text-davinci-003",
|
model="text-davinci-003",
|
||||||
prompt=message.text[7:],
|
prompt=message.text[7:],
|
||||||
@ -22,5 +56,24 @@ async def send(message: types.Message):
|
|||||||
)
|
)
|
||||||
await message.reply(response['choices'][0]['text'])
|
await message.reply(response['choices'][0]['text'])
|
||||||
|
|
||||||
|
elif message.reply_to_message.from_user.is_bot:
|
||||||
|
response = openai.Completion.create(
|
||||||
|
model="text-davinci-003",
|
||||||
|
prompt=message.text,
|
||||||
|
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,201 +1,155 @@
|
|||||||
from aiogram import types
|
from aiogram import types, Dispatcher
|
||||||
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, DickRequest, UserDick
|
|
||||||
import random
|
import random
|
||||||
import traceback
|
|
||||||
from random import randint
|
from random import randint
|
||||||
import datetime
|
import datetime
|
||||||
from dicktxt import ForReadDict
|
from dicktxt import ForReadDict
|
||||||
|
from dicktxt import change_words
|
||||||
size_change = None
|
from pipisa_functions import pipisa_time
|
||||||
dick_plus = None
|
from DBChat import user_operations
|
||||||
dick_minus = None
|
from TopMsg import topdicks
|
||||||
|
from handlers import post_cleaner
|
||||||
|
|
||||||
|
|
||||||
def get_dick(user_id, chat_id):
|
#### https://docs.python-arango.com/en/main/
|
||||||
dick = session.query(UserDick).filter(UserDick.user_id == user_id, UserDick.chat_id == chat_id).first()
|
from arango import ArangoClient
|
||||||
return dick
|
|
||||||
|
|
||||||
def top_dick_message(delimiter, top_list):
|
admins = [317051301, 226765676]
|
||||||
message = f'🏆Топ 10 бубылд чата🏆\n\n'
|
|
||||||
f'🚀 {delimiter.join(map(str, top_list[0]))}\n'
|
|
||||||
f'💫 {delimiter.join(map(str, top_list[1]))}\n'
|
|
||||||
f'🍆 {delimiter.join(map(str, top_list[2]))}\n'
|
|
||||||
f'🍌 {delimiter.join(map(str, top_list[3]))}\n'
|
|
||||||
f'🐍 {delimiter.join(map(str, top_list[4]))}\n'
|
|
||||||
f'🐎 {delimiter.join(map(str, top_list[5]))}\n'
|
|
||||||
f'🌭 {delimiter.join(map(str, top_list[6]))}\n'
|
|
||||||
f'🌶 {delimiter.join(map(str, top_list[7]))}\n'
|
|
||||||
f'👌 {delimiter.join(map(str, top_list[8]))}\n'
|
|
||||||
f'💩 {delimiter.join(map(str, top_list[9]))}\n'
|
|
||||||
return message
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=["maintenance"])
|
|
||||||
async def stop_for_maintenance(message: types.Message): # остановить бота
|
|
||||||
if message.from_user.id == 226765676:
|
|
||||||
bot.send_message(message.from_user.id, "Блять. Как же больно. Мне холодно. Наверное это конец...")
|
|
||||||
close_all_sessions()
|
|
||||||
dp.stop_polling()
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=["dick"])
|
@dp.message_handler(commands=["dick"])
|
||||||
async def up_dick(message: types.Message): # рандомайзер
|
async def up_dick(message: types.Message):
|
||||||
|
#global size_change, dick_plus, dick_minus
|
||||||
|
|
||||||
numb = randint(-6, 10)
|
if message.from_user.id in admins or message.chat.type != 'private':
|
||||||
global size_change, dick_plus, dick_minus
|
|
||||||
dick_plus = ('<b>вырос</b>', '<b>подрос</b>', '<b>привстал</b>')
|
|
||||||
dick_minus = ('<b>уменьшился</b>', '<b>откусила злая гадюка</b>', '<b>продали евреи, он стал дешевле</b>')
|
arango_client = ArangoClient(hosts='https://arango.guaranteedstruggle.host')
|
||||||
if numb > 0:
|
pipisa_db = arango_client.db('pipisa', username='root', password='stolendick527')
|
||||||
size_change = random.choice(dick_plus) + '🚀'
|
dicks_collection = pipisa_db.collection('dicks')
|
||||||
# if numb = 0 size_change = random.choice(const) TODO
|
|
||||||
|
|
||||||
|
|
||||||
|
result = randint(-9,10) # главный рандомайзер
|
||||||
|
|
||||||
|
try: ## Тело метода прокрутки
|
||||||
|
|
||||||
|
candidate_cursor = 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()
|
||||||
|
|
||||||
|
if pipisa_time.rolltime(check_datetime=user['datetimes'], curr_time=datetime.datetime.now()):
|
||||||
|
|
||||||
|
updetedDick = user["dick_size"] + result ## получение итогового результата
|
||||||
|
|
||||||
|
pipisa_word = ForReadDict.RandomDick()
|
||||||
|
change_word = change_words.ChangeWord(result)
|
||||||
|
|
||||||
|
enl_message = await bot.send_message(message.chat.id, f"@\
|
||||||
|
{message.from_user.username}, \
|
||||||
|
ваш <em>{pipisa_word}</em> ,{change_word(result)} на <b>{abs(result)}</b> см!\n\ \
|
||||||
|
Теперь он равен <b>{updetedDick}</b> см! ")
|
||||||
|
|
||||||
|
post_cleaner.cleanup_sequence(user['user_id'], enl_message.id, message.chat.id)
|
||||||
|
|
||||||
|
|
||||||
|
try:
|
||||||
|
dt = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
||||||
|
metadata = dicks_collection.insert(
|
||||||
|
{
|
||||||
|
'_key': user['_key'], ## этот нужен чтобы апдейт прилетал тому же юзеру
|
||||||
|
'user_id': user['user_id'],
|
||||||
|
'user_fullname': message.from_user.full_name,
|
||||||
|
'dick_size': updetedDick,
|
||||||
|
'datetimes': datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
|
||||||
|
'attempts': user['attempts'] + 1,
|
||||||
|
'chat_id': message.chat.id
|
||||||
|
},
|
||||||
|
overwrite_mode='update'
|
||||||
|
)
|
||||||
|
print(f'Успешно апдейтнули пипису @{message.from_user.username}')
|
||||||
|
except Exception as update_err:
|
||||||
|
print(f'ошибка в обновлении пиписы @{message.from_user.username}')
|
||||||
|
print(update_err)
|
||||||
else:
|
else:
|
||||||
size_change = random.choice(dick_minus) + '✂'
|
error_msg = await bot.send_message(message.chat.id, f'@\
|
||||||
|
{message.from_user.username},🚫 вы уже крутили пипису, \
|
||||||
|
ее размер <b>{user["dick_size"]}</b> см! ')
|
||||||
|
|
||||||
|
post_cleaner.cleanup_sequence(user['user_id'], error_msg.id, error_msg.chat.id)
|
||||||
|
|
||||||
# работа с таблицей
|
|
||||||
try:
|
try:
|
||||||
dick_request = DickRequest(size_change=numb, date_changed=datetime.datetime.now())
|
metadata = dicks_collection.insert(
|
||||||
b = session.query(User).filter(User.user_id == message.from_user.id).first()
|
{
|
||||||
if b:
|
'_key': user['_key'], ### этот нужен чтобы апдейт прилетал тому же юзеру
|
||||||
last_time = b.date_updated
|
'user_id': user['user_id'] ,
|
||||||
now_time = datetime.datetime.now()
|
'user_fullname': message.from_user.full_name,
|
||||||
diff = now_time.day - last_time.day
|
'dick_size': user['dick_size'] ,
|
||||||
|
'datetimes': user['datetimes'] ,
|
||||||
|
'attempts': user['attempts'] + 1,
|
||||||
|
'chat_id': message.chat.id,
|
||||||
|
},
|
||||||
|
overwrite_mode='update'
|
||||||
|
)
|
||||||
|
print(f'Успешно апдейтнули попытку крутить пипису @{message.from_user.username}')
|
||||||
|
except Exception as roll_err:
|
||||||
|
print(f'ошибка в обновлении попытки крутить пипису @{message.from_user.username}')
|
||||||
|
print(roll_err)
|
||||||
|
|
||||||
if abs(diff) <= 0:
|
print(f'пользователь @{message.from_user.username} уже крутил')
|
||||||
try:
|
|
||||||
current_dick = b.dick_size
|
|
||||||
await bot.send_message(message.chat.id, f'@{message.from_user.username},'
|
|
||||||
f'🚫 вы уже крутили пипису, '
|
|
||||||
f'ее размер <b>{current_dick}</b> см! ')
|
|
||||||
except Exception as e:
|
|
||||||
print('ошибка в if diff <=0 ' + "\n".join(traceback.format_tb(e.__traceback__)))
|
|
||||||
session.rollback()
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
print('пользователь уже крутил')
|
|
||||||
|
|
||||||
|
else: ## если пользователь не найден то создаём ему запись в БД
|
||||||
|
|
||||||
|
if result > 0:
|
||||||
|
new_reply = await bot.send_message(message.chat.id, f'@\
|
||||||
|
{message.from_user.username}, ваш писюн показал головку 🚀\n\
|
||||||
|
на <b>{abs(result)}</b> см!\n\
|
||||||
|
Теперь он равен <b>{result}</b> см!')
|
||||||
else:
|
else:
|
||||||
try:
|
new_reply =await bot.send_message(message.chat.id, f'@\
|
||||||
usr = session.query(User)
|
{message.from_user.username}, добро пожаловать в игру!🚀\n\
|
||||||
a = usr.filter(User.user_id == message.from_user.id).first() # запрос на поиск в таблице
|
Ты стартуешь не с лучшей позиции, природа наградила тебя всего <b>{result}</b> см 😬')
|
||||||
dick = get_dick(User.user_id, message.chat.id)
|
|
||||||
updated_dick = dick.dick_size = (dick.dick_size + numb)
|
|
||||||
dick_request.user = a
|
|
||||||
a.date_updated = datetime.datetime.now()
|
|
||||||
session.add(dick_request)
|
|
||||||
session.flush()
|
|
||||||
session.commit() # обновляет запись
|
|
||||||
randomword = ForReadDict.RandomDick()
|
|
||||||
|
|
||||||
await bot.send_message(message.chat.id, f'@{message.from_user.username}, '
|
user_operations.UserCreate(dicks_collection, message, result, new_reply.id)
|
||||||
f'ваш <em>{randomword}</em> {size_change}'
|
|
||||||
f'на <b>{abs(numb)}</b> см!\n'
|
|
||||||
f'Теперь он равен <b>{updated_dick}</b> см! ')
|
|
||||||
except Exception as e:
|
|
||||||
session.rollback()
|
|
||||||
print('ошибка в обновлении пиписы, трейсбек-бек-бек: ' +
|
|
||||||
"\n".join(traceback.format_tb(e.__traceback__)))
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
print('пиписа обновлена')
|
|
||||||
|
|
||||||
|
print(f'Успешно добавлен пользователь @{message.from_user.username}')
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
except Exception as crit_err:
|
||||||
|
print('ошибка DB в /dick')
|
||||||
|
print(crit_err)
|
||||||
else:
|
else:
|
||||||
session.close()
|
await message.reply('Растить елду можно только в общих чатах!')
|
||||||
try:
|
|
||||||
numb = randint(1, 10)
|
|
||||||
int_table = User(user_id=message.from_user.id,
|
|
||||||
user_fullname=message.from_user.full_name,
|
|
||||||
date_updated=datetime.datetime.now(),
|
|
||||||
date_created=datetime.datetime.now())
|
|
||||||
new_dick = DickRequest(user_id=message.from_user.id,
|
|
||||||
chat_id=message.chat.id,
|
|
||||||
dick_size=numb)
|
|
||||||
|
|
||||||
dick_request.user = int_table
|
|
||||||
session.add(int_table)
|
|
||||||
session.add(new_dick)
|
|
||||||
session.add(dick_request)
|
|
||||||
session.flush()
|
|
||||||
session.commit()
|
|
||||||
session.close()
|
|
||||||
await bot.send_message(message.chat.id, f'@{message.from_user.username}, '
|
|
||||||
f'ваш писюн показал головку 🚀 \n'
|
|
||||||
f'на <b>{abs(numb)}</b> см!\n'
|
|
||||||
f'Теперь он равен <b>{numb}</b> см!')
|
|
||||||
except Exception as e:
|
|
||||||
print('ошибка в добавлении нового пользователя, трейсбек: ' +
|
|
||||||
'\n'.join(traceback.format_tb(e.__traceback__)))
|
|
||||||
session.rollback()
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
print('Успешно добавлен пользователь')
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
session.rollback()
|
|
||||||
print('ошибка в основном коде, трейсбек:' + '\n'.join(traceback.format_tb(e.__traceback__)))
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=["topdick"])
|
@dp.message_handler(commands=["topdick"])
|
||||||
async def send_top_chat(message: types.Message):
|
async def send_topchat(message: types.Message): # тело метода лучших результатов отдельной конфы
|
||||||
try:
|
if message.from_user.id in admins or message.chat.type != 'private':
|
||||||
top = []
|
|
||||||
chats = message.chat.id
|
|
||||||
changechat_id = session.query(DickRequest).filter(DickRequest.user_id == message.from_user.id,
|
|
||||||
DickRequest.chat_id == chats).first()
|
|
||||||
dicks = session.query(DickRequest)\
|
|
||||||
.filter(DickRequest.chat_id == chats)\
|
|
||||||
.limit(10).all()\
|
|
||||||
.order_by(DickRequest.dick_size.desc())
|
|
||||||
for dick in dicks:
|
|
||||||
user = session.query(User.user_fullname).filter(User.user_id == dick.user_id)
|
|
||||||
top.append([user.full_name, dick.dick_size])
|
|
||||||
delimiter = ': '
|
|
||||||
|
|
||||||
# проверка на длину списка, если меньше limit, то:
|
|
||||||
|
|
||||||
size_len = len(top)
|
|
||||||
if size_len < 10:
|
|
||||||
len_minus = 10 - size_len
|
|
||||||
top.extend([('-'*len_minus),""])
|
|
||||||
await bot.send_message(message.chat.id, top_dick_message(delimiter, top))
|
|
||||||
session.close()
|
|
||||||
# если все нормально и участников минимум 10
|
|
||||||
|
|
||||||
|
dickos = topdicks.topdicks_conf(message.chat.id)
|
||||||
|
await bot.send_message(message.chat.id, '🏆Топ 10 бубылд чата🏆\n\n' + dickos)
|
||||||
else:
|
else:
|
||||||
|
await message.reply('Работает только в общих чатах!\n'\
|
||||||
|
'Вы мсжете посмотреть топ по миру: /globaldick')
|
||||||
|
|
||||||
|
|
||||||
await bot.send_message(message.chat.id, top_dick_message(delimiter, top))
|
|
||||||
except Exception as e:
|
|
||||||
session.rollback()
|
|
||||||
print('ошибка в /topdick. Трейсбэк:\n' + '\n'.join(traceback.format_tb(e.__traceback__)))
|
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=["globaldick"])
|
@dp.message_handler(commands=["globaldick"])
|
||||||
async def send_global_top(message: types.Message):
|
async def send_global_top(message: types.Message): # тело метода лучших результатов
|
||||||
try:
|
tops = topdicks.topdicks_global(message.chat.id,)
|
||||||
top = []
|
|
||||||
dicks = session.query(DickRequest) \
|
|
||||||
.limit(10).all() \
|
|
||||||
.order_by(DickRequest.dick_size.desc())
|
|
||||||
for dick in dicks:
|
|
||||||
user = session.query(User.user_fullname).filter(User.user_id == dick.user_id)
|
|
||||||
top.append([user.full_name, dick.dick_size])
|
|
||||||
delimiter = ': '
|
|
||||||
|
|
||||||
await bot.send_message(message.chat.id, top_dick_message(delimiter, top))
|
await bot.send_message(message.chat.id, '🏆Топ 10 пипис в мире🏆\n\n' + tops)
|
||||||
except Exception as e:
|
|
||||||
session.rollback()
|
|
||||||
print('ошибка в /globaldick, трейсбэк:\n' + '\n'.join(traceback.format_tb(e.__traceback__)))
|
|
||||||
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)
|
|
||||||
|
69
handlers/post_cleaner.py
Normal file
69
handlers/post_cleaner.py
Normal file
@ -0,0 +1,69 @@
|
|||||||
|
import datetime
|
||||||
|
from arango import ArangoClient
|
||||||
|
from create_bot import bot
|
||||||
|
|
||||||
|
def get_remove_collection():
|
||||||
|
arango_client = ArangoClient(hosts=CONFIG['databaso']['host'] )
|
||||||
|
pipisa_db = arango_client.db(
|
||||||
|
CONFIG['databaso']['base'],
|
||||||
|
username=CONFIG['databaso']['user'],
|
||||||
|
password=CONFIG['databaso']['pass']
|
||||||
|
)
|
||||||
|
remove_collection = pipisa_db.collection(CONFIG['databaso']['deleteCollection'])
|
||||||
|
|
||||||
|
return remove_collection
|
||||||
|
|
||||||
|
async def cleanup_sequence(userID: int, messageID:int, chatID:int):
|
||||||
|
|
||||||
|
#remove last reply and check posts older than 5 minutes to delete
|
||||||
|
remove_collection = get_remove_collection
|
||||||
|
try:
|
||||||
|
clean_candidate = remove_collection.find(
|
||||||
|
{
|
||||||
|
'_key': userID,
|
||||||
|
'chat_id': chatID
|
||||||
|
},
|
||||||
|
skip=0, limit=1000)
|
||||||
|
|
||||||
|
prev_enl_message_id = clean_candidate["messageID"]
|
||||||
|
await bot.delete_message(prev_enl_message_id) #delete last reply to this user
|
||||||
|
|
||||||
|
current_time = datetime.datetime.now()
|
||||||
|
too_old = current_time-datetime.timedelta(min=5)
|
||||||
|
old_posts = remove_collection.find(
|
||||||
|
{
|
||||||
|
'chat_id': chatID,
|
||||||
|
},
|
||||||
|
skip=0, limit=1000)
|
||||||
|
|
||||||
|
for old_Post in old_posts:
|
||||||
|
if old_Post["datetimes"] <= too_old:
|
||||||
|
to_delete_ID = old_Post["messageID"]
|
||||||
|
to_delete_UserId = old_Post['_key']
|
||||||
|
|
||||||
|
await bot.delete_message(to_delete_ID) #delete old message
|
||||||
|
old_posts.delete(to_delete_UserId) #delete old message DB entry
|
||||||
|
|
||||||
|
|
||||||
|
except Exception as autoCleanErr:
|
||||||
|
print(autoCleanErr)
|
||||||
|
|
||||||
|
#write last reply post ID to db
|
||||||
|
remove_collection = get_remove_collection
|
||||||
|
message_time = datetime.datetime.now()
|
||||||
|
|
||||||
|
try:
|
||||||
|
lastReply = {
|
||||||
|
'_key': userID,
|
||||||
|
'datetimes': message_time,
|
||||||
|
'chat_id': chatID,
|
||||||
|
'messageID': messageID
|
||||||
|
}
|
||||||
|
|
||||||
|
metadata = remove_collection.insert(lastReply, overwrite_mode='update') #write last reply time
|
||||||
|
|
||||||
|
except Exception as lastWriteErr:
|
||||||
|
print(lastWriteErr)
|
||||||
|
|
||||||
|
return 0
|
||||||
|
|
@ -1,8 +1,8 @@
|
|||||||
import traceback
|
|
||||||
|
|
||||||
from create_bot import dp, bot
|
from create_bot import dp, bot
|
||||||
from aiogram import types, Dispatcher
|
from aiogram import types, Dispatcher
|
||||||
from sqdb import User, session
|
from sqdb import user, session
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@dp.message_handler(commands=['sendall'])
|
@dp.message_handler(commands=['sendall'])
|
||||||
@ -12,20 +12,17 @@ async def sendall(message: types.Message):
|
|||||||
if message.from_user.id == 226765676:
|
if message.from_user.id == 226765676:
|
||||||
text = message.text[9:]
|
text = message.text[9:]
|
||||||
try:
|
try:
|
||||||
chats_id = session.query(User.chat_id).distinct()
|
chats_id = -1001355646177
|
||||||
for row in chats_id:
|
|
||||||
try:
|
try:
|
||||||
await bot.send_message(row[0], text)
|
await bot.send_message(chats_id, text)
|
||||||
except Exception as e:
|
except:
|
||||||
print('вероятно бота нет в чате, на всякий случай трейсбек:\n' +
|
print('вероятно бота нет в чате')
|
||||||
"\n".join(traceback.format_tb(e.__traceback__)))
|
|
||||||
await bot.send_message(message.from_user.id, 'Сообщение успешно отправлено во все чаты')
|
await bot.send_message(message.from_user.id, 'Сообщение успешно отправлено во все чаты')
|
||||||
except Exception as e:
|
except:
|
||||||
session.rollback()
|
session.rollback()
|
||||||
print('error sendall, traceback:\n' +
|
print('eror sendall')
|
||||||
"\n".join(traceback.format_tb(e.__traceback__)))
|
finally: session.close_all()
|
||||||
finally:
|
|
||||||
session.close_all()
|
|
||||||
|
|
||||||
|
|
||||||
# def register_handlers_test(dp: Dispatcher):
|
# def register_handlers_test(dp: Dispatcher):
|
||||||
|
@ -1,28 +1,27 @@
|
|||||||
from aiogram import types
|
from aiogram import types, Dispatcher
|
||||||
|
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)
|
|
||||||
|
@ -1,28 +1,27 @@
|
|||||||
from aiogram import types
|
from aiogram import types, Dispatcher
|
||||||
from create_bot import dp
|
from create_bot import dp, bot
|
||||||
import datetime
|
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(2024, 12, 31) # дата нового года
|
new_year = 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+datetime.timedelta(hours=15) # +14 часов от сервера
|
||||||
ct2 = ct1.strftime('%H:%M') # форматирует дату/время просто во время(ч, м)
|
ct2 = ct1.strftime('%H:%M') # форматирует датувремя просто во время(ч,м)
|
||||||
date_difference = (new_year - ct).days # отнимает от нг текущее время и получаем разницу в днях
|
raznitsa = (new_year - ct).days #отнимает от нг текущее время и получаем разницу в днях
|
||||||
|
|
||||||
days = ['день', 'дня', 'дней']
|
days = ['день', 'дня', 'дней']
|
||||||
if date_difference % 10 == 1 and date_difference % 100 != 11:
|
if raznitsa % 10 == 1 and raznitsa % 100 != 11:
|
||||||
p = 0
|
p = 0
|
||||||
elif 2 <= date_difference % 10 <= 4 and (date_difference % 100 < 10 or date_difference % 100 >= 20):
|
elif 2 <= raznitsa % 10 <= 4 and (raznitsa % 100 < 10 or raznitsa % 100 >= 20):
|
||||||
p = 1
|
p = 1
|
||||||
else:
|
else:
|
||||||
p = 2
|
p = 2
|
||||||
num = (str(date_difference) + ' ' + days[p])
|
num=(str(raznitsa) + ' ' + days[p])
|
||||||
|
|
||||||
await message.reply(f'Сегодня {cd} \nВремя: {ct2} \nДо Нового Года осталось {num}')
|
await message.reply(f'Сегодня {cd} \nВремя: {ct2} \nДо Нового Года осталось {num}')
|
||||||
|
|
||||||
|
|
||||||
# def register_handlers_time(dp: Dispatcher):
|
def register_handlers_time(dp: Dispatcher):
|
||||||
# dp.register_message_handler(send_time)
|
dp.register_message_handler(send_time)
|
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)
|
2
requairements.txt
Normal file
2
requairements.txt
Normal file
@ -0,0 +1,2 @@
|
|||||||
|
aiogram
|
||||||
|
python-arango
|
@ -1,6 +0,0 @@
|
|||||||
aiogram
|
|
||||||
sqdb
|
|
||||||
sqlalchemy
|
|
||||||
graypy
|
|
||||||
openai
|
|
||||||
psycopg2
|
|
35
sqdb.py
35
sqdb.py
@ -1,51 +1,36 @@
|
|||||||
from sqlalchemy import create_engine, MetaData, Table, Integer, String, BIGINT, \
|
from sqlalchemy import create_engine, MetaData, Table, Integer, String, BIGINT, \
|
||||||
Column, DateTime, ForeignKey, Numeric
|
Column, DateTime, ForeignKey, Numeric
|
||||||
from sqlalchemy.ext.declarative import declarative_base
|
from sqlalchemy.ext.declarative import declarative_base
|
||||||
from sqlalchemy.orm import sessionmaker, relationship
|
from sqlalchemy.orm import sessionmaker
|
||||||
from sqlalchemy import select, update
|
from sqlalchemy import select, update
|
||||||
|
|
||||||
engine = create_engine('postgresql+psycopg2://postgres:postgres@postgres.vdk2ch.ru:5432/jack_bot', echo=True)
|
engine = create_engine('postgresql+psycopg2://postgres:postgres@postgres.vdk2ch.ru:5432/jack_bot', echo=True)
|
||||||
Session = sessionmaker(bind=engine)
|
Session = sessionmaker(bind=engine)
|
||||||
session = Session()
|
session = Session()
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
Base = declarative_base()
|
Base = declarative_base()
|
||||||
|
|
||||||
|
|
||||||
class User(Base):
|
|
||||||
|
class user(Base):
|
||||||
__tablename__ = 'users'
|
__tablename__ = 'users'
|
||||||
|
|
||||||
id = Column(Integer, primary_key=True)
|
id = Column(Integer, primary_key=True)
|
||||||
user_id = Column(Integer)
|
user_id = Column(Integer)
|
||||||
date_created = (Column(DateTime))
|
|
||||||
user_fullname = Column(String)
|
user_fullname = Column(String)
|
||||||
date_updated = (Column(DateTime))
|
|
||||||
dick_requests = relationship("DickRequest", back_populates="user")
|
|
||||||
user_dicks = relationship("UserDick", back_populates="user")
|
|
||||||
|
|
||||||
|
|
||||||
class UserDick(Base):
|
|
||||||
__tablename__ = 'user_dicks'
|
|
||||||
|
|
||||||
id = Column(Integer, primary_key=True)
|
|
||||||
user_id = Column(Integer, ForeignKey("users.user_id"))
|
|
||||||
dick_size = (Column(Integer))
|
dick_size = (Column(Integer))
|
||||||
|
datetimes = (Column(DateTime))
|
||||||
chat_id = Column(BIGINT)
|
chat_id = Column(BIGINT)
|
||||||
user = relationship("User", back_populates="user_dicks")
|
|
||||||
|
|
||||||
|
|
||||||
class DickRequest(Base):
|
|
||||||
__tablename__ = 'dick_requests'
|
|
||||||
id = Column(Integer, primary_key=True)
|
|
||||||
size_change = Column(Integer)
|
|
||||||
user_id = Column(Integer, ForeignKey("users.user_id"))
|
|
||||||
user = relationship("User", back_populates="dick_requests")
|
|
||||||
date_changed = (Column(DateTime))
|
|
||||||
|
|
||||||
session.close()
|
session.close()
|
||||||
engine.dispose()
|
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()
|
||||||
# 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)
|
# print(top)
|
||||||
|
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user