https://github.com/oreoxmt/tidbcloudy
Python SDK for TiDB Cloud
https://github.com/oreoxmt/tidbcloudy
sdk-python tidb tidbcloud
Last synced: about 1 month ago
JSON representation
Python SDK for TiDB Cloud
- Host: GitHub
- URL: https://github.com/oreoxmt/tidbcloudy
- Owner: Oreoxmt
- License: apache-2.0
- Created: 2022-08-04T13:01:37.000Z (almost 3 years ago)
- Default Branch: main
- Last Pushed: 2023-10-14T08:29:13.000Z (over 1 year ago)
- Last Synced: 2024-11-03T23:19:40.324Z (7 months ago)
- Topics: sdk-python, tidb, tidbcloud
- Language: Python
- Homepage:
- Size: 139 KB
- Stars: 4
- Watchers: 2
- Forks: 2
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# Python SDK for TiDB Cloud
[](https://github.com/psf/black)
`tidbcloudy` is an **unofficial** Python SDK for [TiDB Cloud](https://tidbcloud.com). If you encounter any problems or have suggestions, feel free to open an issue on the GitHub repository [Oreoxmt/tidbcloudy](https://github.com/Oreoxmt/tidbcloudy).
## Table of contents
- [Introduction](https://github.com/Oreoxmt/tidbcloudy#introduction)
- [Compatibility with TiDB Cloud API](https://github.com/Oreoxmt/tidbcloudy#compatibility-with-tidb-cloud-api)
- [Enhancements comparing to original TiDB Cloud API](https://github.com/Oreoxmt/tidbcloudy#enhancements-comparing-to-original-tidb-cloud-api)- [Installation](https://github.com/Oreoxmt/tidbcloudy#installation)
- [Usage](https://github.com/Oreoxmt/tidbcloudy#usage)- [Prerequisites](https://github.com/Oreoxmt/tidbcloudy#prerequisites)
- [List all resources in your organization](https://github.com/Oreoxmt/tidbcloudy#list-all-resources-in-your-organization)
- [Create a project and manage your AWS CMEK](https://github.com/Oreoxmt/tidbcloudy#create-a-project-and-manage-your-aws-cmek) **New in 1.0.9**
- [Create a cluster](https://github.com/Oreoxmt/tidbcloudy#create-a-cluster)
- [Connect to TiDB](https://github.com/Oreoxmt/tidbcloudy#connect-to-tidb)
- [Modify a cluster](https://github.com/Oreoxmt/tidbcloudy#modify-a-cluster)
- [Backup and restore](https://github.com/Oreoxmt/tidbcloudy#backup-and-restore)
- [Pause or resume your cluster](https://github.com/Oreoxmt/tidbcloudy#pause-or-resume-your-cluster) **New in 1.0.0**
- [Get monthly bills of your organization](https://github.com/Oreoxmt/tidbcloudy#get-monthly-bills-of-your-organization) **New in 1.0.8**
- [Delete all resources](https://github.com/Oreoxmt/tidbcloudy#delete-all-resources)- [Related projects](https://github.com/Oreoxmt/tidbcloudy#related-projects)
## Introduction
For more information about TiDB Cloud API, see TiDB Cloud API Documentation ([v1beta](https://docs.pingcap.com/tidbcloud/api/v1beta) and [v1beta1](https://docs.pingcap.com/tidbcloud/api/v1beta1)).
> TiDB Cloud is a fully-managed Database-as-a-Service (DBaaS) that brings everything great about TiDB to your cloud.
If you do not have a TiDB Cloud account yet, you can sign up [here](https://tidbcloud.com). For more details about TiDB Cloud, refer to [TiDB Cloud Documentation](https://docs.pingcap.com/tidbcloud).
You can use this SDK to access [TiDB Cloud](https://tidbcloud.com) and manage your billings, projects, clusters, backups and restores:
- manage your **billings** of your organization (_get_)
- manage your TiDB Cloud **projects** (_list_, _create_) and manage the **AWS CMEK** of your projects (_get_, _create_)
- list all available cloud providers (AWS and GCP), regions, and specifications before creating or modifying a cluster
- manage your TiDB Serverless or TiDB Dedicated **clusters** (_create_, _modify_, _pause_, _resume_, _get_, _list_, _delete_)
- manage your **backups** of a cluster (_create_, _get_, _list_, _delete_)
- manage your **restores** of a project (_create_, _get_, _list_)### Compatibility with TiDB Cloud API
`tidbcloudy` is compatible with TiDB Cloud API. **Endpoints added in [v1beta Release 20230228](https://docs.pingcap.com/tidbcloud/api/v1beta#section/API-Changelog/20230228) and [v1beta Release 20230905](https://docs.pingcap.com/tidbcloud/api/v1beta#section/API-Changelog/20230905) are not supported for now**. The following table lists the compatibility between `tidbcloudy` and TiDB Cloud API.
TiDB Cloud API v1beta1
TiDB Cloud API v1beta
2023-09-28
20230905
20230801
20230602
20230328
20230321
20230228
20230214
20230104
20221028
20220920
20220906
20220823
20220809
1.1.1
✅
❌
✅
✅
✅
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.1.0
✅
❌
✅
✅
✅
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.0.10
✅
❌
✅
✅
✅
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.0.9
✅
❌
✅
✅
✅
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.0.8
✅
❌
✅
✅
❌
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.0.7
❌
❌
✅
✅
❌
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.0.6
❌
❌
✅
✅
❌
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.0.5
❌
❌
✅
✅
❌
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.0.4
/
/
/
/
❌
✅
❌
✅
✅
✅
✅
✅
✅
✅
1.0.3
/
/
/
/
/
/
/
/
/
✅
✅
✅
✅
✅
1.0.2
/
/
/
/
/
/
/
/
/
✅
✅
✅
✅
✅
1.0.1
/
/
/
/
/
/
/
/
/
/
/
✅
✅
✅
1.0.0
/
/
/
/
/
/
/
/
/
/
/
/
✅
✅
0.2.2
/
/
/
/
/
/
/
/
/
/
/
/
/
✅
0.2.1
/
/
/
/
/
/
/
/
/
/
/
/
/
✅
0.2.0
/
/
/
/
/
/
/
/
/
/
/
/
/
✅
### Enhancements comparing to original [TiDB Cloud API](https://docs.pingcap.com/tidbcloud/api/v1beta)
- **Iterate over** resources instead of manual pagination
- **Connect to a TiDB cluster** using the [MySQL client](https://github.com/PyMySQL/mysqlclient)
- **Get** a Project using a Project **ID**
- Configure your cluster with **method chaining**
- Add your **current IP address automatically**
- **Wait for the cluster to be ready** when creating/modifying a cluster
- **Case-insensitive** when setting cluster type, cloud provider, and component name## Installation
```bash
pip3 install tidbcloudy
```⚠️ Make sure that you have installed **mysql client** in your environment. For more information, see [PyMySQL/mysqlclient](https://github.com/PyMySQL/mysqlclient#install).
## Usage
### Prerequisites
- Create a TiDB Cloud account.
- Create a TiDB Cloud API key. To manage your API keys, see [TiDB Cloud API Documentation](https://docs.pingcap.com/tidbcloud/api/v1beta#section/Authentication/API-Key-Management).
- Install the latest version of `tidbcloudy`.To get full code examples, see the [`examples`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples) folder.
> Note:
>
> It is recommended to set environment variables for your API public and private key. For example, in bash, you can:
>
> ```bash
> export PUBLIC_KEY=your_api_public_key
> export PRIVATE_KEY=your_api_private_key
> ```### List all resources in your organization
**This feature is available in `tidbcloudy` 1.0.9 or later.**
To get the full code example of listing all projects, clusters, backup tasks, and restore tasks in your organization, see [`0_list_resources.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/0_list_resources.py).
```python
import osimport tidbcloudy
from tidbcloudy.specification import ClusterTypepublic_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
for project in api.iter_projects():
print(project)
if project.aws_cmek_enabled:
for aws_cmek in project.iter_aws_cmek():
print(aws_cmek)
for cluster in project.iter_clusters():
print(cluster)
if cluster.cluster_type == ClusterType.DEDICATED:
for backup in cluster.iter_backups():
print(backup)
for restore in project.iter_restores():
print(restore)
```### Create a project and manage your AWS CMEK
To create a project, run the [`8_manage_project.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/8_manage_project.py).
```python
import osimport tidbcloudy
public_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")
debug_mode = os.environ.get("TIDBCLOUDY_LOG")api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
# Create a project with AWS CMEK enabled
project = api.create_project(name="0", aws_cmek_enabled=True, update_from_server=True)
print(project)# Configure AWS CMEK for the project
project.create_aws_cmek([("your_aws_region_1", "your_aws_kms_arn_1"), ("your_aws_region_2", "your_aws_kms_arn_2")])# List all AWS CMEKs of the project
for cmek in project.iter_aws_cmek():
print(cmek)
```### Create a cluster
Before creating a cluster, you should list all available provider regions and cluster configuration specifications. For more details, run the [`1_list_provider_regions.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/1_list_provider_regions.py).
```python
import osimport tidbcloudy
public_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
for spec in api.list_provider_regions():
print(f"- type: {spec.cluster_type.value}")
print(f" provider: {spec.cloud_provider.value}")
print(f" region: {spec.region}")
print(f" components:")
for tidb in spec.tidb:
print(f" - tidb: {tidb.node_size}; "
f"min={tidb.node_quantity_range.min} step={tidb.node_quantity_range.step}")
for tikv in spec.tikv:
print(f" - tikv: {tikv.node_size}; "
f"min={tikv.node_quantity_range.min} "
f"step={tikv.node_quantity_range.step}; "
f"{tikv.storage_size_gib_range.min}..{tikv.storage_size_gib_range.max} GiB")
for tiflash in spec.tiflash:
print(
f" - tiflash: {tiflash.node_size}; "
f"min={tiflash.node_quantity_range.min} step={tiflash.node_quantity_range.step}; "
f"{tiflash.storage_size_gib_range.min}..{tiflash.storage_size_gib_range.max} GiB")
```> Note:
>
> Creating a cluster might cost money. For more details, see [TiDB Cloud pricing details](https://www.pingcap.com/tidb-cloud-pricing-details).To create a TiDB Serverless cluster, run the [`2_1_create_serverless_cluster.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/2_1_create_serverless_cluster.py).
To create a TiDB Dedicated cluster, run the [`2_2_create_dedicated_cluster.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/2_2_create_dedicated_cluster.py).
The following takes creating a TiDB Serverless cluster as an example:
```python
import os
import tidbcloudy
from tidbcloudy.specification import CreateClusterConfigpublic_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")
debug_mode = os.environ.get("TIDBCLOUDY_LOG")
project_id = os.environ.get("PROJECT_ID", "1234567890123456789")api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
project = api.get_project(project_id, update_from_server=True)config = CreateClusterConfig()
config\
.set_name("serverless-0") \
.set_cluster_type("DEVELOPER") \
.set_cloud_provider("AWS") \
.set_region("us-west-2") \
.set_root_password("your_root_password")
cluster = project.create_cluster(config)
print(cluster)cluster.wait_for_available(interval_sec=1)
print(cluster)
```### Connect to TiDB
To connect to your TiDB cluster, run the [`3_connect_mysql.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/3_connect_mysql.py).
```python
import osimport tidbcloudy
from tidbcloudy.specification import ClusterStatuspublic_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")
project_id = os.environ.get("PROJECT_ID", "1234567890123456789")
cluster_id = os.environ.get("CLUSTER_ID", "1234567890123456789")print("Connecting to TiDB Cloud...")
api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
project = api.get_project(project_id, update_from_server=True)
cluster = project.get_cluster(cluster_id)
print(cluster)if cluster.status.cluster_status == ClusterStatus.AVAILABLE:
connection_strings = cluster.status.connection_strings
connection = cluster.connect(type="standard", database="test", password=os.environ.get("CLUSTER_PWD", "your_root_password"))
print(connection)
with connection:
with connection.cursor() as cursor:
cursor.execute("SELECT DATABASE();")
m = cursor.fetchone()
print(m[0])
```### Modify a cluster
> Note:
>
> Modify a cluster might cost money. For more details, see [TiDB Cloud pricing details](https://www.pingcap.com/tidb-cloud-pricing-details).To modify a cluster, run the [`4_scale_a_cluster.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/4_scale_a_cluster.py).
```python
import osimport tidbcloudy
from tidbcloudy.specification import UpdateClusterConfigpublic_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")
project_id = os.environ.get("PROJECT_ID", "1234567890123456789")
cluster_id = os.environ.get("CLUSTER_ID", "1234567890123456789")api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
project = api.get_project(project_id, update_from_server=True)
cluster = project.get_cluster(cluster_id)print("The original config is: {}".format(cluster.config.components.to_object()))
new_config = UpdateClusterConfig()
new_config.update_component("tiflash", node_quantity=1, node_size="8C64G", storage_size_gib=500)
cluster.update(new_config)
cluster.wait_for_available()print("The new config is: {}".format(cluster.config.components.to_object()))
```### Backup and restore
> Note:
>
> Backup or restore a cluster might cost money. For more details, see [TiDB Cloud pricing details](https://www.pingcap.com/tidb-cloud-pricing-details).To create a backup and restore, run the [`5_backup_restore.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/5_backup_restore.py)
```python
import osimport tidbcloudy
from tidbcloudy.specification import CreateClusterConfigpublic_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")
project_id = os.environ.get("PROJECT_ID", "1234567890123456789")
cluster_id = os.environ.get("CLUSTER_ID", "1234567890123456789")
backup_id = "1234567"api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
project = api.get_project(project_id, update_from_server=True)
cluster = project.get_cluster(cluster_id)
print("Create a manual backup task")
backup = cluster.create_backup(name="backup-1", description="automatically generated by tidbcloudy")
print(backup)config = CreateClusterConfig()
config \
.set_cluster_type("DEDICATED") \
.set_cloud_provider("AWS") \
.set_region("us-west-2") \
.set_port(4399) \
.set_root_password("your_root_password") \
.set_component("tidb", "8C16G", 1) \
.set_component("tikv", "8C32G", 3, 500) \
.set_component("tiflash", "8C64G", 1, 500) \
.add_current_ip_access()
print("Create a restore task from backup_id={}".format(backup_id))
restore = project.create_restore(backup_id=backup_id, name="restore-1", cluster_config=config)
restore_task = project.get_restore(restore.id)
print(restore_task.to_object())
for restore in project.iter_restores():
print(restore)
```### Pause or resume your cluster
**This feature is available in `tidbcloudy` 1.0.0 or later.**
To pause or resume your cluster, run the [`6_pause_cluster.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/6_pause_cluster.py).
```python
import osimport tidbcloudy
from tidbcloudy.specification import ClusterStatuspublic_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")
project_id = os.environ.get("PROJECT_ID", "1234567890123456789")
cluster_id = os.environ.get("CLUSTER_ID", "1234567890123456789")api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
project = api.get_project(project_id, update_from_server=True)
cluster = project.get_cluster(cluster_id)if cluster.status.cluster_status == ClusterStatus.AVAILABLE:
print("Pause the cluster id={}".format(cluster_id))
cluster.pause()
if cluster.status.cluster_status == ClusterStatus.PAUSED:
print("Resume the cluster id={}".format(cluster_id))
cluster.resume()
if cluster.status.cluster_status == ClusterStatus.RESUMING:
print("Wait for the RESUMING cluster id={} to be available".format(cluster_id))
cluster.wait_for_available()
```### Get monthly bills of your organization
**This feature is available in `tidbcloudy` 1.0.8 or later.**
To get the billing information of your organization, run the [`v1beta1_get_monthly_bill.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/v1beta1_get_monthly_bill.py).
```python
import osimport tidbcloudy
public_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")
debug_mode = os.environ.get("TIDBCLOUDY_LOG")api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
billing = api.get_monthly_bill(month="2023-10")
# billing = api.get_monthly_bill(month="202310")
# billing = api.get_current_month_bill()
print(billing)
print(billing.overview)
print(billing.summaryByProject)
print(billing.summaryByService)
```### Delete all resources
> Warning:
>
> This is a destructive operation. It will delete all resources in the project. **DO NOT** run this script in a production environment.To delete all clusters and backup tasks in your project, run the [`7_delete_resources.py`](https://github.com/Oreoxmt/tidbcloudy/tree/main/examples/7_delete_resources.py).
```python
import osimport tidbcloudy
from tidbcloudy.specification import ClusterTypepublic_key = os.environ.get("PUBLIC_KEY")
private_key = os.environ.get("PRIVATE_KEY")
project_id = os.environ.get("PROJECT_ID", "1234567890123456789")api = tidbcloudy.TiDBCloud(public_key=public_key, private_key=private_key)
project = api.get_project(project_id, update_from_server=True)
for cluster in project.iter_clusters():
print(cluster)
if cluster.cluster_type == ClusterType.DEDICATED:
for backup in cluster.iter_backups():
print(backup)
backup.delete()
cluster.delete()
```## Related projects
- Go SDK: [go-tidbcloud-sdk-v1](https://github.com/c4pt0r/go-tidbcloud-sdk-v1) by [@c4pt0r](https://github.com/c4pt0r)
- Official TiDB Cloud CLI: [tidbcloud-cli](https://github.com/tidbcloud/tidbcloud-cli) | [User documentation](https://docs.pingcap.com/tidbcloud/cli-reference)
- Official code samples in Go and Python: [tidbcloud-api-samples](https://github.com/tidbcloud/tidbcloud-api-samples)