diff sat/bridge/bridge_constructor/bridge_constructor.py @ 2562:26edcf3a30eb

core, setup: huge cleaning: - moved directories from src and frontends/src to sat and sat_frontends, which is the recommanded naming convention - move twisted directory to root - removed all hacks from setup.py, and added missing dependencies, it is now clean - use https URL for website in setup.py - removed "Environment :: X11 Applications :: GTK", as wix is deprecated and removed - renamed sat.sh to sat and fixed its installation - added python_requires to specify Python version needed - replaced glib2reactor which use deprecated code by gtk3reactor sat can now be installed directly from virtualenv without using --system-site-packages anymore \o/
author Goffi <goffi@goffi.org>
date Mon, 02 Apr 2018 19:44:50 +0200
parents src/bridge/bridge_constructor/bridge_constructor.py@0046283a285d
children 56f94936df1e
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/sat/bridge/bridge_constructor/bridge_constructor.py	Mon Apr 02 19:44:50 2018 +0200
@@ -0,0 +1,97 @@
+#!/usr/bin/env python2
+#-*- coding: utf-8 -*-
+
+# SàT: a XMPP client
+# Copyright (C) 2009-2018 Jérôme Poisson (goffi@goffi.org)
+
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero 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 Affero General Public License for more details.
+
+# 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/>.
+
+
+from sat.bridge import bridge_constructor
+from sat.bridge.bridge_constructor.constants import Const as C
+from sat.bridge.bridge_constructor import constructors, base_constructor
+import argparse
+from ConfigParser import SafeConfigParser as Parser
+from importlib  import import_module
+import os
+import os.path
+
+#consts
+__version__ = C.APP_VERSION
+
+
+class BridgeConstructor(object):
+
+    def importConstructors(self):
+        constructors_dir = os.path.dirname(constructors.__file__)
+        self.protocoles = {}
+        for dir_ in os.listdir(constructors_dir):
+            init_path = os.path.join(constructors_dir, dir_, '__init__.py')
+            constructor_path = os.path.join(constructors_dir, dir_, 'constructor.py')
+            module_path = "sat.bridge.bridge_constructor.constructors.{}.constructor".format(dir_)
+            if os.path.isfile(init_path) and os.path.isfile(constructor_path):
+                mod = import_module(module_path)
+                for attr in dir(mod):
+                    obj = getattr(mod, attr)
+                    if not isinstance(obj, type):
+                        continue
+                    if issubclass(obj, base_constructor.Constructor):
+                        name = obj.NAME or dir_
+                        self.protocoles[name] = obj
+                        break
+        if not self.protocoles:
+            raise ValueError("no protocole constructor found")
+
+    def parse_args(self):
+        """Check command line options"""
+        parser = argparse.ArgumentParser(description=C.DESCRIPTION, formatter_class=argparse.RawDescriptionHelpFormatter)
+
+        parser.add_argument("--version", action="version", version= __version__)
+        default_protocole = C.DEFAULT_PROTOCOLE if C.DEFAULT_PROTOCOLE in self.protocoles else self.protocoles[0]
+        parser.add_argument("-p", "--protocole", choices=sorted(self.protocoles), default=default_protocole,
+            help="generate bridge using PROTOCOLE (default: %(default)s)") # (default: %s, possible values: [%s])" % (DEFAULT_PROTOCOLE, ", ".join(MANAGED_PROTOCOLES)))
+        parser.add_argument("-s", "--side", choices=("core", "frontend"), default="core",
+            help="which side of the bridge do you want to make ?") # (default: %default, possible values: [core, frontend])")
+        default_template = os.path.join(os.path.dirname(bridge_constructor.__file__), 'bridge_template.ini')
+        parser.add_argument("-t", "--template", type=file, default=default_template,
+            help="use TEMPLATE to generate bridge (default: %(default)s)")
+        parser.add_argument("-f", "--force", action="store_true",
+            help=("force overwritting of existing files"))
+        parser.add_argument("-d", "--debug", action="store_true",
+            help=("add debug information printing"))
+        parser.add_argument("--no-unicode", action="store_false", dest="unicode",
+            help=("remove unicode type protection from string results"))
+        parser.add_argument("--flags", nargs='+', default=[],
+            help=("constructors' specific flags"))
+        parser.add_argument("--dest-dir", default=C.DEST_DIR_DEFAULT,
+            help=("directory when the generated files will be written (default: %(default)s)"))
+
+        return parser.parse_args()
+
+    def go(self):
+        self.importConstructors()
+        args = self.parse_args()
+        template_parser = Parser()
+        try:
+            template_parser.readfp(args.template)
+        except IOError:
+            print ("The template file doesn't exist or is not accessible")
+            exit(1)
+        constructor = self.protocoles[args.protocole](template_parser, args)
+        constructor.generate(args.side)
+
+
+if __name__ == "__main__":
+    bc = BridgeConstructor()
+    bc.go()