Python User.get_or_createの例

コード例 #1

ファイル: collect_users.py プロジェクト: init10xa/telegram_parser-1

def main():
    print('Starting script...')

    client = ApiHandler(API_ID, API_HASH)
    print('Auth - successfully!')

    messages = Message.select()
    print('Getting messages...\nGot {} messages'.format(len(messages)))

    user_ids = []

    for message in messages:
        if message.from_id and message.from_id not in user_ids:
            try:
                user_ids.append(message.from_id)
            except Exception:
                pass

    print('Got {} unique users'.format(len(user_ids)))

    users = client.get_users(user_ids)

    for user in users:
        User.get_or_create(user_id=user['id'],
                           username=user['username'],
                           first_name=user['first_name'],
                           last_name=user['last_name'])

    print('Done!')

コード例 #2

ファイル: user.py プロジェクト: artemmarkaryan/price_checker

def get_or_create(user_id: int, platform: str) -> User:
    platform = Platform.get(name=platform)
    return User.get_or_create(user_id=user_id, platform=platform)