mwgencode 1.4.1__py3-none-any.whl

gencode/gencode/gen_state_code.py

@@ -0,0 +1,171 @@
+import os
+
+from gencode.importxmi.import_states import ImportStates
+from gencode.utils import saveUTF8File
+
+
+class Gen_State_Code():
+    def __init__(self,root_path,xmifile):
+        self.state_imp = ImportStates()
+        self.state_imp.impxml(xmifile)
+        self.root_path = root_path
+        self.codes = []
+
+    def _gen_state_code(self):
+        def get_states_code():
+            self.codes.append('  ')
+            for states in self.state_imp.states:
+                for key,values in states.items():
+                    self.codes.append('%s = ['%key)
+                    for state in values:
+                        state_doc = '        #%s:%s'%(state['name'],state['doc'])
+                        state_doc = state_doc+ ','.join([";%s:%s"%(ent['name'],ent['doc']) for ent in state['onenter']])
+                        state_doc = state_doc + ','.join([";%s:%s" % (ext['name'], ext['doc']) for ext in state['onexit']])
+                        self.codes.append(state_doc)
+                        state_code = "        State(name='%s',"%state['name']
+                        on_enter_code = ','.join(["'%s'"%ent['name'] for ent in state['onenter']])
+                        if on_enter_code:
+                            state_code = state_code+ "on_enter=[%s],"%on_enter_code
+                        on_exit_code = ','.join(["'%s'"%ext['name'] for ext in state['onexit']])
+                        if on_exit_code:
+                            state_code = state_code + "on_exit=[%s],"%on_exit_code
+                        state_code = state_code + 'ignore_invalid_triggers=True,'
+                        state_code = state_code + '),'
+                        self.codes.append(state_code)
+                    self.codes.append(']')
+
+        def get_trans_code():
+            self.codes.append('  ')
+            def get_conditions(trg):
+                return ','.join(["'%s'" % g for g in trg['guard'].split('&')])
+
+            self.codes.append('transitions = [')
+            for trg in self.state_imp.trans:
+                trg_code = "    {'trigger': '%s', "%trg['trigger']
+                trg_code = trg_code + "'source': '%s', "%trg['source']
+                trg_code = trg_code +"'dest': '%s', "%trg['target']
+                if trg['guard']:
+                    code_guard = ','.join(["'%s'" % g for g in trg['guard'].split('&')])
+                    trg_code = trg_code +"'conditions' : [%s], "%code_guard
+                if  trg['after']:
+                    trg_code = trg_code +"'after' : '%s',"%trg['after']
+                if trg['before']:
+                    trg_code = trg_code +"'before' : '%s_before',"%trg['before'].split('(')[0]
+                trg_code = trg_code +'},'
+                self.codes.append(trg_code)
+            self.codes.append(']')
+
+
+        self.codes.append('from transitions import State')
+        self.codes.append('  ')
+        get_states_code()
+        get_trans_code()
+        filename = os.path.join(self.root_path, 'states.py')
+        if os.path.exists(filename):
+            print('the file is exists,filename:%s'%filename)
+            print('codes :')
+            for c in self.codes:
+                print(c)
+            return
+        saveUTF8File(filename,self.codes)
+
+    def gen_state_code(self):
+        self.codes = []
+        self._gen_state_code()
+
+    def _gen_model_code(self):
+        self.codes.append('from transitions import Machine')
+        self.codes.append('from fsm.states import states,transitions')
+        self.codes.append('import logging')
+        self.codes.append('from transitions import logger')
+        self.codes.append(' ')
+        self.codes.append('logger.setLevel(logging.DEBUG)')
+        self.codes.append(' ')
+        self.codes.append('class State_Model(object):')
+        self.codes.append('    def __init__(self):')
+        self.codes.append('        self._add_extend_state()')
+        self.codes.append("        self.machine = Machine(model=self, states=states, transitions=transitions, initial='free')")
+        self.codes.append('')
+
+        def gen_add_extend_state():
+            self.codes.append('    def _add_extend_state(self):')
+            self.codes.append('        #fault = State(name="fault", )')
+            self.codes.append('        #states.append(fault)')
+            self.codes.append("        #transitions.append({'trigger': 'fail', 'source': '*', 'dest': 'fault',})")
+            self.codes.append("        #transitions.append({'trigger': 'fix', 'source': 'fault', 'dest': 'free',})")
+            self.codes.append("        pass")
+
+        def gen_failure_recovery():
+            self.codes.append('def failure_recovery(statemodel,state):')
+            self.codes.append('    if state == statemodel.state:')
+            self.codes.append('        return')
+            for sts in self.state_imp.states:
+                for key, values in sts.items():
+                    for value in values:
+                        self.codes.append("    elif state == '%s':"%value['name'])
+                        self.codes.append("        statemodel.to_%s()"%value['name'])
+            self.codes.append("    logger.debug('recovery state to (%s)' % statemodel.state)")
+
+        def get_func():
+            for sts in self.state_imp.states:
+                for key,values in sts.items():
+                    for value in values:
+                        for ent in value['onenter']:
+                            self.codes.append('')
+                            self.codes.append('    #%s' % ent['doc'])
+                            self.codes.append('    def %s(self):' % ent['name'])
+                            self.codes.append('        logger.debug("onenter:%s")' % ent['name'])
+                        for ext in value['onexit']:
+                            self.codes.append('')
+                            self.codes.append('    #%s' % ext['doc'])
+                            self.codes.append('    def %s(self):' % ext['name'])
+                            self.codes.append('        logger.debug("onexit:%s")' % ext['name'])
+            for tr in self.state_imp.trans:
+                if tr['before']:
+                    self.codes.append('')
+                    self.codes.append('    #%s' % tr['doc'])
+                    # check_in(carid) 解析成 ['check_in','carid)']
+                    func_bfr =tr['before'].split('(')
+                    args_bfr = lambda x :', %s:'%func_bfr[1] if len(func_bfr)>1 else '):'
+                    self.codes.append('    def %s_before(self %s:' % (func_bfr[0],args_bfr(func_bfr)))
+                    self.codes.append('        logger.debug("before:%s")' % tr['before'])
+                if tr['after']:
+                    self.codes.append('')
+                    self.codes.append('    #%s' % tr['doc'])
+                    self.codes.append('    def %s(self):' % tr['after'])
+                    self.codes.append('        logger.debug("after:%s")' % tr['after'])
+                if tr['guard']:
+                    for g in tr['guard'].split('&'):
+                        self.codes.append('')
+                        self.codes.append('    #%s' % tr['doc'])
+                        self.codes.append('    def %s(self):' % g)
+                        self.codes.append('        logger.debug("guard:%s")' % g)
+
+        gen_add_extend_state()
+        get_func()
+        gen_failure_recovery()
+        filename = os.path.join(self.root_path, 'model.py')
+        if os.path.exists(filename):
+            print('the file is exists,filename:%s'%filename)
+            print('code :')
+            for c in self.codes:
+                print(c)
+            return
+        saveUTF8File(filename,self.codes)
+
+    def gen_model_code(self):
+        self.codes = []
+        self._gen_model_code()
+
+
+if __name__ == '__main__':
+    g = Gen_State_Code(r"d:\temp",r"D:\mwwork\projects\iparking\docs-parking\parking-state.xml")
+    # g.gen_state_code()
+    g.gen_model_code()
+    # def state(self):
+    #     gen_code = Gen_State_Code(os.path.join(self.rootpath, 'fsm'), self.modelfile)
+    #     gen_code.gen_state_code()
+    #     gen_code.gen_model_code()
+
+
+

gencode/gencode/gen_swagger_code.py

@@ -0,0 +1,94 @@
+from gencode.utils import  saveUTF8File, get_merge_file
+from gencode.importmdj.import_swagger2_class import  ImportSwagger
+from gencode.swg2_class_models import Swagger2,Path2,Action2,Tag2,Param2,Resp2
+import os
+from jinja2 import FileSystemLoader, Environment
+from enum import Enum
+from gencode.ext import Session
+import logging
+match = lambda fld, evalue: True if fld.fieldtype == evalue.value else False
+
+required = lambda st: ('noreq' not in st) and ('nr' not in st)
+
+class ProjectType(Enum):
+    flask = 'flask'
+    aiohttp = 'aiohttp'
+
+right_type = {'post':'insert','put':'edit','delete':'delete','get':'view'}
+
+def get_params(params):
+    result = []
+    for param in params:
+        p_name = param.name
+        if not param.required:
+            p_name = p_name + ' = None'
+        # heard 的不要写入到参数中
+        if param.is_header:
+            continue
+        result.append(p_name)
+    return ','.join(result)
+
+class GenSwaggerCodeFromUml():
+    def __init__(self, root_path, xmifile, type='flask'):
+        self.type = type
+        self.swagger_imp = ImportSwagger()
+        self.swager = self.swagger_imp.impUMLModels(xmifile)
+        self.root_path = root_path
+        tmp_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'template')
+        load = FileSystemLoader(tmp_path)
+        self.env = Environment(loader=load)
+        self.session = Session()
+
+    def gen_swagger_code(self,outfile):
+        codes = []
+        swg = self.swager
+        if self.type == ProjectType.aiohttp.value:
+            swg.host = ''
+        else:
+            swg.host = 'host: localhost:8000'
+        swg.swghost = '{{host}}'
+        swg.swgtitle = '{{title}}'
+        paths = self.session.query(Path2).\
+            join(Tag2,Tag2.id==Path2.tagid).\
+            join(Swagger2,Swagger2.id==Tag2.swaggerid).\
+            filter(Swagger2.id==self.swager.id).\
+            order_by(Swagger2.name,Tag2.name,Path2.name).all()
+        template = self.env.get_template('swagger_file.yaml')
+        result = template.render(swg=swg,paths=paths)  # self.defines)
+        result = '\n'.join([line for line in result.split('\n') if line.strip() != ''])
+        # print(result)
+        codes.append(result)
+        version_dir = swg.version.replace('.', '_')
+        if outfile:
+            filename = outfile
+        else:
+            fname = '%s.yaml' % swg.name
+            filename = os.path.join(self.root_path, 'swagger', version_dir, fname)
+        if os.path.exists(filename):
+            logging.info('the file is exists,将被覆盖！,filename:%s ' % filename)
+        saveUTF8File(filename, codes,writegeninfo=False)
+
+    def gen_swagger_ctr_code(self):
+        template = self.env.get_template('swg_ctrl_code.pys')
+        version_dir = self.swager.version.replace('.', '_')
+        for tag in self.swager.tags:
+            codes = template.render(paths = tag.paths,
+                                    right_type=right_type,
+                                    get_params=get_params)
+            codes = [line for line in codes.split('\n') if line.strip() != '']
+            fname = '%s.py' % tag.name
+            filename = os.path.join(self.root_path, 'app', 'api', version_dir, fname)
+            if os.path.exists(filename):
+                logging.info('文件存在,将被合并,文件:%s' % filename)
+                saveUTF8File(filename, get_merge_file(codes, filename))
+            else:
+                saveUTF8File(filename, codes)
+
+
+if __name__ == '__main__':
+    g = GenSwaggerCodeFromUml(r"d:\temp\swg", r"D:\mwwork\projects\its\mobile_gateway_server\docs\uml analyse.xml")
+    g.gen_swagger_code()
+
+
+
+

gencode/gencode/gen_tests_code.py

@@ -0,0 +1,113 @@
+from gencode.utils import  saveUTF8File, get_merge_file
+from gencode.importmdj.import_swagger2_class import  ImportSwagger
+from gencode.swg2_class_models import Swagger2,Path2,Action2,Tag2,Param2,Resp2
+import os
+from jinja2 import FileSystemLoader, Environment
+from enum import Enum
+from gencode.ext import Session
+import json
+import codecs
+import logging
+
+match = lambda fld, evalue: True if fld.fieldtype == evalue.value else False
+
+required = lambda st: ('noreq' not in st) and ('nr' not in st)
+
+class ProjectType(Enum):
+    flask = 'flask'
+    aiohttp = 'aiohttp'
+
+right_type = {'post':'insert','put':'edit','delete':'delete','get':'view'}
+
+def get_params(params):
+    result = []
+    for param in params:
+        p_name = param.name
+        if param.is_query:
+            p_name = p_name + '=None'
+        # heard 的不要写入到参数中
+        else:
+            continue
+        result.append(p_name)
+    return '?'+'&'.join(result) if result else ''
+
+def get_define(ref_name,swg):
+    for defn in swg.defines:
+        if defn.name==ref_name:
+            return defn
+
+def get_define_js(defn,swg):
+
+    result ={}
+    for att in defn.attrs:
+        if att.type_attr=='object':
+            defn1 = get_define(att.ref_name,swg)
+            assert defn1, "%s's is null" % (defn.call_path())
+            defn1_js = get_define_js(defn1,swg)
+            if hasattr(defn1,'is_array') and defn1.is_array:
+                result[att.name] = [defn1_js]
+            else:
+                result[att.name] = defn1_js
+            continue
+        result[att.name] = '%s'%att.type
+    return result
+def get_resp_js(act):
+    def_resp = act.default_resp
+    swg = act.path.tag.swagger
+    defn = get_define(def_resp.ref_name,swg)
+    if not defn:
+        result =''
+    else:
+        result = get_define_js(defn,swg)
+    if def_resp.is_array:
+        return [result]
+    return result
+
+class GenTestsCodeFromUml():
+    def __init__(self, root_path, xmifile, type='flask'):
+        self.type = type
+        self.swagger_imp = ImportSwagger()
+        self.swager = self.swagger_imp.impUMLModels(xmifile)
+        self.root_path = root_path
+        self.tmp_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'template','tests')
+        load = FileSystemLoader(self.tmp_path)
+        self.env = Environment(loader=load)
+        self.session = Session()
+    def save_as(self,sfile,dfile):
+        if os.path.exists(dfile):
+            logging.info('the file(%s) is exist'%dfile)
+            return
+        with codecs.open(sfile,encoding='utf8') as f:
+            codes = f.readlines()
+            saveUTF8File(dfile,[code.rstrip() for code in codes])
+
+    def __gen_test_classmng_code(self):
+        template = self.env.get_template('test_classmng.pys')
+        for tag in self.swager.tags:
+            fname = 'test_%s.py' % tag.name
+            filename = os.path.join(self.root_path, 'tests', fname)
+            if os.path.exists(filename):
+                logging.info('测试文件存在,文件:%s' % filename)
+                continue
+            codes = template.render(tag = tag,
+                                    get_params=get_params,
+                                    get_resp_js = get_resp_js
+                                    )
+            codes = [line for line in codes.split('\n') if line.strip() != '']
+            saveUTF8File(filename, codes)
+
+    def gen_tests_codes(self):
+            self.__gen_test_classmng_code()
+            self.save_as(os.path.join(self.tmp_path ,'__init__.pys'),
+                         os.path.join(os.path.realpath(self.root_path),'tests', '__init__.py'))
+            self.save_as(os.path.join(self.tmp_path, 'run.pys'),
+                         os.path.join(os.path.realpath(self.root_path), 'tests', 'run.py'))
+            self.save_as(os.path.join(self.tmp_path, 'init_test_data.pys'),
+                         os.path.join(os.path.realpath(self.root_path), 'tests', 'init_test_data.py'))
+            saveUTF8File(os.path.join(os.path.realpath(self.root_path),'tests', 'test_base.py'),
+                         [self.env.get_template('test_base.pys').render(swagger=self.swager)],
+                         exist_ok=False
+                         )
+
+
+

gencode/gencode/sample/__init__.py

@@ -0,0 +1,2 @@
+#
+#

gencode/gencode/sample/babel.cfg

@@ -0,0 +1,3 @@
+[python: **.py]
+[jinja2: **/templates/**.html]
+extensions=jinja2.ext.autoescape,jinja2.ext.with_

gencode/gencode/sample/config-sample.ini

@@ -0,0 +1,9 @@
+[web]
+#web port
+port = 8000
+
+[reg_service]
+#true: register service to kong,false:do not
+kong = false
+#true : register service to consul,false :do not
+consul = false

gencode/gencode/sample/config.ini

@@ -0,0 +1,14 @@
+[web]
+#web port
+port = 8000
+
+[reg_service]
+#true: register service to kong,false:do not
+kong = False
+#true : register service to consul,false :do not
+consul = False
+
+[app_config]
+# select development config,can take away kong auth
+config = development
+

gencode/gencode/sample/create_new_table_run.pys

@@ -0,0 +1,19 @@
+########################################
+# create by :cxh-PC
+# create time :2018-02-08 15:02:13.462231
+########################################
+from app import create_app_swagger
+
+def create_dev_db():
+    from app import create_app_swagger, db
+    # from app.models import App_confirm
+    app_sw = create_app_swagger('default')
+    app = app_sw.app
+    app_context = app.app_context()
+    app_context.push()
+    #db.d*r*o*p*_*all()
+    db.create_all()
+
+if __name__ == '__main__':
+    # 开了这项会显示不了log？？？
+    create_dev_db()

gencode/gencode/sample/dockerignore.dock

@@ -0,0 +1,57 @@
+# Created by .ignore support plugin (hsz.mobi)
+__pycache__/
+*.py[cod]
+*$py.class
+*.so
+.Python
+env/
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+*.manifest
+*.spec
+pip-log.txt
+pip-delete-this-directory.txt
+htmlcov/
+.tox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*,cover
+.hypothesis/
+*.mo
+*.pot
+*.log
+local_settings.py
+instance/
+.webassets-cache
+.scrapy
+docs/_build/
+target/
+.ipynb_checkpoints
+.python-version
+celerybeat-schedule
+*.sage.py
+.env
+.venv
+venv/
+ENV/
+.spyderproject
+.ropeproject
+*.md
+*.bak
+*.xmi