rdxz2-utill 0.0.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

utill/cmd/utill.py

@@ -2,89 +2,231 @@ import click
 
 
 @click.group()
-def main(): pass
+def main():
+    pass
 
 
-# Conf
+# MARK: Conf
 
 
-@main.group('conf', help='Configure this library')
-def main__conf(): pass
-@main__conf.command('init', help='Initialize env files')
-@click.argument('mode', type=click.Choice(['google-cloud', 'postgresql', 'metabase']))
-def main__conf__init(**kwargs): from ._conf import _init; _init(**kwargs)
-@main__conf.command('list', help='List all configs')
-@click.option('-m', 'module', type=click.Choice(['postgresql', 'metabase']), help='List config for a specific modules')
-def main__conf__list(**kwargs): from ._conf import _list; _list(**kwargs)
-@main__conf.command('set', help='Set configuration variables')
-@click.option('-e', 'vars', type=(str, str), multiple=True, required=True, help='Variables -> K V')
-def main__conf__set(**kwargs): from ._conf import _set; _set(**kwargs)
+@main.group("conf", help="Configure this library")
+def main__conf():
+    pass
 
 
-# PG
+@main__conf.command("init", help="Initialize env files")
+@click.argument("mode", type=click.Choice(["google-cloud", "postgresql", "metabase"]))
+def main__conf__init(**kwargs):
+    from ._conf import _init
 
+    _init(**kwargs)
 
-@main.group('pg', help='PostgreSQL utility')
-def main__pg(): pass
-@main__pg.command('pg-to-pg', help='Copy table from one PG instance to another')
-@click.argument('src_profile', type=str)
-@click.argument('src_table', type=str)
-@click.argument('dst_profile', type=str)
-@click.argument('dst_table', type=str)
-@click.option('-c', '--columns', type=str, default='*', help='Columns to copy')
-def main__pg__pg_to_pg(**kwargs): from ._pg import _pg_to_pg; _pg_to_pg(**kwargs)
-@main__pg.command('upload-csv', help='Upload CSV file into PG table')
-@click.argument('profile', type=str)
-@click.argument('src_filename', type=click.Path())
-@click.argument('dst_table', type=str)
-def main__pg__upload_csv(**kwargs): from ._pg import _upload_csv; _upload_csv(**kwargs)
 
+@main__conf.command("list", help="List all configs")
+@click.option(
+    "-m",
+    "module",
+    type=click.Choice(["postgresql", "metabase"]),
+    help="List config for a specific modules",
+)
+def main__conf__list(**kwargs):
+    from ._conf import _list
 
-# BQ
+    _list(**kwargs)
 
 
-@main.group('bq', help='BigQuery utility')
-def main__bq(): pass
-@main__bq.command('upload-csv', help='Upload CSV file into BQ table')
-@click.argument('src_filename', type=click.Path())
-@click.argument('dst_table_fqn', type=str)
-@click.option('-c', 'columns', type=(str, str), required=True, multiple=True, help='Columns -> Name DataType')
-@click.option('--partition-col', 'partition_col', type=str, help='Partition column')
-@click.option('--cluster-col', 'cluster_cols', type=str, multiple=True, help='Cluster column(s)')
-@click.option('--project', type=str, help='Billing project')
-def main__bq__upload_csv(**kwargs): from ._bq import _upload_csv; _upload_csv(**kwargs)
-@main__bq.command('download-table', help='Download a BQ table into CSV file')
-@click.argument('src_table_fqn', type=str)
-@click.argument('dst_filename', type=str)
-@click.option('--project', type=str, help='Billing project')
-def main__bq__download_table(**kwargs): from ._bq import _download_table; _download_table(**kwargs)
+@main__conf.command("set", help="Set configuration variables")
+@click.option(
+    "-e", "vars", type=(str, str), multiple=True, required=True, help="Variables -> K V"
+)
+def main__conf__set(**kwargs):
+    from ._conf import _set
 
+    _set(**kwargs)
 
-# Encyrption
 
+# MARK: Metabase
 
-@main.group('enc', help='Encryption utility')
-def main__enc(): pass
-@main__enc.command('encrypt', help='Encrypt a string / file')
-@click.argument('src', type=str)
-@click.option('-p', 'password', type=str, required=True, help='The password')
-def main__enc__encrypt(**kwargs): from ._enc import _encrypt; _encrypt(**kwargs)
 
+@main.group("mb", help="Metabase utility commands")
+def main__mb():
+    pass
 
-# Other utilities
 
+@main__mb.command(
+    "jl-grant", help="Grant access to Metabase questions/collections/dashboards"
+)
+@click.option("-u", "--email", "emails", type=str, multiple=True, help="User emails")
+@click.option("-l", "--url", "urls", type=str, multiple=True, help="URLs")
+@click.option(
+    "-c",
+    "create_user_if_not_exists",
+    type=bool,
+    is_flag=True,
+    help="Create user if not exists, also reactivate user if it's already exists default: False",
+)
+def main__mb__grant(**kwargs):
+    from ._mb import _jl_grant
 
-@main.command('random', help='Generate random string')
-@click.option('-l', 'length', type=int, default=32, help='Length of the random string')
-@click.option('-a', 'alphanum', is_flag=True, default=False, help='Use alphanumeric only (a-Z, 0-9)')
-def main__random(**kwargs): from ._main import _random; _random(**kwargs)
+    _jl_grant(**kwargs)
 
 
-@main.command('unique', help='Get unique values')
-@click.argument('strings', nargs=-1)
-@click.option('-s', 'sort', type=bool, is_flag=True, help='Sort the output')
-def main__unique(**kwargs): from ._main import _unique; _unique(**kwargs)
+@main__mb.command(
+    "copy-permissions", help="Copy all permissions from one user to another"
+)
+@click.argument("src_email", type=str)
+@click.argument("dst_emails", type=str, nargs=-1)
+def main__mb__copy_permissions(**kwargs):
+    from ._mb import _copy_permissions
 
+    _copy_permissions(**kwargs)
 
-if __name__ == '__main__':
+
+@main__mb.command("reset-password", help="Reset Metabase user password")
+@click.option(
+    "-u",
+    "--email",
+    "emails",
+    type=str,
+    required=True,
+    multiple=True,
+    help="User emails",
+)
+def main__mb__reset_password(**kwargs):
+    from ._mb import _reset_password
+
+    _reset_password(**kwargs)
+
+
+@main__mb.command("disable-user", help="Disable Metabase user")
+@click.option(
+    "-u",
+    "--email",
+    "emails",
+    type=str,
+    required=True,
+    multiple=True,
+    help="User emails",
+)
+def main__mb__disable_user(**kwargs):
+    from ._mb import _disable_user
+
+    _disable_user(**kwargs)
+
+
+# MARK: PG
+
+
+@main.group("pg", help="PostgreSQL utility")
+def main__pg():
+    pass
+
+
+@main__pg.command("pg-to-pg", help="Copy table from one PG instance to another")
+@click.argument("src_profile", type=str)
+@click.argument("src_table", type=str)
+@click.argument("dst_profile", type=str)
+@click.argument("dst_table", type=str)
+@click.option("-c", "--columns", type=str, default="*", help="Columns to copy")
+def main__pg__pg_to_pg(**kwargs):
+    from ._pg import _pg_to_pg
+
+    _pg_to_pg(**kwargs)
+
+
+@main__pg.command("upload-csv", help="Upload CSV file into PG table")
+@click.argument("profile", type=str)
+@click.argument("src_filename", type=click.Path())
+@click.argument("dst_table", type=str)
+def main__pg__upload_csv(**kwargs):
+    from ._pg import _upload_csv
+
+    _upload_csv(**kwargs)
+
+
+# MARK: BQ
+
+
+@main.group("bq", help="BigQuery utility")
+def main__bq():
+    pass
+
+
+@main__bq.command("upload-csv", help="Upload CSV file into BQ table")
+@click.argument("src_filename", type=click.Path())
+@click.argument("dst_table_fqn", type=str)
+@click.option(
+    "-c",
+    "columns",
+    type=(str, str),
+    required=True,
+    multiple=True,
+    help="Columns -> Name DataType",
+)
+@click.option("--partition-col", "partition_col", type=str, help="Partition column")
+@click.option(
+    "--cluster-col", "cluster_cols", type=str, multiple=True, help="Cluster column(s)"
+)
+@click.option("--project", type=str, help="Billing project")
+def main__bq__upload_csv(**kwargs):
+    from ._bq import _upload_csv
+
+    _upload_csv(**kwargs)
+
+
+@main__bq.command("download-table", help="Download a BQ table into CSV file")
+@click.argument("src_table_fqn", type=str)
+@click.argument("dst_filename", type=str)
+@click.option("--project", type=str, help="Billing project")
+def main__bq__download_table(**kwargs):
+    from ._bq import _download_table
+
+    _download_table(**kwargs)
+
+
+# MARK: Encyrption
+
+
+@main.group("enc", help="Encryption utility")
+def main__enc():
+    pass
+
+
+@main__enc.command("encrypt", help="Encrypt a string / file")
+@click.argument("src", type=str)
+@click.option("-p", "password", type=str, required=True, help="The password")
+def main__enc__encrypt(**kwargs):
+    from ._enc import _encrypt
+
+    _encrypt(**kwargs)
+
+
+# MARK: Other utilities
+
+
+@main.command("random", help="Generate random string")
+@click.option("-l", "length", type=int, default=32, help="Length of the random string")
+@click.option(
+    "-a",
+    "alphanum",
+    is_flag=True,
+    default=False,
+    help="Use alphanumeric only (a-Z, 0-9)",
+)
+def main__random(**kwargs):
+    from ._main import _random
+
+    _random(**kwargs)
+
+
+@main.command("unique", help="Get unique values")
+@click.argument("strings", nargs=-1)
+@click.option("-s", "sort", type=bool, is_flag=True, help="Sort the output")
+def main__unique(**kwargs):
+    from ._main import _unique
+
+    _unique(**kwargs)
+
+
+if __name__ == "__main__":
     main()