(self)
| 467 | return os_data |
| 468 | |
| 469 | def populate_env(self): |
| 470 | env_file = '' |
| 471 | bcolors.printcolor('Generating Environment Variables file...', bcolors.OKBLUE) |
| 472 | if self.static_storage_provider == 'gcp': |
| 473 | env_file = f"GCP_SERVICE_ACCOUNT_FILE_PATH={self.gcp_credentials_path}\n" |
| 474 | env_file += f"CLOUD_STORAGE_BUCKET={self.bucket_name}\n" |
| 475 | env_file += f"ML__CLOUD_STORAGE_BUCKET={self.bucket_name}\n" |
| 476 | env_file += f"DIFFGRAM_STATIC_STORAGE_PROVIDER={self.static_storage_provider}\n" |
| 477 | elif self.static_storage_provider == 'aws': |
| 478 | env_file = f"DIFFGRAM_AWS_ACCESS_KEY_ID={self.s3_access_id}\n" |
| 479 | env_file += f"DIFFGRAM_AWS_ACCESS_KEY_SECRET={self.s3_access_secret}\n" |
| 480 | env_file += f"DIFFGRAM_S3_BUCKET_NAME={self.bucket_name}\n" |
| 481 | env_file += f"DIFFGRAM_S3_BUCKET_REGION={self.bucket_region}\n" |
| 482 | env_file += f"IS_DIFFGRAM_S3_V4_SIGNATURE={self.is_aws_signature_v4}\n" |
| 483 | env_file += f"ML__DIFFGRAM_S3_BUCKET_NAME={self.bucket_name}\n" |
| 484 | env_file += f"DIFFGRAM_STATIC_STORAGE_PROVIDER={self.static_storage_provider}\n" |
| 485 | env_file += "GCP_SERVICE_ACCOUNT_FILE_PATH=/dev/null\n" |
| 486 | elif self.static_storage_provider == 'minio': |
| 487 | env_file = f"DIFFGRAM_MINIO_ENDPOINT_URL={self.s3_endpoint_url}\n" |
| 488 | env_file += f"DIFFGRAM_MINIO_ACCESS_KEY_ID={self.s3_access_id}\n" |
| 489 | env_file += f"DIFFGRAM_MINIO_ACCESS_KEY_SECRET={self.s3_access_secret}\n" |
| 490 | env_file += f"DIFFGRAM_S3_BUCKET_NAME={self.bucket_name}\n" |
| 491 | env_file += f"DIFFGRAM_S3_BUCKET_REGION={self.bucket_region}\n" |
| 492 | env_file += f"IS_DIFFGRAM_S3_V4_SIGNATURE={True}\n" |
| 493 | env_file += f"ML__DIFFGRAM_S3_BUCKET_NAME={self.bucket_name}\n" |
| 494 | env_file += f"DIFFGRAM_STATIC_STORAGE_PROVIDER={self.static_storage_provider}\n" |
| 495 | env_file += "GCP_SERVICE_ACCOUNT_FILE_PATH=/dev/null\n" |
| 496 | elif self.static_storage_provider == 'azure': |
| 497 | env_file = f"DIFFGRAM_AZURE_CONNECTION_STRING={self.azure_connection_string}\n" |
| 498 | env_file += f"DIFFGRAM_AZURE_CONTAINER_NAME={self.bucket_name}\n" |
| 499 | env_file += f"ML__DIFFGRAM_AZURE_CONTAINER_NAME={self.bucket_name}\n" |
| 500 | env_file += f"DIFFGRAM_STATIC_STORAGE_PROVIDER={self.static_storage_provider}\n" |
| 501 | env_file += "GCP_SERVICE_ACCOUNT_FILE_PATH=/dev/null\n" |
| 502 | |
| 503 | fernet_key = base64.urlsafe_b64encode(os.urandom(32)).decode("utf-8") |
| 504 | env_file += f"FERNET_KEY={fernet_key}\n" |
| 505 | env_file += f"USER_PASSWORDS_SECRET={create_random_string(10)}\n" |
| 506 | env_file += f"INTER_SERVICE_SECRET={create_random_string(10)}\n" |
| 507 | env_file += f"SECRET_KEY={create_random_string(18)}\n" |
| 508 | env_file += "WALRUS_SERVICE_URL_BASE=http://walrus:8082/\n" |
| 509 | |
| 510 | env_file += f"DIFFGRAM_ERROR_SEND_TRACES_IN_RESPONSE={True}\n" |
| 511 | |
| 512 | install_fingerprint = self.gen_install_finger_print() |
| 513 | env_file += f"DIFFGRAM_SYSTEM_MODE=sandbox\n" |
| 514 | env_file += f"SERVICE_ACCOUNT_FULL_PATH=/gcp/gcp-service-account.json\n" |
| 515 | env_file += f"GOOGLE_APPLICATION_CREDENTIALS=/gcp/gcp-service-account.json\n" |
| 516 | env_file += f"PYTHONPATH=/gcp/gcp-service-account.json\n" |
| 517 | env_file += f"DIFFGRAM_INSTALL_FINGERPRINT={install_fingerprint}\n" |
| 518 | env_file += f"DIFFGRAM_VERSION_TAG={self.diffgram_version}\n" |
| 519 | env_file += f"DIFFGRAM_HOST_OS={self.get_system_os()}\n" |
| 520 | |
| 521 | env_file += f"RABBITMQ_DEFAULT_USER={self.rabbit_username}\n" |
| 522 | env_file += f"RABBITMQ_DEFAULT_PASS={self.rabbit_pass}\n" |
| 523 | env_file += f"RABBITMQ_HOST={self.rabbit_host}\n" |
| 524 | env_file += f"RABBITMQ_PORT={self.rabbit_port}\n" |
| 525 | |
| 526 | if self.local_database: |
no test coverage detected