-
Notifications
You must be signed in to change notification settings - Fork 13
/
registry_module_versions.py
95 lines (74 loc) · 3.82 KB
/
registry_module_versions.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
"""
Module for Terraform Enterprise/Cloud Migration Worker: Registry Module Versions.
"""
import os
from .base_worker import TFCMigratorBaseWorker
class RegistryModuleVersionsWorker(TFCMigratorBaseWorker):
"""
A class to represent the worker that will migrate all registry module
versions from one TFC/E org to another TFC/E org.
"""
_api_module_used = "registry_modules"
_required_entitlements = ["private-module-registry"]
def migrate(self):
"""
Function to migrate all registry module versions from one TFC/E org to another TFC/E org.
"""
self._logger.info("Migrating registry module versions...")
source_modules = self._api_source.registry_modules.list()["data"]
target_modules = self._api_target.registry_modules.list()["data"]
target_module_names = \
[target_module["attributes"]["name"] for target_module in target_modules]
for source_module in source_modules:
source_module_name = source_module["attributes"]["name"]
source_module_provider = source_module["attributes"]["provider"]
source_module_version = source_module["attributes"]["version-statuses"][0]["version"]
if source_module_name in target_module_names:
self._logger.info("Registry module: %s, exists. Skipped.", source_module_name)
continue
# Build the new module payload
new_module_payload = {
"data": {
"type": "registry-modules",
"attributes": {
"name": source_module_name,
"provider": source_module_provider
}
}
}
# Create the module in the target organization
self._api_target.registry_modules.create(new_module_payload)
self._logger.info("Registry module: %s, created.", source_module_name)
# Build the new module version payload
new_module_version_payload = {
"data": {
"type": "registry-module-versions",
"attributes": {
"version": source_module_version
}
}
}
# Create the module version in the target organization
new_module_version = self._api_target.registry_modules.create_version(\
source_module_name, source_module_provider, new_module_version_payload)["data"]
self._logger.info("Module version: %s, for module: %s, created.", \
source_module_version, source_module_name)
# Pull module version file and upload to target organization
source_module_file_path = "/tmp/%s.tar.gz" % (source_module_name)
self._api_source.registry_modules.download_latest_source( \
source_module_name, source_module_provider, source_module_file_path)
self._api_target.registry_modules.upload_version(\
source_module_file_path, new_module_version["links"]["upload"])
self._logger.info("Module version file for version: %s, for module: %s, uploaded.", \
source_module_version, source_module_name)
os.remove(source_module_file_path)
self._logger.info("Registry module versions migrated.")
def delete_all_from_target(self):
self._logger.info("Deleting registry modules...")
# TODO: list all of the modules
modules = self._api_target.registry_modules.list()["data"]
if modules:
for module in modules:
self._api_target.registry_modules.destroy(module["attributes"]["name"])
self._logger.info("Registry module: %s, deleted.", module["attributes"]["name"])
self._logger.info("Registry modules deleted.")