github/github/classes/chat_controller.py at master · codex-bot/github · GitHub

34 lines (27 loc) · 1021 Bytes

import random

import string

from time import time

from settings import USERS_COLLECTION_NAME

class ChatController:

def __init__(self, sdk):

self.sdk = sdk

def get_chat(self, chat_id, bot_id):

registered_chat = self.sdk.db.find_one(USERS_COLLECTION_NAME, {'chat': chat_id, 'bot': bot_id})

if registered_chat:

return registered_chat

else:

user_token = self.generate_user_token()

new_chat = {

'chat': chat_id,

'user': user_token,

'bot': bot_id,

'dt_register': time(),

'branch': "*",

'verbose': True

}

self.sdk.log("New user registered with token {}".format(user_token))

self.sdk.db.insert(USERS_COLLECTION_NAME, new_chat)

return new_chat

@staticmethod

def generate_user_token():

return ''.join(random.SystemRandom().choice(string.ascii_uppercase + string.digits) for _ in range(8))

Expand file tree