fix imports

iamdual · Aug 13, 2024 · f32a2ff · f32a2ff
1 parent 9595d6d
commit f32a2ff
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/src/ua_generator/data/generator.py b/src/ua_generator/data/generator.py
@@ -3,8 +3,6 @@
 Copyright: 2022-2024 Ekin Karadeniz (github.com/iamdual)
 License: Apache License 2.0
 """
-from .browsers import chrome, safari, firefox, edge
-from .platforms import ios, android, linux, windows, macos
 from .. import utils, exceptions
 from ..options import Options
 
@@ -21,6 +19,8 @@ def __init__(self, device, platform, browser, options: Options):
         self.user_agent = self.__user_agent()
 
     def __platform_version(self):
+        from .platforms import ios, android, linux, windows, macos
+
         if self.platform == 'windows':
             return windows.get_version(options=self.options)
         elif self.platform == 'macos':
@@ -33,6 +33,8 @@ def __platform_version(self):
             return android.get_version(options=self.options)
 
     def __browser_version(self):
+        from .browsers import chrome, safari, firefox, edge
+
         if self.browser == 'chrome':
             return chrome.get_version(options=self.options)
         elif self.browser == 'safari':

diff --git a/src/ua_generator/user_agent.py b/src/ua_generator/user_agent.py
@@ -6,9 +6,8 @@
 import typing
 
 from . import utils, exceptions
-from .data import platforms, browsers  # TODO: learn why the error is triggered when moving this to the bottom
 from .client_hints import ClientHints
-from .data import devices, platforms_desktop, platforms_mobile
+from .data import devices, browsers, platforms, platforms_desktop, platforms_mobile
 from .data.generator import Generator
 from .headers import Headers
 from .options import Options