blob: 87f368a04a875c9781d3f5fdc3f7c1c215bd23e9 (
plain)
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
|
#!/bin/env python
import sys
import subprocess
def main():
output = "pdfunite.pdf"
files = []
for i, arg in zip(range(len(sys.argv)), sys.argv):
if i == 0:
continue
if arg == "-o":
# output = '_'.join(sys.argv[i + 1 : len(sys.argv)])
output = str(sys.argv[i + 1])
continue
if not arg.startswith("-") and sys.argv[i - 1] != "-o":
files.append(arg)
convert_to_pdf = ["jpg", "jpeg", "png"]
for i, file in zip(range(len(files)), files):
for extension in convert_to_pdf:
if file.endswith(extension):
tmpfile = f"/tmp/{file.replace('/', '_')}.pdf"
subprocess.run(["convert", file, tmpfile])
files[i] = tmpfile
cmd_str = f"/usr/bin/pdfunite {' '.join(files)} {output}"
subprocess.run(cmd_str, shell=True)
if __name__ == "__main__":
main()
|
