[pkg] missing snap boilerplate

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)))