|
14 | 14 | # See the License for the specific language governing permissions and |
15 | 15 | # limitations under the License. |
16 | 16 |
|
17 | | -import configparser |
18 | | -import os |
19 | 17 | import re |
| 18 | +import sys |
20 | 19 |
|
21 | 20 | from google.cloud import spanner |
22 | 21 |
|
23 | | -config = configparser.ConfigParser() |
24 | | -if os.path.exists("test.cfg"): |
25 | | - config.read("test.cfg") |
26 | | -else: |
27 | | - config.read("setup.cfg") |
28 | | -db_url = config.get("db", "default") |
29 | 22 |
|
30 | | -project = re.findall(r"projects(.*?)instances", db_url) |
31 | | -instance_id = re.findall(r"instances(.*?)databases", db_url) |
| 23 | +def main(argv): |
| 24 | + db_url = argv[0] |
32 | 25 |
|
33 | | -client = spanner.Client(project="".join(project).replace("/", "")) |
34 | | -instance = client.instance(instance_id="".join(instance_id).replace("/", "")) |
35 | | -database = instance.database("compliance-test") |
| 26 | + project = re.findall(r"projects(.*?)instances", db_url) |
| 27 | + instance_id = re.findall(r"instances(.*?)databases", db_url) |
36 | 28 |
|
37 | | -database.update_ddl(["DROP TABLE account", "DROP TABLE alembic_version"]).result(120) |
| 29 | + client = spanner.Client(project="".join(project).replace("/", "")) |
| 30 | + instance = client.instance(instance_id="".join(instance_id).replace("/", "")) |
| 31 | + database = instance.database("compliance-test") |
| 32 | + |
| 33 | + database.update_ddl(["DROP TABLE account", "DROP TABLE alembic_version"]).result(120) |
| 34 | + |
| 35 | + |
| 36 | +if __name__ == "__main__": |
| 37 | + main(sys.argv[1:]) |
0 commit comments