Update synth.py to include Firestore Admin API

schmidt-sebastian · schmidt-sebastian · commit 8f3063a1dcb9 · 2019-03-21T14:45:46.000-07:00
diff --git a/google-cloud-clients/google-cloud-firestore/synth.py b/google-cloud-clients/google-cloud-firestore/synth.py
@@ -20,34 +20,21 @@
 
 gapic = gcp.GAPICGenerator()
 
-service = 'firestore'
-version='v1beta1'
-
-library = gapic.java_library(
-    service=service,
-    version='v1beta1',
-    config_path=f'/google/firestore/artman_firestore.yaml',
-    artman_output_name='')
-
-s.copy(library / f'gapic-google-cloud-{service}-{version}/src', 'src')
-s.copy(library / f'grpc-google-cloud-{service}-{version}/src', f'../../google-api-grpc/grpc-google-cloud-{service}-{version}/src')
-s.copy(library / f'proto-google-cloud-{service}-{version}/src', f'../../google-api-grpc/proto-google-cloud-{service}-{version}/src')
-
-java.format_code('./src')
-java.format_code(f'../../google-api-grpc/grpc-google-cloud-{service}-{version}/src')
-java.format_code(f'../../google-api-grpc/proto-google-cloud-{service}-{version}/src')
-
-version='v1'
-library = gapic.java_library(
-    service=service,
-    version='v1',
-    config_path=f'/google/firestore/artman_firestore_v1.yaml',
-    artman_output_name='')
-
-s.copy(library / f'gapic-google-cloud-{service}-{version}/src', 'src')
-s.copy(library / f'grpc-google-cloud-{service}-{version}/src', f'../../google-api-grpc/grpc-google-cloud-{service}-{version}/src')
-s.copy(library / f'proto-google-cloud-{service}-{version}/src', f'../../google-api-grpc/proto-google-cloud-{service}-{version}/src')
-
-java.format_code('./src')
-java.format_code(f'../../google-api-grpc/grpc-google-cloud-{service}-{version}/src')
-java.format_code(f'../../google-api-grpc/proto-google-cloud-{service}-{version}/src')
+def generate_client(service, version, config_path):
+  library = gapic.java_library(
+      service=service,
+      version=version,
+      config_path=config_path,
+      artman_output_name='')
+
+  s.copy(library / f'gapic-google-cloud-{service}-{version}/src', 'src')
+  s.copy(library / f'grpc-google-cloud-{service}-{version}/src', f'../../google-api-grpc/grpc-google-cloud-{service}-{version}/src')
+  s.copy(library / f'proto-google-cloud-{service}-{version}/src', f'../../google-api-grpc/proto-google-cloud-{service}-{version}/src')
+
+  java.format_code('./src')
+  java.format_code(f'../../google-api-grpc/grpc-google-cloud-{service}-{version}/src')
+  java.format_code(f'../../google-api-grpc/proto-google-cloud-{service}-{version}/src')
+
+generate_client('firestore-admin', 'v1', '/google/firestore/admin/artman_firestore_v1.yaml')
+generate_client('firestore', 'v1beta1', '/google/firestore/artman_firestore.yaml')
+generate_client('firestore', 'v1', '/google/firestore/artman_firestore_v1.yaml')
