forked from mookfist/repo
/
pypi_addon_generator.py
executable file
·134 lines (100 loc) · 3.19 KB
/
pypi_addon_generator.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
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
#!/usr/bin/env python
###
# Simple tool to turn any pypi module into a kodi addon
#
# by mookfist
###
import os
import sys
import pprint
try:
from urllib.request import Request
from urllib.request import urlopen
except ImportError:
from urllib2 import Request
from urllib2 import urlopen
import tarfile
import json
import shutil
import zipfile
from xml.sax.saxutils import escape
# https://pypi.python.org/packages/source/p/pybuilder-nose/pybuilder-nose-0.0.5.tar.gz#md5=f8794d709967d109c3e1ffb86096196b
pp = pprint.PrettyPrinter(indent=2)
PROVIDER_NAME = "Mookfist"
XBMC_PYTHON_VER = "2.14.0"
ADDON_XML = """<?xml version="1.0" encoding="utf-8"?>
<addon id="script.module.python-%(modname)s" version="%(version)s" name="python-%(modname)s" provider-name="%(modauthor)s">
<requires>
<import addon="xbmc.python" version="%(xbmc_python_ver)s" />
</requires>
<extension point="xbmc.python.module" library="lib" />
<extension point="xbmc.addon.metadata">
<summary lang="en">%(summary)s</summary>
<description lang="en">%(description)s</description>
<disclaimer>This addon is just a wrapper around the original module found at PyPI</disclaimer>
<language>en</language>
<platform>all</platform>
<source>%(source_url)s</source>
<email>%(author_email)s</email>
<website>%(website)s</website>
</extension>
</addon>"""
BASE_PATH = os.path.realpath(sys.argv[1])
mod = sys.argv[2]
metadata_url = "https://pypi.python.org/pypi/%s/json" % (mod)
tmp_dir = 'tmp/%s' % mod
if not os.path.exists(tmp_dir):
os.makedirs(tmp_dir)
os.chdir(tmp_dir)
print("Getting metadata from %s" % metadata_url)
req = Request(metadata_url)
res = urlopen(req)
metadata = json.loads(res.read().decode("utf-8"))
ver = metadata['info']['version']
for release in metadata['releases'][ver]:
if release['packagetype'] == 'sdist':
break
url = release['url']
ext = url.split('.')[-1]
filename = os.path.basename(url)
print("Downloading %s" % (url))
req = Request(url)
res = urlopen(req)
f = open(filename, 'wb')
f.write(res.read())
f.close()
print("Unpacking")
if ext == "whl" or ext == "egg" or ext == "zip":
archive = zipfile.ZipFile(filename)
elif ext == "gz" or ext == "tar" or ext == "tgz":
archive = tarfile.open(filename)
else:
print("Extension unknown: %s" % ext)
sys.exit(1)
archive.extractall()
archive.close()
if ext == "whl" or ext == "egg":
src_dir = tmp_dir + "/" + mod
else:
src_dir = tmp_dir + "/%s-%s/%s" % (mod, ver, mod)
addon = ADDON_XML % {
'modname': mod,
'version': ver,
'modauthor': escape(metadata['info']['author']),
'xbmc_python_ver': XBMC_PYTHON_VER,
'summary': escape(metadata['info']['summary']),
'description': escape(metadata['info']['description']),
'source_url': escape(metadata['info']['package_url']),
'author_email': escape(metadata['info']['author_email']),
'website': escape(metadata['info']['package_url'])
}
os.chdir(BASE_PATH)
plugin_dir = 'script.module.python-%s' % mod
if os.path.exists(plugin_dir):
shutil.rmtree(plugin_dir)
os.makedirs(plugin_dir + "/lib")
shutil.copyfile('python-icon.png', plugin_dir + '/icon.png')
shutil.move(src_dir, plugin_dir + "/lib")
f = open(plugin_dir + '/addon.xml','w')
f.write(addon)
shutil.rmtree(tmp_dir)