tinybird 0.0.1.dev17__py3-none-any.whl → 0.0.1.dev19__py3-none-any.whl

tinybird/tb/modules/common.py

@@ -17,7 +17,7 @@ from contextlib import closing
 from copy import deepcopy
 from enum import Enum
 from functools import wraps
-from os import chmod, environ, getcwd, getenv
+from os import environ, getcwd, getenv
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Callable, Dict, Iterable, List, Optional, Set, Tuple, Union
 from urllib.parse import urlparse
@@ -61,9 +61,7 @@ if TYPE_CHECKING:
     from tinybird.connectors import Connector
 
 from tinybird.feedback_manager import FeedbackManager, warning_message
-from tinybird.git_settings import DEFAULT_TINYENV_FILE
 from tinybird.syncasync import async_to_sync, sync_to_async
-from tinybird.tb.modules.cicd import APPEND_FIXTURES_SH, DEFAULT_REQUIREMENTS_FILE, EXEC_TEST_SH
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.exceptions import (
     CLIAuthException,
@@ -371,111 +369,6 @@ async def folder_init(
             for path in Path(folder).glob(f"*.{format}"):
                 await _generate_datafile(str(path), client, format=format, force=force)
 
-    if generate_releases:
-        base = Path(".")
-        f = base / (".tinyenv")
-        if not f.exists() or force:
-            async with aiofiles.open(".tinyenv", "w") as file:
-                await file.write(DEFAULT_TINYENV_FILE)
-                click.echo(FeedbackManager.info_file_created(file=".tinyenv"))
-        else:
-            click.echo(FeedbackManager.info_dottinyenv_already_exists())
-
-        base = Path(".")
-        f = base / ("requirements.txt")
-        if not f.exists() or force:
-            async with aiofiles.open("requirements.txt", "w") as file:
-                await file.write(DEFAULT_REQUIREMENTS_FILE)
-                click.echo(FeedbackManager.info_file_created(file="requirements.txt"))
-
-        base = Path("scripts")
-        if not base.exists():
-            base = Path()
-        f = base / ("exec_test.sh")
-        if not f.exists() or force:
-            async with aiofiles.open(f"{f}", "w") as t_file:
-                await t_file.write(EXEC_TEST_SH)
-            click.echo(FeedbackManager.info_file_created(file="scripts/exec_test.sh"))
-        chmod(f, 0o755)
-
-        f = base / ("append_fixtures.sh")
-        if not f.exists() or force:
-            async with aiofiles.open(f"{f}", "w") as t_file:
-                await t_file.write(APPEND_FIXTURES_SH)
-            click.echo(FeedbackManager.info_file_created(file="scripts/append_fixtures.sh"))
-        chmod(f, 0o755)
-
-        base = Path("tests")
-        if not base.exists():
-            base = Path()
-        f = base / ("example.yml")
-        if not base.exists() or force:
-            async with aiofiles.open(f"{f}", "w") as t_file:
-                await t_file.write(
-                    """
-##############################################################################################################################
-###   Visit https://www.tinybird.co/docs/production/implementing-test-strategies.html#data-quality-tests              ###
-###   for more details on Data Quality tests                                                                               ###
-##############################################################################################################################
-
-- example_no_negative_numbers:
-    max_bytes_read: null
-    max_time: null
-    sql: |
-        SELECT
-            number
-        FROM numbers(10)
-        WHERE
-            number < 0
-
-#    - example_top_products_params_no_empty_top_10_on_2023:
-#       max_bytes_read: null
-#       max_time: null
-#       sql: |
-#          SELECT *
-#          FROM top_products_params
-#          WHERE empty(top_10)
-#       pipe:
-#          name: top_products_params
-#          params:
-#             start: '2023-01-01'
-#             end: '2023-12-31'
-
-"""
-                )
-
-        f = base / ("regression.yaml")
-        if not base.exists() or force:
-            async with aiofiles.open(f"{f}", "w") as t_file:
-                await t_file.write(
-                    """
-############################################################################################################################
-###   Visit https://www.tinybird.co/docs/production/implementing-test-strategies.html#regression-tests              ###
-###   for more details on Regression tests                                                                               ###
-############################################################################################################################
-
-###
-### New pipes are covered by this rule, rules below this one supersede this setting
-###
-- pipe: '.*'
-  tests:
-    - coverage:
-
-
-
-###
-### These are rules to customize regression testing by pipe using regular expressions
-### For instance skip regression tests for the pipes matching `endpoint_name.*`
-###
-- pipe: 'endpoint_name.*'
-  tests:
-    - coverage:
-        config:
-          skip: True
-
-"""
-                )
-
 
 async def configure_connector(connector):
     if connector not in SUPPORTED_CONNECTORS:

tinybird/tb/modules/create.py

@@ -1,5 +1,4 @@
 import os
-import subprocess
 from os import getcwd
 from pathlib import Path
 from typing import Optional
@@ -126,7 +125,7 @@ async def create(
             test_name = "api_token_usage"
             test_path = Path(folder) / "tests" / f"{test_name}.yaml"
             test_content = fetch_gist_content(
-                "https://gist.githubusercontent.com/gnzjgo/e58620bbb977d6f42f1d0c2a7b46ac8f/raw/a5f61d5019111f937484f941111829dfce69f648/api_token_usage.yaml"
+                "https://gist.githubusercontent.com/gnzjgo/e58620bbb977d6f42f1d0c2a7b46ac8f/raw/a3a1cd0ce3a90bcd2f6dfce00da51e6051443612/api_token_usage.yaml"
             )
             test_path.write_text(test_content)
             click.echo(FeedbackManager.info(message=f"✓ /tests/{test_name}.yaml"))
@@ -147,7 +146,7 @@ async def create(
                 datasource_content = datasource_path.read_text()
                 has_json_path = "`json:" in datasource_content
                 if has_json_path:
-                    sql = await llm.generate_sql_sample_data(schema=datasource_content, rows=rows, context=prompt)
+                    sql = await llm.generate_sql_sample_data(schema=datasource_content, rows=rows, prompt=prompt)
                     result = await tb_client.query(f"{sql} FORMAT JSON")
                     data = result.get("data", [])
                     fixture_name = build_fixture_name(datasource_path.absolute(), datasource_name, datasource_content)
@@ -214,9 +213,6 @@ def init_git(folder: str):
     try:
         path = Path(folder)
         gitignore_file = path / ".gitignore"
-        git_folder = path / ".git"
-        if not git_folder.exists():
-            subprocess.run(["git", "init"], cwd=path, check=True, capture_output=True)
 
         if gitignore_file.exists():
             content = gitignore_file.read_text()