tinybird 0.0.1.dev18__tar.gz → 0.0.1.dev20__tar.gz

{tinybird-0.0.1.dev18 → tinybird-0.0.1.dev20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tinybird
-Version: 0.0.1.dev18
+Version: 0.0.1.dev20
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird

{tinybird-0.0.1.dev18 → tinybird-0.0.1.dev20}/tinybird/client.py

@@ -103,7 +103,7 @@ class TinyB:
         self.send_telemetry = send_telemetry
         self.semver = semver
 
-    async def _req(
+    async def _req_raw(
         self,
         endpoint: str,
         data=None,
@@ -155,10 +155,15 @@ class TinyB:
                     response = await sync_to_async(session.get, thread_sensitive=False)(
                         url, verify=verify_ssl, **kwargs
                     )
-
         except Exception as e:
             raise e
 
+        if self.send_telemetry:
+            try:
+                add_telemetry_event("api_request", endpoint=url, token=self.token, status_code=response.status_code)
+            except Exception as ex:
+                logging.exception(f"Can't send telemetry: {ex}")
+
         logging.debug("== server response ==")
         logging.debug(response.content)
         logging.debug("==      end        ==")
@@ -169,6 +174,21 @@ class TinyB:
             except Exception as ex:
                 logging.exception(f"Can't send telemetry: {ex}")
 
+        return response
+
+    async def _req(
+        self,
+        endpoint: str,
+        data=None,
+        files=None,
+        method: str = "GET",
+        retries: int = LIMIT_RETRIES,
+        use_token: Optional[str] = None,
+        **kwargs,
+    ):
+        token_to_use = use_token if use_token else self.token
+        response = await self._req_raw(endpoint, data, files, method, retries, use_token, **kwargs)
+
         if response.status_code == 403:
             error = parse_error_response(response)
             if not token_to_use:
@@ -500,7 +520,7 @@ class TinyB:
         payload = {"datasource_a": datasource_a, "datasource_b": datasource_b}
         return await self._req("/v0/datasources/exchange", method="POST", data=payload)
 
-    async def datasource_events(self, datasource_name: str, data: Dict[str, Any]):
+    async def datasource_events(self, datasource_name: str, data: str):
         return await self._req(f"/v0/events?name={datasource_name}", method="POST", data=data)
 
     async def analyze_pipe_node(
@@ -816,7 +836,7 @@ class TinyB:
         params = {"with_token": "true" if with_token else "false"}
         return await self._req(f"/v0/workspaces/{workspace_id}?{urlencode(params)}")
 
-    async def workspace_info(self):
+    async def workspace_info(self) -> Dict[str, Any]:
         return await self._req("/v0/workspace")
 
     async def wait_for_job(

{tinybird-0.0.1.dev18 → tinybird-0.0.1.dev20}/tinybird/config.py

@@ -20,7 +20,7 @@ CURRENT_VERSION = f"{__cli__.__version__}"
 VERSION = f"{__cli__.__version__} (rev {__revision__})"
 DEFAULT_UI_HOST = "https://app.tinybird.co"
 SUPPORTED_CONNECTORS = ["bigquery", "snowflake"]
-PROJECT_PATHS = ["datasources", "pipes", "tests"]
+PROJECT_PATHS = ["datasources", "datasources/fixtures", "endpoints", "pipes", "tests", "scripts", "deploy"]
 DEPRECATED_PROJECT_PATHS = ["endpoints"]
 MIN_WORKSPACE_ID_LENGTH = 36
 LEGACY_HOSTS = {