Add __main__.py to do single file conversion from command line

Author: rschroll

rmrl/__main__.py

@@ -0,0 +1,47 @@
+'''
+Copyright 2021 Robert Schroll
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <https://www.gnu.org/licenses/>.
+'''
+
+import argparse
+import io
+import sys
+import zipfile
+
+from . import render
+from .sources import ZipSource
+
+def main():
+    parser = argparse.ArgumentParser(description="Render a PDF file from a Remarkable document.")
+    parser.add_argument('input', help="Filename of zip file, or root-level unpacked file of document.  Use '-' to read zip file from stdin.")
+    parser.add_argument('output', nargs='?', default='', help="Filename where PDF file should be written.  Omit to write to stdout.")
+    args = parser.parse_args()
+
+    source = args.input
+    if source == '-':
+        # zipfile needs to seek, so we need to read this all in
+        source = ZipSource(zipfile.ZipFile(io.BytesIO(sys.stdin.buffer.read())))
+    if args.output:
+        fout = open(args.output, 'wb')
+    else:
+        fout = sys.stdout.buffer
+
+    stream = render(source)
+    fout.write(stream.read())
+    fout.close()
+    return 0
+
+if __name__ == '__main__':
+    sys.exit(main())