quillsql 2.1.6__py3-none-any.whl → 2.2.1__py3-none-any.whl

quillsql/utils/run_query_processes.py

@@ -1,19 +1,20 @@
-import json
-
 def remove_fields(query_result, fields_to_remove):
-  fields = [
-      {"name": field['name'], "dataTypeID": field['dataTypeID']}
-      for field in query_result['fields']
-      if field['name'] not in fields_to_remove
-  ]
-  rows = [row for row in query_result['rows']]
-  for row in rows:
-    for field in fields_to_remove:
-      if field in row:
-          del row[field]
-  return {"fields": fields, "rows": rows}
+    fields = [
+        {"name": field["name"], "dataTypeID": field["dataTypeID"]}
+        for field in query_result["fields"]
+        if field["name"] not in fields_to_remove
+    ]
+    rows = [row for row in query_result["rows"]]
+    for row in rows:
+        for field in fields_to_remove:
+            if field in row:
+                del row[field]
+    return {"fields": fields, "rows": rows}
+
 
 def array_to_map(queries, array_to_map, metadata, target_pool):
-  for i in range(len(queries)):
-    query_result = target_pool.query(queries[i])
-    metadata[array_to_map.get("arrayName")][i][array_to_map.get("field")] = query_result.get("rows")
+    mapped_array = []
+    for i in range(len(queries)):
+        query_result = target_pool.query(queries[i])
+        mapped_array.append(query_result.get("rows"))
+    return mapped_array

quillsql/utils/schema_conversion.py

@@ -1,6 +1,9 @@
 from quillsql.assets.pgtypes import PG_TYPES
 
+
 def convert_type_to_postgres(data_type_id):
-   # find the object in PG_TYPES that matches the type
-    pg_type = next((pg_type for pg_type in PG_TYPES if pg_type['oid'] == data_type_id), None)
-    return pg_type['typname'] if pg_type else data_type_id
+    # find the object in PG_TYPES that matches the type
+    pg_type = next(
+        (pg_type for pg_type in PG_TYPES if pg_type["oid"] == data_type_id), None
+    )
+    return pg_type["typname"] if pg_type else data_type_id

quillsql/utils/tenants.py

@@ -0,0 +1,60 @@
+from typing import Union, List, Dict
+
+# Type aliases for clarity
+TenantId = Union[str, int]
+TenantInfo = Dict[str, Union[str, List[TenantId]]]
+Tenants = Union[List[TenantId], List[TenantInfo]]
+
+def extract_tenant_ids(tenants: Tenants) -> List[TenantId]:
+    """
+    Extract tenant IDs from the tenants parameter, which can be either a list of IDs
+    or a list of tenant info dictionaries.
+    
+    Args:
+        tenants: Either a list of tenant IDs (strings/integers) or a list of tenant info dictionaries
+        
+    Returns:
+        List of tenant IDs
+        
+    Raises:
+        ValueError: If the tenants parameter format is invalid
+    """
+    if not tenants:
+        raise ValueError("Invalid format for tenants: empty list")
+
+    first_tenant = tenants[0]
+    
+    if isinstance(first_tenant, (str, int)):
+        return tenants  # type: ignore
+    elif isinstance(first_tenant, dict) and "tenantIds" in first_tenant:
+        # TODO: support multiple tenants in future
+        return first_tenant["tenantIds"]
+    else:
+        raise ValueError("Invalid format for tenants")
+
+def extract_tenant_field(tenants: Tenants, dashboard_owner: str) -> str:
+    """
+    Extract tenant field from the tenants parameter, falling back to dashboard_owner
+    if tenants is a simple list of IDs.
+    
+    Args:
+        tenants: Either a list of tenant IDs (strings/integers) or a list of tenant info dictionaries
+        dashboard_owner: The default tenant field to use if tenants is a simple list
+        
+    Returns:
+        The tenant field string
+        
+    Raises:
+        ValueError: If the tenants parameter format is invalid
+    """
+    if not tenants:
+        raise ValueError("Invalid format for tenants: empty list")
+
+    first_tenant = tenants[0]
+    
+    if isinstance(first_tenant, (str, int)):
+        return dashboard_owner
+    elif isinstance(first_tenant, dict) and "tenantField" in first_tenant:
+        return first_tenant["tenantField"]
+    else:
+        raise ValueError("Invalid format for tenants") 

quillsql-2.2.1.dist-info/METADATA

@@ -0,0 +1,69 @@
+Metadata-Version: 2.4
+Name: quillsql
+Version: 2.2.1
+Summary: Quill SDK for Python.
+Home-page: https://github.com/quill-sql/quill-python
+Author: Quill
+Author-email: shawn@quill.co
+Description-Content-Type: text/markdown
+Requires-Dist: psycopg2-binary
+Requires-Dist: requests
+Requires-Dist: redis
+Requires-Dist: python-dotenv
+Requires-Dist: pytest
+Requires-Dist: google-cloud-bigquery
+Requires-Dist: google-auth
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: summary
+
+# Quill Python SDK
+
+## Quickstart
+
+First, install the quillsql package by running:
+
+```bash
+$ pip install quillsql
+```
+
+Then, add a `/quill` endpoint to your existing python server. For example, if
+you were running a FASTAPI app, you would just add the endpoint like this:
+
+```python
+from quillsql import Quill
+
+quill = Quill(
+    private_key=os.getenv("QULL_PRIVATE_KEY"),
+    database_connection_string=os.getenv("POSTGRES_READ"),
+    database_type="postgresql"
+)
+
+security = HTTPBearer()
+
+async def authenticate_jwt(token: str = Depends(security)):
+    # Your JWT validation logic here
+    # Return user object or raise HTTPException
+    user = validate_jwt_token(token.credentials)
+    return user
+
+@app.post("/quill")
+async def quill_post(data: Request, user: dict = Depends(authenticate_jwt)):
+    # assuming user fetched via auth middleware has an userId
+    user_id = user["user_id"]
+    body = await data.json()
+    metadata = body.get("metadata")
+
+    result = quill.query(
+        tenants=[{"tenantField": "user_id", "tenantIds": [user_id]}],
+        metadata=metadata
+    )
+    return result
+```
+
+Then you can run your app like normally. Pass in this route to our react library
+on the frontend and you all set!

quillsql-2.2.1.dist-info/RECORD

@@ -0,0 +1,20 @@
+quillsql/__init__.py,sha256=FiuoxaNZveKXOPB0hkpfGNlpZKmSn3pRwcqm9HKYbCQ,180
+quillsql/core.py,sha256=cWWu_ME3kapzv1fBhoJLmkA7QIlbVCAOMHyce_VE3BU,22464
+quillsql/error.py,sha256=n9VKHw4FAgg7ZEAz2YQ8L_8FdRG_1shwGngf2iWhUSM,175
+quillsql/assets/__init__.py,sha256=oXQ2ZS5XDXkXTYjADxNfGt55cIn_rqfgWL2EDqjTyoI,45
+quillsql/assets/pgtypes.py,sha256=-B_2wUaoAsdX7_HnJhUlx4ptZQ6x-cXwuST9ACgGFdE,33820
+quillsql/db/__init__.py,sha256=sPgYMF3eqKuc2k96i1Re8NHazLdGKCVN-rs15F1sovw,63
+quillsql/db/bigquery.py,sha256=xJv3WUjyDmnfwaAAeLTTezUEw_qxrxQy-CTCwj8Hmjs,4811
+quillsql/db/cached_connection.py,sha256=Pp0R2XGY-QuQYh8ThiDxSwNb6SEuEKfZcQRcvzjeC28,2556
+quillsql/db/db_helper.py,sha256=qiIP-BM7R-3PhvWBELYjNazi-92EcQB0q9eN7Ej7XUA,2111
+quillsql/db/postgres.py,sha256=ZTLtUVTJHkqAj7nZkcwbmkSOwR2ySN7vS5snoxqLRN0,4156
+quillsql/utils/__init__.py,sha256=C2k9Xe0sG5XrP0XJo9K_-iej1S9PLMRKOYMeLxj7NYE,210
+quillsql/utils/filters.py,sha256=REXOLIQZDHL4EDKtConXY9_GaqUmc2uTcyUa2_4MvAg,6786
+quillsql/utils/pivot_template.py,sha256=NzHO7Ux8GVAKY-DUtsOmE7TUls2q6FJG2kgJxVWq-wQ,18282
+quillsql/utils/run_query_processes.py,sha256=FRmNvjTDLUBr7MqDKQmivdC0anwybMXUyzQbKnaZx70,698
+quillsql/utils/schema_conversion.py,sha256=TFfMibN9nOsxNRhHw5YIFl3jGTvipG81bxX4LFDulUY,314
+quillsql/utils/tenants.py,sha256=ZD2FuKz0gjBVSsThHDv1P8PU6EL8E009NWihE5hAH-Q,2022
+quillsql-2.2.1.dist-info/METADATA,sha256=AImLKL4nUys0Yz6i_12iPMHLxi4q5KHnxZJWRoIyw7U,1786
+quillsql-2.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+quillsql-2.2.1.dist-info/top_level.txt,sha256=eU2vHnVqwpYQJ3ADl1Q-DIBzbYejZRUhcMdN_4zMCz8,9
+quillsql-2.2.1.dist-info/RECORD,,

{quillsql-2.1.6.dist-info → quillsql-2.2.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.42.0)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

quillsql-2.1.6.dist-info/METADATA

@@ -1,72 +0,0 @@
-Metadata-Version: 2.1
-Name: quillsql
-Version: 2.1.6
-Summary: Quill SDK for Python.
-Home-page: https://github.com/quill-sql/quill-python
-Author: Quill
-Author-email: shawn@quillsql.com
-Description-Content-Type: text/markdown
-Requires-Dist: psycopg2-binary
-Requires-Dist: requests
-Requires-Dist: redis
-Requires-Dist: python-dotenv
-Requires-Dist: pytest
-Requires-Dist: google-cloud-bigquery
-Requires-Dist: google-auth
-
-# Quill Python SDK
-
-## Quickstart
-
-First, install the quillsql package by running:
-
-```bash
-$ pip install quillsql
-```
-
-Then, add a `/quill` endpoint to your existing python server. For example, if
-you were running a FASTAPI app, you would just add the endpoint like this:
-
-```python
-from quillsql import Quill
-from fastapi import FastAPI, Request
-
-app = FastAPI()
-
-quill = Quill(
-    private_key=<YOUR_PRIVATE_KEY_HERE>,
-    database_connection_string=<YOUR_DB_CONNECTION_STRING_HERE>,
-)
-
-# ... your existing endpoints here ...
-
-@app.post("/quill")
-async def quill_post(data: Request):
-    body = await data.json()
-    return quill.query(org_id="2", data=body)
-```
-
-Then you can run your app like normally. Pass in this route to our react library
-on the frontend and you all set!
-
-## For local testing (dev purposes only)
-
-```
-pipenv install
-pipenv shell
-uvicorn examples.fastapi-server.app:app --reload --port 3000
-```
-
-You are now ready to ping your local server at http://localhost:3000.
-
-## Troubleshooting
-
-If you run into issues with `Library not loaded: @rpath/libpq.5.dylib` or `no LC_RPATH's found`, try uninstalling and reinstalling postgres on your machine. For example, using homebrew:
-
-```bash
-$ brew uninstall c
-$ brew update
-$ brew install postgresql
-```
-
-If you're still having this issue, this resource might also be useful for you: https://www.psycopg.org/docs/install.html.

quillsql-2.1.6.dist-info/RECORD

@@ -1,17 +0,0 @@
-quillsql/__init__.py,sha256=xxVDrfIXLIbt45yHKGYRbqmgXSPe03uNbqENMLUnUC8,70
-quillsql/core.py,sha256=cCoCLwudlfmOj-o2hW-7Bwx5-XrCvXxQg3iNpTDX6z0,7670
-quillsql/error.py,sha256=XltjRgdBcq3lQN0KdUba5eDzA21yQ7KisIjvqqUSELI,161
-quillsql/assets/__init__.py,sha256=PX43SCko4w5P197YXghGI2VAq3-X1J91IO_tWY7rpiM,44
-quillsql/assets/pgtypes.py,sha256=qWAcGO6QfWN6XXtGI7cMreNSeTjw-xNgcc1aLoq9EV0,40768
-quillsql/db/__init__.py,sha256=mBGMd22nAtp9CdTpEG1Ty47zvSpO8gGejORCACF3-3c,62
-quillsql/db/bigquery.py,sha256=Hzs2DjJ9p8l-HywvaaAk5PbadueNgTVDyDMpCGTcd48,4220
-quillsql/db/cached_connection.py,sha256=1y_GzW5mIAYpmGKJjxpBqmFafpmgUA0Qm9oW_ufyPko,2570
-quillsql/db/db_helper.py,sha256=PFAcX2V26sIaR-hF8orGGltKXdEzR5b_enMSOHlAMUg,2054
-quillsql/db/postgres.py,sha256=iw71feFvnCDBMKpeGbhiJ_4Z4h7AlUi4Hk12Pfyzx6c,2799
-quillsql/utils/__init__.py,sha256=jJLp0OkFQdWohu8vIijfggiWf4eWmeAafVzx3YpdE4E,75
-quillsql/utils/run_query_processes.py,sha256=hElCVACq6bz1_jVIjQUrZzI1CPY188I3_5sHc21jtOc,673
-quillsql/utils/schema_conversion.py,sha256=MwFk95SIZ8OX5EAv141QXLjmCef-1z8jiVKlW71-sk8,297
-quillsql-2.1.6.dist-info/METADATA,sha256=HNtsQyw_xKnQasZ5OKDvKNZf_1OS_qCbPQguu-YTk_U,1809
-quillsql-2.1.6.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-quillsql-2.1.6.dist-info/top_level.txt,sha256=eU2vHnVqwpYQJ3ADl1Q-DIBzbYejZRUhcMdN_4zMCz8,9
-quillsql-2.1.6.dist-info/RECORD,,