klemek/meme-otron
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

code cleaning

Browse Source
This commit is contained in:
klemek
2020-04-27 19:14:28 +02:00
parent 2223770fb8
commit 4bb9bbb23a
8 changed files with 103 additions and 158 deletions
Download Patch File Download Diff File Expand all files Collapse all files
meme_otron/__main__.py
+14 -15
View File
@@ -1,15 +1,14 @@
import logging
import sys
import os
from . import img_factory as imgf
from . import meme_db as db
from . import img_factory
from . import meme_db
from . import meme_otron
from . import VERSION
if __name__ == "__main__":
db.load_memes()
imgf.load_fonts()
meme_db.load_memes()
img_factory.load_fonts()
# TODO better arguments reading (-h, -o, -v)
@@ -21,29 +20,29 @@ if __name__ == "__main__":
file=sys.stderr)
sys.exit(1)
else:
output_f = None
output_file = None
if "-o" in sys.argv:
i = sys.argv.index("-o")
if len(sys.argv) >= i:
output_f = sys.argv[i + 1]
output_file = sys.argv[i + 1]
del sys.argv[i + 1]
del sys.argv[i]
img = meme_otron.compute(*sys.argv[1:])
if img is None:
hint = db.find_nearest(sys.argv[1])
if hint is not None:
print(f"Did you mean '{hint}'?", file=sys.stderr)
proposal = meme_db.find_nearest(sys.argv[1])
if proposal is not None:
print(f"Did you mean '{proposal}'?", file=sys.stderr)
sys.exit(1)
if output_f is None:
if output_file is None:
with os.fdopen(os.dup(sys.stdout.fileno())) as output:
img.save(output, format="jpeg")
else:
try:
img.save(output_f)
print(f"Wrote '{output_f}'")
img.save(output_file)
print(f"Wrote '{output_file}'")
except OSError as e:
print(f"Cannot write '{output_f}': {e}", file=sys.stderr)
print(f"Cannot write '{output_file}': {e}", file=sys.stderr)
sys.exit(1)
except ValueError as e:
print(f"Cannot write '{output_f}': {e}", file=sys.stderr)
print(f"Cannot write '{output_file}': {e}", file=sys.stderr)
sys.exit(1)