1 files changed, 7 insertions, 2 deletions
diff --git a/branding/templates/snap/generate.py b/branding/templates/snap/generate.py
index 41d563d..b5ca641 100755
--- a/branding/templates/snap/generate.py
+++ b/branding/templates/snap/generate.py
@@ -13,15 +13,20 @@ from string import Template
 TEMPLATE = 'snapcraft-template.yaml'
 SNAPCRAFT = 'snapcraft.yaml'
 
-
 here = os.path.split(os.path.realpath(__file__))[0]
 data = json.load(open(os.path.join(here, 'data.json')))
 
+DESKTOP_TEMPLATE = 'local/app.desktop'
+DESKTOP = 'local/{}.desktop'.format(data['binaryName'])
 
 template = Template(open(TEMPLATE).read())
-
 with open(SNAPCRAFT, 'w') as output:
     output.write(template.safe_substitute(data))
 
+template = Template(open(DESKTOP_TEMPLATE).read())
+with open(DESKTOP, 'w') as output:
+    output.write(template.safe_substitute(data))
+os.remove(DESKTOP_TEMPLATE)
+
 print("[+] Snapcraft spec written to {path}".format(
     path=os.path.abspath(SNAPCRAFT)))