code reformat

This commit is contained in:
2023-10-30 23:30:16 +07:00
parent ce88bb6dc8
commit 57f401ee5f
8 changed files with 35 additions and 29 deletions
+2 -1
View File
@@ -1,4 +1,5 @@
from vkbottle.user import Message from vkbottle.user import Message
from main import user from main import user
TOKEN = "test" TOKEN = "test"
@@ -19,5 +20,5 @@ async def edit_message(
kwargs.setdefault('dont_parse_links', False) kwargs.setdefault('dont_parse_links', False)
return await user.api.messages.edit( return await user.api.messages.edit(
**kwargs **kwargs
) )
+2 -1
View File
@@ -9,6 +9,7 @@ class Base(DeclarativeBase):
class UsersTokens(Base): class UsersTokens(Base):
__tablename__ = "users_tokens" __tablename__ = "users_tokens"
id = Column("id", BigInteger, Identity(start=1, increment=1, minvalue=1, cycle=False, cache=1), nullable=False, primary_key=True) id = Column("id", BigInteger, Identity(start=1, increment=1, minvalue=1, cycle=False, cache=1), nullable=False,
primary_key=True)
vkid = Column("vkid", BigInteger, nullable=False) vkid = Column("vkid", BigInteger, nullable=False)
token = Column("token", Text, nullable=False) token = Column("token", Text, nullable=False)
+2 -1
View File
@@ -1,5 +1,6 @@
from loguru import logger from loguru import logger
from sqlalchemy import select, update from sqlalchemy import select
import db import db
+3 -1
View File
@@ -1,9 +1,11 @@
import asyncio import asyncio
import sys import sys
from loguru import logger
from vkbottle.user import User from vkbottle.user import User
import db.requests import db.requests
from routes import labelers from routes import labelers
from loguru import logger
logger.add(sys.stderr, format="{time} {level} {message}", filter="my_module", level="INFO") logger.add(sys.stderr, format="{time} {level} {message}", filter="my_module", level="INFO")
user = None user = None
+16 -16
View File
@@ -17,32 +17,32 @@ DD_SCRIPT = (
@bl.message( @bl.message(
Permission(), Permission(),
text=[prefix_dd + " <count:int>"], text=[prefix_dd + " <count:int>"],
) )
async def dd_handler(message: Message, count: int = 2): async def dd_handler(message: Message, count: int = 2):
await user.api.execute( await user.api.execute(
DD_SCRIPT % ( DD_SCRIPT % (
count, count,
message.peer_id, message.peer_id,
message.from_id, message.from_id,
int(datetime.datetime.now().timestamp()) int(datetime.datetime.now().timestamp())
) )
) )
@bl.message( @bl.message(
Permission(), Permission(),
text=["все", "всё"], text=["все", "всё"],
) )
async def dd_all_handler(message: Message): async def dd_all_handler(message: Message):
count = 1000 count = 1000
await user.api.execute( await user.api.execute(
DD_SCRIPT % ( DD_SCRIPT % (
count, count,
message.peer_id, message.peer_id,
message.from_id, message.from_id,
int(datetime.datetime.now().timestamp()) int(datetime.datetime.now().timestamp())
) )
) )
+4 -3
View File
@@ -1,15 +1,16 @@
import asyncio import asyncio
from vkbottle.user import Message, UserLabeler from vkbottle.user import Message, UserLabeler
from config import prefix_bot, edit_message from config import prefix_bot, edit_message
from custom_rules.permission import Permission from custom_rules.permission import Permission
bl = UserLabeler(custom_rules=Permission) bl = UserLabeler(custom_rules=Permission)
@bl.message( @bl.message(
Permission(), Permission(),
text=[prefix_bot + " игра"], text=[prefix_bot + " игра"],
) )
async def game(message: Message): async def game(message: Message):
loader = ( loader = (
+4 -5
View File
@@ -1,11 +1,10 @@
import pytz
import git import git
import pytz
from custom_rules.permission import Permission
from vkbottle.user import Message, UserLabeler from vkbottle.user import Message, UserLabeler
from config import edit_message, prefix_bot
from utils import rewards_list
from config import edit_message, prefix_bot
from custom_rules.permission import Permission
from utils import rewards_list
bl = UserLabeler(custom_rules=Permission) bl = UserLabeler(custom_rules=Permission)
+2 -1
View File
@@ -1,6 +1,7 @@
from custom_rules.permission import Permission
from vkbottle.user import Message, UserLabeler from vkbottle.user import Message, UserLabeler
from config import edit_message, prefix_bot from config import edit_message, prefix_bot
from custom_rules.permission import Permission
from utils import rewards_list from utils import rewards_list
listToStr = ' '.join(map(str, rewards_list)) listToStr = ' '.join(map(str, rewards_list))