fix codegen (airbytehq#605)

airbyte-integrations/base-python/airbyte_protocol/models/__init__.py

@@ -1,8 +1,2 @@
-from .airbyte_catalog import AirbyteStream
-from .airbyte_catalog import AirbyteCatalog
-from .airbyte_connection_status import AirbyteConnectionStatus
-from .airbyte_message import AirbyteMessage
-from .airbyte_message import AirbyteLogMessage
-from .airbyte_message import AirbyteRecordMessage
-from .airbyte_message import AirbyteStateMessage
-from .airbyte_type import AirbyteType
+# generated by generate-protocol-files
+from .airbyte_message import *

airbyte-integrations/base-python/bin/generate-protocol-files.sh

@@ -2,15 +2,24 @@
 
 set -e
 
+[ -z "$ROOT_DIR" ] && exit 1
+
 YAML_DIR=airbyte-protocol/models/src/main/resources/airbyte_protocol
 OUTPUT_DIR=airbyte-integrations/base-python/airbyte_protocol/models
 
-FILES=$(cd ../../$YAML_DIR; ls -1 | grep yaml)
-for f in $FILES
-do
-  FILE_WITHOUT_EXTENSION=""$(echo -n $f | cut -d'.' -f 1)
-  docker run -v "$(pwd)"/../..:/airbyte airbyte/code-generator:dev \
-    --input airbyte/"$YAML_DIR"/"$FILE_WITHOUT_EXTENSION".yaml \
-    --output airbyte/"$OUTPUT_DIR"/"$FILE_WITHOUT_EXTENSION".py \
-    --disable-timestamp
+function main() {
+  rm -rf "$ROOT_DIR/$OUTPUT_DIR"/*.py
+  echo "# generated by generate-protocol-files" > "$ROOT_DIR/$OUTPUT_DIR"/__init__.py
+
+  for f in "$ROOT_DIR/$YAML_DIR"/*.yaml; do
+    filename_wo_ext=$(basename "$f" | cut -d . -f 1)
+    echo "from .$filename_wo_ext import *" >> "$ROOT_DIR/$OUTPUT_DIR"/__init__.py
+
+    docker run -v "$ROOT_DIR":/airbyte airbyte/code-generator:dev \
+      --input "/airbyte/$YAML_DIR/$filename_wo_ext.yaml" \
+      --output "/airbyte/$OUTPUT_DIR/$filename_wo_ext.py" \
+      --disable-timestamp
 done
+}
+
+main "$@"

airbyte-integrations/base-python/build.gradle

@@ -1,10 +1,11 @@
 apply from: rootProject.file('tools/gradle/commons/integrations/image.gradle')
 
 task generateProtocolClassFiles(type: Exec) {
-    workingDir '.'
+    environment 'ROOT_DIR', rootDir.absolutePath
     commandLine 'bin/generate-protocol-files.sh'
-    dependsOn ':airbyte-integrations:base-python:code-generator:buildImage'
+
     outputs.upToDateWhen { false }
+    dependsOn ':airbyte-integrations:base-python:code-generator:buildImage'
 }
 
 build.dependsOn generateProtocolClassFiles