Added 4 more methods

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "resin-release-tool"
-version = "0.1.0"
+version = "0.1.1"
 readme = "README.md"
 description = "Release tool to have canary in resin.io"
 homepage = "https://github.com/mobilityhouse/resin-release-tool"

resin_release_tool/cli.py

@@ -116,5 +116,36 @@ def releases(releaser, count):
         click.echo(f'{release["end_timestamp"]} {release["commit"]}')
 
 
+@cli.command()
+@pass_releaser
+def show_tags_per_device(releaser):
+    """ Shows device's tags"""
+    tag = releaser.get_tags_per_device()
+    click.echo(tag)
+
+@cli.command()
+@pass_releaser
+def show_device_env_vars(releaser):
+    """Shows overriden ENV VARS in each device"""
+    tag = releaser.get_tags_per_device()
+    env_vars = releaser.get_device_env_vars()
+    click.echo(env_vars)
+
+@cli.command()
+@pass_releaser
+def show_app_env_vars(releaser):
+    """Shows application ENV VARS with default values"""
+    env_vars = releaser.get_app_env_vars()
+    click.echo(env_vars)
+
+@cli.command()
+@pass_releaser
+def set_device_env_vars(releaser):
+    """Override ENV VARS with with values of tags"""
+    tags_per_device = releaser.get_tags_per_device()
+    device_env_vars = releaser.get_device_env_vars()
+    releaser.set_device_env_vars_from_tags(tags_per_device, device_env_vars)
+
+
 if __name__ == '__main__':
     cli()

resin_release_tool/releaser.py

@@ -11,6 +11,8 @@ def __init__(self, token, app_id):
 
         self.app_id = app_id
 
+        self.uuid_list = [] #list of all the UUIDs in the app
+
     def get_info(self):
         return self.models.application.get_by_id(self.app_id)
 
@@ -21,6 +23,69 @@ def get_canaries(self):
             if tag['tag_key'] == 'CANARY']
         return canaries
 
+    def get_tags_per_device(self):
+        all_apps = self.models.application.get_all()
+        if len(next((app for app in all_apps if app['id'] == int(self.app_id)),{})) != 0: #checks if the app exists
+            print("Found app with app_id ", self.app_id, "in Balena")
+            app_info = next((app for app in all_apps if app['id'] == int(self.app_id)),{})
+            all_devices = self.models.device.get_all_by_application_id(app_info['id'])
+            tags_per_device = [{'uuid':device['uuid'],
+                                'tags':self.models.tag.device.get_all_by_device(device['uuid'])}
+                               for device in all_devices]
+            entries_to_remove = ('id', 'device', '__metadata')
+            for device in tags_per_device: #removes all the key/values except, "tag_key" and "value"
+                for tag in device['tags']:
+                    for k in entries_to_remove:
+                        tag.pop(k)
+            for device in tags_per_device:
+                dict_of_vars = {}
+                for tag in device['tags']:
+                    dict_of_vars[tag['tag_key']] = tag['value']
+                del(device['tags'])
+                device['tags'] = dict_of_vars
+            self.uuid_list = [uuid['uuid'] for uuid in tags_per_device]
+            return tags_per_device
+        else:
+            print("Could not find the app with app_id ", self.app_id)
+
+    def get_app_env_vars(self):        
+        allvars = self.models.environment_variables.application.get_all(int(self.app_id))
+        list_of_app_env_vars = [{'id':var['id'], var['name']:var['value']} for var in allvars]
+        return list_of_app_env_vars
+
+    def get_device_env_vars(self):
+        list_of_env_vars_per_device = []
+        for device in self.uuid_list:
+            allvars = self.models.environment_variables.device.get_all(device)
+            list_of_vars = [{'id':var['id'], var['name']:var['value']} for var in allvars]
+            list_of_env_vars_per_device.append({device:list_of_vars})
+        return list_of_env_vars_per_device
+
+    def set_device_env_vars_from_tags(self, tags_per_device, device_env_vars):
+        for device in tags_per_device:
+            uuid = device['uuid']
+            print("----------------------------------------------------------")
+            print("Device: ", uuid)
+            for tag,value in device['tags'].items():
+                print('--')
+                try:
+                    self.models.environment_variables.device.create(device['uuid'], tag, value)
+                    print("creating/overriding var: ", device['uuid'], tag, value)
+                except:
+                    print("var", tag, "already exists!")
+                    for elem in device_env_vars:
+                        if uuid in elem:
+                            list_of_env_vars = elem[uuid]
+                            for var in list_of_env_vars:
+                                if tag in var:
+                                    var_id = var['id']
+                                    old_val = var[tag]
+                                    if old_val != value:
+                                        print("Updating var ", tag, "with value", value)
+                                        self.models.environment_variables.device.update(var_id, value)
+                                    else:
+                                        print("Not updating var ", tag, "since the value did not change!")
+
     @lru_cache()
     def get_releases(self):
         releases = self.models.release.get_all_by_application(self.app_id)