mirror of https://github.com/HandBrake/HandBrake
151 lines
4.8 KiB
Python
Executable File
151 lines
4.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import types
|
|
import os
|
|
import sys
|
|
import json
|
|
import getopt
|
|
import posixpath
|
|
from collections import OrderedDict
|
|
try:
|
|
from urlparse import urlsplit
|
|
from urllib import unquote
|
|
except ImportError: # Python 3
|
|
from urllib.parse import urlsplit, unquote
|
|
|
|
|
|
def url2filename(url):
|
|
path = urlsplit(url).path
|
|
return posixpath.basename(unquote(path))
|
|
|
|
def islocal(url):
|
|
result = urlsplit(url)
|
|
return result.scheme == "" and result.netloc == ""
|
|
|
|
class SourceType:
|
|
contrib = 1
|
|
archive = 2
|
|
|
|
class SourceEntry:
|
|
def __init__(self, url, entry_type, sha256=None):
|
|
self.url = url
|
|
self.entry_type = entry_type
|
|
self.sha256 = sha256
|
|
|
|
class FlatpakManifest:
|
|
def __init__(self, source_list, runtime, template=None):
|
|
if template != None:
|
|
with open(template, 'r') as fp:
|
|
self.manifest = json.load(fp, object_pairs_hook=OrderedDict)
|
|
|
|
self.finish_args = self.manifest["finish-args"]
|
|
self.modules = self.manifest["modules"]
|
|
self.hbmodule = self.modules[len(self.modules) - 1]
|
|
self.sources = [None]
|
|
|
|
self.hbmodule["sources"] = self.sources
|
|
else:
|
|
self.manifest = OrderedDict()
|
|
self.modules = []
|
|
self.hbmodule = OrderedDict()
|
|
self.sources = [None]
|
|
|
|
self.manifest["finish-args"] = self.finish_args
|
|
self.manifest["modules"] = self.modules
|
|
self.modules[0] = self.hbmodule
|
|
self.hbmodule["sources"] = self.sources
|
|
|
|
if runtime != None:
|
|
self.manifest["runtime-version"] = runtime
|
|
|
|
handbrake_found = False
|
|
for key, value in source_list.items():
|
|
source = OrderedDict()
|
|
if islocal(value.url):
|
|
source["path"] = value.url
|
|
else:
|
|
if value.sha256 == "" or value.sha256 == None:
|
|
continue
|
|
source["url"] = value.url
|
|
source["sha256"] = value.sha256
|
|
|
|
if value.entry_type == SourceType.archive:
|
|
if handbrake_found:
|
|
print "Error: only one archive source permitted"
|
|
sys.exit(3)
|
|
|
|
source["type"] = "archive"
|
|
source["strip-components"] = 1
|
|
self.sources[0] = source
|
|
handbrake_found = True
|
|
|
|
elif value.entry_type == SourceType.contrib:
|
|
source["type"] = "file"
|
|
source["dest"] = "download"
|
|
source["dest-filename"] = url2filename(value.url)
|
|
self.sources.append(source)
|
|
|
|
|
|
def usage():
|
|
print "create_flatpak_manifest [-a <archive>] [-c <contrib>] [-s <sha265>] [-t <template>] [-r <sdk-runtime-version] [-h] [<dst>]"
|
|
print " -a --archive - Main archive (a.k.a. HB sources)"
|
|
print " -c --contrib - Contrib download URL (can be repeated)"
|
|
print " -s --sha256 - sha256 of previous file on command line"
|
|
print " -t --template - Flatpak manifest template"
|
|
print " -r --runtime - Flatpak SDK runtime version"
|
|
print " -h --help - Show this message"
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
opts, args = getopt.getopt(sys.argv[1:], "a:c:s:t:r:h",
|
|
["archive=", "contrib=", "sha265=",
|
|
"template=", "runtime=", "help"])
|
|
except getopt.GetoptError:
|
|
print "Error: Invalid option"
|
|
usage()
|
|
sys.exit(2)
|
|
|
|
if len(args) > 1:
|
|
usage()
|
|
exit(2)
|
|
|
|
source_list = OrderedDict()
|
|
current_source = None
|
|
runtime = None
|
|
for opt, arg in opts:
|
|
if opt in ("-h", "--help"):
|
|
usage()
|
|
sys.exit()
|
|
elif opt in ("-a", "--archive"):
|
|
if arg != None and arg != "":
|
|
current_source = arg
|
|
source_list[arg] = SourceEntry(arg, SourceType.archive)
|
|
else:
|
|
current_source = None
|
|
elif opt in ("-c", "--contrib"):
|
|
if arg != None and arg != "":
|
|
current_source = arg
|
|
source_list[arg] = SourceEntry(arg, SourceType.contrib)
|
|
else:
|
|
current_source = None
|
|
elif opt in ("-s", "--sha256"):
|
|
if current_source != None:
|
|
source_list[current_source].sha256 = arg
|
|
elif opt in ("-t", "--template"):
|
|
template = arg
|
|
elif opt in ("-r", "--runtime"):
|
|
runtime = arg
|
|
|
|
if len(args) > 0:
|
|
dst = args[0]
|
|
else:
|
|
dst = None
|
|
|
|
manifest = FlatpakManifest(source_list, runtime, template)
|
|
if dst != None:
|
|
with open(dst, 'w') as fp:
|
|
json.dump(manifest.manifest, fp, ensure_ascii=False, indent=4)
|
|
else:
|
|
print json.dumps(manifest.manifest, ensure_ascii=False, indent=4)
|
|
|