redsandy/partner-tg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Добавлено новое окружение с COMPANY_KEY и обновлена логика регистрации пользователя в функции send_welcome для передачи company_key в запросе.

Browse Source
This commit is contained in:
Redsandyg 2025-06-03 14:03:55 +03:00
parent 9f215802a6
commit baf9cd0880
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.env
View File

@ -1,2 +1,3 @@
TG_BOT_TOKEN=7727936968:AAELs0gg5VNGzg6wMnG66CkpYaEv0K1Tp5w TG_BOT_TOKEN=7727936968:AAELs0gg5VNGzg6wMnG66CkpYaEv0K1Tp5w
API_URL=http://127.0.0.1:8000 API_URL=http://127.0.0.1:8000
COMPANY_KEY=re-premium-key

5
main.py
View File

@ -15,7 +15,7 @@ load_dotenv()
API_TOKEN = os.getenv('TG_BOT_TOKEN') API_TOKEN = os.getenv('TG_BOT_TOKEN')
API_URL = os.getenv('API_URL') API_URL = os.getenv('API_URL')
COMPANY_KEY = os.getenv('COMPANY_KEY')
bot = Bot(token=API_TOKEN) bot = Bot(token=API_TOKEN)
dp = Dispatcher() dp = Dispatcher()
@ -118,6 +118,7 @@ async def send_welcome(message: types.Message):
token = None token = None
error = None error = None
hash_value = sha256(f"{tg_id}sold".encode()).hexdigest() hash_value = sha256(f"{tg_id}sold".encode()).hexdigest()
company_key = COMPANY_KEY
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
try: try:
# 1. Пробуем авторизоваться по hash # 1. Пробуем авторизоваться по hash
@ -126,7 +127,7 @@ async def send_welcome(message: types.Message):
token = hash_value token = hash_value
else: else:
# 2. Если не найден — регистрируем # 2. Если не найден — регистрируем
async with session.post(f'{API_URL}/register', json={'tg_id': tg_id, 'chat_id': chat_id, 'name': name, 'login': login}) as reg_resp: async with session.post(f'{API_URL}/register', json={'tg_id': tg_id, 'chat_id': chat_id, 'name': name, 'login': login, 'company_key': company_key}) as reg_resp:
if reg_resp.status == 200: if reg_resp.status == 200:
# После регистрации снова пробуем авторизоваться # После регистрации снова пробуем авторизоваться
async with session.post(f'{API_URL}/tg_auth', json={'hash': hash_value}) as auth_resp2: async with session.post(f'{API_URL}/tg_auth', json={'hash': hash_value}) as auth_resp2: