From 6d19016f7809dc0458994e1504edb7de3ae57504 Mon Sep 17 00:00:00 2001 From: d60 Date: Sat, 24 Aug 2024 00:11:28 +0900 Subject: [PATCH] Fixed key error --- twikit/__init__.py | 2 +- twikit/client/client.py | 37 ++++++++++++++++++++++++++++--------- 2 files changed, 29 insertions(+), 10 deletions(-) diff --git a/twikit/__init__.py b/twikit/__init__.py index 59facd9c..442fa4d4 100644 --- a/twikit/__init__.py +++ b/twikit/__init__.py @@ -7,7 +7,7 @@ A Python library for interacting with the Twitter API. """ -__version__ = '2.1.2' +__version__ = '2.1.3' import asyncio import os diff --git a/twikit/client/client.py b/twikit/client/client.py index e4c3be1a..1f865f4d 100644 --- a/twikit/client/client.py +++ b/twikit/client/client.py @@ -17,12 +17,24 @@ from ..bookmark import BookmarkFolder from ..community import Community, CommunityMember from ..constants import TOKEN -from ..errors import (AccountLocked, AccountSuspended, BadRequest, - CouldNotTweet, Forbidden, InvalidMedia, NotFound, - RequestTimeout, ServerError, TooManyRequests, - TweetNotAvailable, TwitterException, Unauthorized, - UserNotFound, UserUnavailable, - raise_exceptions_from_response) +from ..errors import ( + AccountLocked, + AccountSuspended, + BadRequest, + CouldNotTweet, + Forbidden, + InvalidMedia, + NotFound, + RequestTimeout, + ServerError, + TooManyRequests, + TweetNotAvailable, + TwitterException, + Unauthorized, + UserNotFound, + UserUnavailable, + raise_exceptions_from_response +) from ..geo import Place, _places_from_response from ..group import Group, GroupMessage from ..list import List @@ -32,8 +44,15 @@ from ..trend import Location, PlaceTrend, PlaceTrends, Trend from ..tweet import CommunityNote, Poll, ScheduledTweet, Tweet, tweet_from_data from ..user import User -from ..utils import (Flow, Result, build_tweet_data, build_user_data, - find_dict, find_entry_by_type, httpx_transport_to_url) +from ..utils import ( + Flow, + Result, + build_tweet_data, + build_user_data, + find_dict, + find_entry_by_type, + httpx_transport_to_url +) from .gql import GQLClient from .v11 import V11Client @@ -388,7 +407,7 @@ async def login( if not flow.response['subtasks']: return - self._user_id = flow.response['subtasks'][0]['open_account']['user']['id_str'] + self._user_id = find_dict(flow.response, 'id_str', find_one=True)[0] if flow.task_id == 'LoginTwoFactorAuthChallenge': if totp_secret is None: