import datetime
import sqlalchemy

import sqlalchemy_bigquery
from google.cloud.bigquery import TimePartitioning

!bq rm -f tmp.test_table_create
engine = sqlalchemy.create_engine(f"bigquery://github.com/tmp", echo=True)
meta = sqlalchemy.MetaData()
table = sqlalchemy.Table(
    f"tmp.test_table_create",
    meta,
    sqlalchemy.Column("integer_c", sqlalchemy.Integer, doc="column description"),
    sqlalchemy.Column("float_c", sqlalchemy.Float),
    sqlalchemy.Column("decimal_c", sqlalchemy.DECIMAL),
    sqlalchemy.Column("string_c", sqlalchemy.String),
    sqlalchemy.Column("text_c", sqlalchemy.Text),
    sqlalchemy.Column("boolean_c", sqlalchemy.Boolean),
    sqlalchemy.Column("timestamp_c", sqlalchemy.TIMESTAMP),
    sqlalchemy.Column("datetime_c", sqlalchemy.DATETIME),
    sqlalchemy.Column("date_c", sqlalchemy.DATE),
    sqlalchemy.Column("time_c", sqlalchemy.TIME),
    sqlalchemy.Column("binary_c", sqlalchemy.BINARY),
    bigquery_description="test table description",
    bigquery_friendly_name="test table name",
    bigquery_expiration_timestamp=datetime.datetime(2183, 3, 26, 8, 30, 0),
    bigquery_time_partitioning=TimePartitioning(
        field="date_c",
        expiration_ms=1000 * 60 * 60 * 24 * 30,  # 30 days
        type_="DAY"
    ),
    bigquery_require_partition_filter=True,
    bigquery_default_rounding_mode="ROUND_HALF_EVEN",
    bigquery_clustering_fields=["integer_c", "decimal_c"],
)
table.create(engine)