add complete implementation for app sdk in python sdk

Author: parthlambdatest

selenium/lambdatest_selenium_driver/smartui_app_snapshot.py

@@ -233,9 +233,17 @@ def _process_screenshot_capture(self, driver: WebDriver, screenshot_name: str,
 
     def _process_upload_options(self, upload_request: UploadSnapshotRequest, options: Dict[str, str]):
         """Process upload-related options."""
+        self._set_upload_chunk_options(upload_request, options)
         self._set_navigation_options(upload_request, options)
         upload_request.crop_footer = "false"
         upload_request.crop_status_bar = "false"
+
+    def _set_upload_chunk_options(self, upload_request: UploadSnapshotRequest, options: Dict[str, str]):
+        """Set upload chunk related options."""
+        #get boolean value for upload chunk from options
+        upload_chunk = self._parse_boolean_option(options, "uploadChunk", False)
+        if upload_chunk:
+            upload_request.upload_chunk = "true"
 
     def _set_navigation_options(self, upload_request: UploadSnapshotRequest, options: Dict[str, str]):
         """Set navigation bar and status bar height options."""
@@ -255,11 +263,12 @@ def _handle_screenshot_capture(self, driver: WebDriver, screenshot_name: str,
         temp_dir = tempfile.mkdtemp(prefix=f"smartui_{screenshot_name}_")
 
         try:
+            full_page_util = FullPageScreenshotUtil(
+                    driver, temp_dir, config.test_type, config.precise_scroll
+            )
+            upload_request.device_name = full_page_util.device_name
             if config.full_page:
                 # Full-page screenshot
-                full_page_util = FullPageScreenshotUtil(
-                    driver, temp_dir, config.test_type, config.precise_scroll
-                )
                 result = full_page_util.capture_full_page_screenshot(page_count)
                 screenshots = result["screenshots"]
                 self._validate_screenshots(screenshots)
@@ -297,7 +306,6 @@ def _upload_single_screenshot(self, screenshot_path: str, upload_request: Upload
     def _upload_multiple_screenshots(self, screenshots: List[str], upload_request: UploadSnapshotRequest):
         """Upload multiple screenshots as chunks."""
         total_screenshots = len(screenshots)
-        upload_request.upload_chunk = "true"
 
         # Upload all but last screenshot
         for i in range(total_screenshots - 1):

utils/lambdatest_sdk_utils/app_api.py

@@ -10,7 +10,7 @@
 # API URLs
 SMARTUI_CLIENT_API_URL = os.getenv("SMARTUI_CLIENT_API_URL", "https://api.lambdatest.com/visualui/1.0")
 SMARTUI_UPLOAD_URL = os.getenv("SMARTUI_UPLOAD_URL", "https://api.lambdatest.com/")
-TEST_TYPE = "lambdatest-python-app-sdk"
+TEST_TYPE = "lambdatest-java-app-sdk"
 
 # Routes
 SMARTUI_AUTH_ROUTE = "/token/verify"
@@ -124,7 +124,7 @@ def upload_screenshot(screenshot_path: str, upload_request: UploadSnapshotReques
         Exception: If upload fails
     """
     try:
-        url = get_upload_host_url() + SMARTUI_UPLOAD_SCREENSHOT_ROUTE
+        url = get_host_url() + SMARTUI_UPLOAD_SCREENSHOT_ROUTE
 
         # Prepare multipart form data
         with open(screenshot_path, 'rb') as f:

utils/lambdatest_sdk_utils/models.py

@@ -61,6 +61,7 @@ class UploadSnapshotRequest:
     status_bar_height: Optional[str] = None
     crop_footer: Optional[str] = "false"
     crop_status_bar: Optional[str] = "false"
+    project_type: Optional[str] = None
 
     def to_dict(self):
         """Convert to dictionary for multipart form data."""
@@ -97,6 +98,8 @@ def to_dict(self):
             data["cropFooter"] = self.crop_footer
         if self.crop_status_bar:
             data["cropStatusBar"] = self.crop_status_bar
+        if self.project_type:
+            data["projectType"] = self.project_type
 
         return data