Switch all headers to python3

[fdroidserver.git] / fdroidserver / rewritemeta.py

diff --git a/fdroidserver/rewritemeta.py b/fdroidserver/rewritemeta.py
index d28dc936d1e1464b8c9810edbbeb1af9b395cd1b..651349cd3fe60cc98663d3e57a6379550fd72540 100644 (file)
--- a/fdroidserver/rewritemeta.py
+++ b/fdroidserver/rewritemeta.py
@@ -1,7 +1,7 @@
-#!/usr/bin/env python2
-# -*- coding: utf-8 -*-
+#!/usr/bin/env python3
 #
 # rewritemeta.py - part of the FDroid server tools
+# This cleans up the original .txt metadata file format.
 # Copyright (C) 2010-12, Ciaran Gultnieks, ciaran@ciarang.com
 #
 # This program is free software: you can redistribute it and/or modify
@@ -17,41 +17,83 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
-import sys
+from argparse import ArgumentParser
 import os
-from optparse import OptionParser
+import logging
+try:
+    from cStringIO import StringIO
+except:
+    from StringIO import StringIO
+
 import common
+import metadata
 
 config = None
 options = None
 
+
+def proper_format(app):
+    s = StringIO()
+    # TODO: currently reading entire file again, should reuse first
+    # read in metadata.py
+    with open(app.metadatapath, 'r') as f:
+        cur_content = f.read()
+    metadata.write_txt_metadata(s, app)
+    content = s.getvalue()
+    s.close()
+    return content == cur_content
+
+
 def main():
 
     global config, options
 
     # Parse command line...
-    parser = OptionParser()
-    parser.add_option("-v", "--verbose", action="store_true", default=False,
-                      help="Spew out even more information than normal")
-    parser.add_option("-p", "--package", default=None,
-                      help="Process only the specified package")
-    (options, args) = parser.parse_args()
+    parser = ArgumentParser(usage="%(prog)s [options] [APPID [APPID ...]]")
+    common.setup_global_opts(parser)
+    parser.add_argument("-l", "--list", action="store_true", default=False,
+                        help="List files that would be reformatted")
+    parser.add_argument("-t", "--to", default=None,
+                        help="Rewrite to a specific format")
+    parser.add_argument("appid", nargs='*', help="app-id in the form APPID")
+    options = parser.parse_args()
 
     config = common.read_config(options)
 
     # Get all apps...
-    apps = common.read_metadata(package=options.package)
+    allapps = metadata.read_metadata(xref=True)
+    apps = common.read_app_args(options.appid, allapps, False)
+
+    if options.list and options.to is not None:
+        parser.error("Cannot use --list and --to at the same time")
+
+    supported = ['txt', 'yaml']
 
-    if len(apps) == 0 and options.package:
-        print "No such package"
-        sys.exit(1)
+    if options.to is not None and options.to not in supported:
+        parser.error("Must give a valid format to --to")
 
-    for app in apps:
-        print "Writing " + app['id']
-        common.write_metadata(os.path.join('metadata', app['id']) + '.txt', app)
+    for appid, app in apps.iteritems():
+        base, ext = common.get_extension(app.metadatapath)
+        if not options.to and ext not in supported:
+            logging.info("Ignoring %s file at '%s'" % (ext, app.metadatapath))
+            continue
 
-    print "Finished."
+        to_ext = ext
+        if options.to is not None:
+            to_ext = options.to
+
+        if options.list:
+            if not proper_format(app):
+                print app.metadatapath
+            continue
+
+        with open(base + '.' + to_ext, 'w') as f:
+            metadata.write_metadata(to_ext, f, app)
+
+        if ext != to_ext:
+            os.remove(app.metadatapath)
+
+    logging.debug("Finished.")
 
 if __name__ == "__main__":
     main()
-