From 66165a9d5d18dea92fc0105a73a4944da0b8502a Mon Sep 17 00:00:00 2001 From: Alexander Graf <17130992+aandergr@users.noreply.github.com> Date: Sat, 11 Apr 2026 12:26:09 +0200 Subject: [PATCH] Improve typing on import_session() --- instaloader/__main__.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/instaloader/__main__.py b/instaloader/__main__.py index e76dfea..7b2670d 100644 --- a/instaloader/__main__.py +++ b/instaloader/__main__.py @@ -82,7 +82,7 @@ def filterstr_to_filterfunc(filter_str: str, item_type: type): return filterfunc -def get_cookies_from_instagram(domain, browser, cookie_file='', cookie_name=''): +def get_cookies_from_instagram(domain: str, browser: str, cookie_file: Optional[str] = None): supported_browsers = { "brave": browser_cookie3.brave, "chrome": browser_cookie3.chrome, @@ -114,13 +114,10 @@ def get_cookies_from_instagram(domain, browser, cookie_file='', cookie_name=''): raise LoginException(f"No cookies found for Instagram in {browser}, " f"Are you logged in successfully in {browser}?") - if cookie_name: - return cookies.get(cookie_name, {}) - else: - return cookies + return cookies -def import_session(browser, instaloader, cookiefile): +def import_session(browser: str, instaloader: Instaloader, cookiefile: Optional[str]): cookie = get_cookies_from_instagram('instagram', browser, cookiefile) if cookie is not None: instaloader.context.update_cookies(cookie)