black formatting

Author: casey-crawford-cfa

samples/extracts.py

@@ -42,7 +42,6 @@ def main():
     server.add_http_options({"verify": False})
     server.use_server_version()
     with server.auth.sign_in(tableau_auth):
-
         wb = None
         ds = None
         if args.workbook:

tableauserverclient/models/extract_item.py

@@ -6,7 +6,7 @@
 class ExtractItem:
     """
     An extract refresh task item.
-    
+
     Attributes
     ----------
     id : str
@@ -22,11 +22,7 @@ class ExtractItem:
     """
 
     def __init__(
-        self,
-        priority: int,
-        refresh_type: str,
-        workbook_id: Optional[str] = None,
-        datasource_id: Optional[str] = None
+        self, priority: int, refresh_type: str, workbook_id: Optional[str] = None, datasource_id: Optional[str] = None
     ):
         self._id: Optional[str] = None
         self._priority = priority
@@ -65,22 +61,22 @@ def from_xml_element(cls, parsed_response: ET.Element, ns: dict) -> List["Extrac
         """Create ExtractItem objects from XML element."""
         all_extract_items = []
         all_extract_xml = parsed_response.findall(".//t:extract", namespaces=ns)
-        
+
         for extract_xml in all_extract_xml:
             extract_id = extract_xml.get("id", None)
             priority = int(extract_xml.get("priority", 0))
             refresh_type = extract_xml.get("type", None)
-            
+
             # Check for workbook or datasource
             workbook_elem = extract_xml.find(".//t:workbook", namespaces=ns)
             datasource_elem = extract_xml.find(".//t:datasource", namespaces=ns)
-            
+
             workbook_id = workbook_elem.get("id", None) if workbook_elem is not None else None
             datasource_id = datasource_elem.get("id", None) if datasource_elem is not None else None
-            
+
             extract_item = cls(priority, refresh_type, workbook_id, datasource_id)
             extract_item._id = extract_id
-            
+
             all_extract_items.append(extract_item)
-            
-        return all_extract_items
+
+        return all_extract_items

tableauserverclient/server/endpoint/groups_endpoint.py

@@ -173,10 +173,12 @@ def delete(self, group_id: str) -> None:
         logger.info(f"Deleted single group (ID: {group_id})")
 
     @overload
-    def update(self, group_item: GroupItem, as_job: Literal[False]) -> GroupItem: ...
+    def update(self, group_item: GroupItem, as_job: Literal[False]) -> GroupItem:
+        ...
 
     @overload
-    def update(self, group_item: GroupItem, as_job: Literal[True]) -> JobItem: ...
+    def update(self, group_item: GroupItem, as_job: Literal[True]) -> JobItem:
+        ...
 
     @api(version="2.0")
     def update(self, group_item, as_job=False):
@@ -258,10 +260,12 @@ def create(self, group_item: GroupItem) -> GroupItem:
         return GroupItem.from_response(server_response.content, self.parent_srv.namespace)[0]
 
     @overload
-    def create_AD_group(self, group_item: GroupItem, asJob: Literal[False]) -> GroupItem: ...
+    def create_AD_group(self, group_item: GroupItem, asJob: Literal[False]) -> GroupItem:
+        ...
 
     @overload
-    def create_AD_group(self, group_item: GroupItem, asJob: Literal[True]) -> JobItem: ...
+    def create_AD_group(self, group_item: GroupItem, asJob: Literal[True]) -> JobItem:
+        ...
 
     @api(version="2.0")
     def create_AD_group(self, group_item, asJob=False):

tableauserverclient/server/endpoint/schedules_endpoint.py

@@ -151,17 +151,19 @@ def _add_to(
             return OK
 
     @api(version="2.3")
-    def get_extract_refresh_tasks(self, schedule_id: str, req_options: Optional["RequestOptions"] = None) -> tuple[list["ExtractItem"], "PaginationItem"]:
+    def get_extract_refresh_tasks(
+        self, schedule_id: str, req_options: Optional["RequestOptions"] = None
+    ) -> tuple[list["ExtractItem"], "PaginationItem"]:
         """Get all extract refresh tasks for the specified schedule."""
         if not schedule_id:
             error = "Schedule ID undefined"
             raise ValueError(error)
-        
+
         logger.info(f"Querying extract refresh tasks for schedule (ID: {schedule_id})")
         url = f"{self.siteurl}/{schedule_id}/extracts"
         server_response = self.get_request(url, req_options)
-        
+
         pagination_item = PaginationItem.from_response(server_response.content, self.parent_srv.namespace)
         extract_items = ExtractItem.from_response(server_response.content, self.parent_srv.namespace)
-        
+
         return extract_items, pagination_item

tableauserverclient/server/pager.py

@@ -12,12 +12,14 @@
 
 @runtime_checkable
 class Endpoint(Protocol[T]):
-    def get(self, req_options: Optional[RequestOptions]) -> tuple[list[T], PaginationItem]: ...
+    def get(self, req_options: Optional[RequestOptions]) -> tuple[list[T], PaginationItem]:
+        ...
 
 
 @runtime_checkable
 class CallableEndpoint(Protocol[T]):
-    def __call__(self, __req_options: Optional[RequestOptions], **kwargs) -> tuple[list[T], PaginationItem]: ...
+    def __call__(self, __req_options: Optional[RequestOptions], **kwargs) -> tuple[list[T], PaginationItem]:
+        ...
 
 
 class Pager(Iterable[T]):

tableauserverclient/server/query.py

@@ -97,10 +97,12 @@ def __iter__(self: Self) -> Iterator[T]:
                 return
 
     @overload
-    def __getitem__(self: Self, k: Slice) -> list[T]: ...
+    def __getitem__(self: Self, k: Slice) -> list[T]:
+        ...
 
     @overload
-    def __getitem__(self: Self, k: int) -> T: ...
+    def __getitem__(self: Self, k: int) -> T:
+        ...
 
     def __getitem__(self, k):
         page = self.page_number

tableauserverclient/server/request_factory.py

@@ -1008,9 +1008,9 @@ def update_req(self, workbook_item, parent_srv: Optional["Server"] = None):
             if data_freshness_policy_config.option == "FreshEvery":
                 if data_freshness_policy_config.fresh_every_schedule is not None:
                     fresh_every_element = ET.SubElement(data_freshness_policy_element, "freshEverySchedule")
-                    fresh_every_element.attrib["frequency"] = (
-                        data_freshness_policy_config.fresh_every_schedule.frequency
-                    )
+                    fresh_every_element.attrib[
+                        "frequency"
+                    ] = data_freshness_policy_config.fresh_every_schedule.frequency
                     fresh_every_element.attrib["value"] = str(data_freshness_policy_config.fresh_every_schedule.value)
                 else:
                     raise ValueError(f"data_freshness_policy_config.fresh_every_schedule must be populated.")

test/request_factory/test_task_requests.py

@@ -5,7 +5,6 @@
 
 
 class TestTaskRequest(unittest.TestCase):
-
     def setUp(self):
         self.task_request = TaskRequest()
         self.xml_request = ET.Element("tsRequest")

test/test_schedule.py

@@ -409,7 +409,6 @@ def test_add_flow(self) -> None:
 
     def test_get_extract_refresh_tasks(self) -> None:
         self.server.version = "2.3"
-        
 
         with open(GET_EXTRACT_TASKS_XML, "rb") as f:
             response_xml = f.read().decode("utf-8")
@@ -420,7 +419,7 @@ def test_get_extract_refresh_tasks(self) -> None:
             # Fix the URL construction to match the endpoint pattern
             # url = f"{self.baseurl}/{schedule_id}/extracts"
             m.get(baseurl, text=response_xml)
-            
+
             extracts = self.server.schedules.get_extract_refresh_tasks(schedule_id)
 
             self.assertIsNotNone(extracts)