mwgencode 1.4.1__py3-none-any.whl

gencode/gen_code.py

@@ -0,0 +1,343 @@
+from . import upgrade
+from .gencode.gen_bo_models_code import Gen_bo_models
+from .gencode.gen_swagger_code import  GenSwaggerCodeFromUml
+from .gencode.gen_tests_code import  GenTestsCodeFromUml
+import os
+import logging
+from gencode.utils import saveUTF8File
+import codecs
+from gencode.gencode.export_class2swgclass import ExportClass2SWGClass
+
+class GenCode():
+    def __init__(self,modelfile,rootpath):
+        '''
+        :param modelfile: mdj 文件
+        :param rootpath: 专案根目录
+        :param appdir:   model文件放入的目录
+        '''
+        assert modelfile.endswith('.mdj') or modelfile.endswith('.json'), \
+               'modelfile 必须为 modelfile，扩展名为.mdj'
+        self.modelfile = os.path.abspath(modelfile)
+        self.rootpath = os.path.abspath(rootpath)
+
+    def model(self, outfile='models_base.py', type='flask',appdir='app',exists2cover=True):
+        '''
+        产生基本资料的base类别的代码
+        :param outfile: bomodel的文件名
+        :param type:  flask:flask_sqlalchemy 的 model
+                      sql :' sqlalchemy 的 model
+        :return:
+        '''
+        if appdir:
+            outfile_name = os.path.join(self.rootpath,appdir, outfile)
+        else:
+            outfile_name = os.path.join(self.rootpath, outfile)
+        gen_code = Gen_bo_models(self.modelfile,
+                                 outfile_name,
+                                 type=type)
+        gen_code.gen_code(exists2cover)
+
+    def swagger(self,outfile='swagger.yaml',type='flask'):
+        '''
+        产生swagger.yaml文件
+        :param outfile:
+        :param type:
+        :return:
+        '''
+        imp = GenSwaggerCodeFromUml(self.rootpath,self.modelfile,type)
+        imp.gen_swagger_code(os.path.join(self.rootpath,outfile))
+
+    def export(self,source_umlfile,dest_umlfile,exclude_classes=None):
+        export = ExportClass2SWGClass(source_umlfile,dest_umlfile)
+        export.export(exclude_classes)
+
+    # def add_swgmodel_file(self,swgmodel_name):
+    #     '''
+    #     增加swagger model文件
+    #     :param swgmodel_name: 如：主model名为estopserver，swgmodel为 basedata时，api为estopserver-basedata
+    #     :return:
+    #     '''
+    #     pass
+
+class GenSwagger():
+    def __init__(self,swg_modelfile):
+        '''
+        :param modelfile: mdj 文件
+        :param rootpath: 专案根目录
+        :param appdir:   model文件放入的目录
+        '''
+        assert swg_modelfile.endswith('.mdj') or swg_modelfile.endswith('.json'), \
+               'modelfile 必须为 modelfile，扩展名为.mdj'
+        self.swg_modelfile = os.path.abspath(swg_modelfile)
+        # self.rootpath = os.path.abspath(rootpath)
+        self.export_swg = None
+
+    def export(self,source_umlfile:str='',exclude_classes:list=None,include_classes:list=None):
+        '''
+        汇入类别到swagger，当exclude_classes,include_classes同时为None时，汇入所有的类到到swagger
+        :param source_umlfile:原
+        :param exclude_classes:不用汇出成swagger class的类名
+        :param include_classes:只需要汇入的swagger class的类名
+        '''
+        export = ExportClass2SWGClass(source_umlfile,self.swg_modelfile)
+        export.export(exclude_classes,include_classes)
+
+    def export_one_swgclass(self,bocls:str,source_modelfile:str=''):
+        '''
+        把bocls汇入到swagger，bocls不存在时会创建
+        :param bocls:
+        :param source_modelfile: bocls 所在的umlfile，为空时，为本身swagger file
+        :return:
+        '''
+        export = ExportClass2SWGClass(source_modelfile if source_modelfile else self.swg_modelfile, self.swg_modelfile)
+        export.export_one_swgclass(bocls)
+
+    def add_operation(self, swgpackage_name: str, opname: str, method: str = 'get'):
+        '''
+        在swagger类上增加一个方法
+        :param swgpackage_name: swagger package 名称，如：companymng
+        :param opname: companys
+        :param method: http method，如：get,put,post,delete等
+        :return:
+        '''
+        if self.export_swg is None:
+            self.export_swg = ExportClass2SWGClass(self.swg_modelfile, self.swg_modelfile)
+        self.export_swg.add_operation(swgpackage_name,opname,method)
+
+class GenProject_base():
+    def __init__(self,modelfile,rootpath):
+        # project 类型，flask，aiohttp
+        self.modelfile = os.path.abspath(modelfile)
+        self.rootpath = os.path.abspath(rootpath)
+        self.tmp_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'gencode', 'template')
+        self.sample_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'gencode', 'sample')
+        # 创建其他专案文件
+        from jinja2 import FileSystemLoader, Environment
+        load = FileSystemLoader(self.tmp_path)
+        self.env = Environment(loader=load)
+
+    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])
+
+class GenProject_Sample(GenProject_base):
+    def __init__(self,modelfile,rootpath):
+        super().__init__(modelfile,rootpath)
+        
+    def gen_code(self,include_gencoderun):
+        p_name = os.path.split(self.rootpath)[-1]
+        if not os.path.exists(self.modelfile):
+            if not os.path.exists('docs'):
+                os.makedirs('docs')
+
+            template = self.env.get_template(r'sample.mdj')
+            # 不要覆盖原有的文件
+            saveUTF8File(os.path.join(os.path.realpath(self.rootpath),'docs', '%s.mdj'%p_name),
+                         [template.render(service_name=p_name)],
+                         writegeninfo=False,exist_ok=False)
+        gen_code_file = os.path.join(os.path.realpath(self.rootpath), 'gen_code_run.py')
+        if include_gencoderun and not os.path.exists(gen_code_file):
+            template = self.env.get_template(r'gen_code_run.pys')
+            saveUTF8File(gen_code_file,
+                         [template.render(pro_name=p_name)])
+        gen_code_yaml = os.path.join(os.path.realpath(self.rootpath), 'gen_code.yaml')
+        if not os.path.exists(gen_code_yaml):
+            template = self.env.get_template(r'gen_code_flask.yaml')
+            saveUTF8File(gen_code_yaml,
+                         [template.render(pro_name=p_name,pro_type='flask')])
+
+class GenProject_Aiohttp(GenProject_base):
+    def __init__(self,modelfile,rootpath):
+        super().__init__(modelfile,rootpath)
+
+    def gen_code(self):
+        gen = GenSwaggerCodeFromUml(self.rootpath, self.modelfile, type='aiohttp')
+        # outfile为’‘时，采用默认的路径
+        gen.gen_swagger_code(outfile='')
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), f'{gen.swager.name}-k8s.yml'),
+                     [self.env.get_template('k8s-tmp.yml').render(root_path=os.path.split(self.rootpath)[-1],
+                                                                          swagger=gen.swager, plugins=[])],
+                     exist_ok=False)
+
+class GenProject_Flask(GenProject_base):
+    def __init__(self,modelfile,rootpath):
+        super().__init__(modelfile,rootpath)
+
+    def gen_code(self,
+                 include_swagger:bool=True,
+                 include_model:bool=True,
+                 include_test:bool = False,
+                 include_seeds:bool = False,
+                 use_uwsgi:bool = True,
+                 plugins:list = None):
+        '''
+        创建专案文件
+        :param include_swagger: False 不产生swagger file 和 swagger相关代码
+        :param include_model: False 不产生model，model_base等相关代码
+        :param include_test: False 不产生test相关代码
+        :param include_seeds: False 不产生seeds相关代码
+        :param use_uwsgi: False 不产生uwsgi相关代码
+        :param plugins: 支持redis,kafka,cassandra等,默认支持redis
+        :return:
+        '''
+        if plugins is None:
+            plugins= ['redis']
+        elif isinstance(plugins,list):
+            plugins.append('redis')
+        else:
+            plugins = ['redis',plugins]
+        # outfile为’‘时，采用默认的路径
+        gen = GenSwaggerCodeFromUml(self.rootpath, self.modelfile, type='flask')
+        if include_swagger:
+            gen.gen_swagger_code(outfile='')
+            gen.gen_swagger_ctr_code()
+        else:
+            logging.info('提示：不产生swagger file')
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath),'app','__init__.py'),
+                     [self.env.get_template('__init__.pys').render(swagger = gen.swager,
+                                                             include_model=include_model,plugins=plugins)],
+                     exist_ok=False
+                     )
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'README.md'),
+                     [self.env.get_template('README.md').render(swagger = gen.swager)],
+                     exist_ok=False
+                     )
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'Dockerfile'),
+                     [self.env.get_template('Dockerfile.tmp').render(swagger=gen.swager)],
+                     exist_ok=False)
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), '.drone.yml'),
+                     [self.env.get_template('drone.tmp').render(swagger=gen.swager)],
+                     exist_ok=False)
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'setup.py'),
+                     [self.env.get_template('setup.tmp').render(swagger=gen.swager)],
+                     exist_ok=False)
+
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'docker-compose.yaml'),
+                     [self.env.get_template('docker-compose.yaml').render(root_path=os.path.split(self.rootpath)[-1],
+                                                                swagger=gen.swager,plugins=plugins)],
+                     exist_ok=False)
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'docker-compose-dev.yaml'),
+                     [self.env.get_template('docker-compose-dev.yaml').render(root_path=os.path.split(self.rootpath)[-1],
+                                                                swagger=gen.swager,plugins=plugins)],
+                     exist_ok=False)
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'run.py'),
+                     [self.env.get_template('run.pys').render(root_path=os.path.split(self.rootpath)[-1],
+                                                                swagger=gen.swager)],
+                     exist_ok=False)
+
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'uwsgi_run.py'),
+                         [self.env.get_template('uwsgi_run.pys').render(root_path=os.path.split(self.rootpath)[-1],
+                                                                swagger=gen.swager)],
+                         exist_ok=False
+                         )
+
+        # saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'gen_code_run.py'),
+        #                  [self.env.get_template('gen_code_run.pys').render(root_path=os.path.split(self.rootpath)[-1],
+        #                                                         swagger=gen.swager)],
+        #                  exist_ok=False
+        #                  )
+
+
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath), f'{gen.swager.name}-k8s.yml'),
+                     [self.env.get_template('k8s-tmp.yml').render(root_path=os.path.split(self.rootpath)[-1],
+                                                                          swagger=gen.swager, plugins=plugins)],
+                     exist_ok=False)
+        self.save_as(os.path.join(self.sample_path, '__init__.py'),
+                     os.path.join(os.path.realpath(self.rootpath), 'app', 'api', '__init__.py'))
+        self.save_as(os.path.join(self.sample_path, '__init__.py'),
+                     os.path.join(os.path.realpath(self.rootpath), 'app', 'api','v1_0', '__init__.py'))
+        self.save_as(os.path.join(self.sample_path,'config.ini'),
+                     os.path.join(os.path.realpath(self.rootpath), 'config.ini'))
+
+        self.save_as(os.path.join(self.sample_path,'create_new_table_run.pys'),
+                     os.path.join(os.path.realpath(self.rootpath),'create_new_table_run.py'))
+
+        saveUTF8File(os.path.join(os.path.realpath(self.rootpath),'app', 'config.py'),
+                     [self.env.get_template('config.pys').render(include_model=include_model)],
+                     exist_ok=False
+                     )
+        # self.save_as(os.path.join(self.sample_path,'config-sample.ini'),
+        #              os.path.join(os.path.realpath(self.rootpath),'config-sample.ini'))
+        self.save_as(os.path.join(self.sample_path,'babel.cfg'),
+                     os.path.join(os.path.realpath(self.rootpath), 'babel.cfg'))
+        # self.save_as(os.path.join(self.sample_path,'gen_code_run.pys'),
+        #              os.path.join(os.path.realpath(self.rootpath),'gen_code_run.py'))
+        self.save_as(os.path.join(self.sample_path, 'requirements.txt'),
+                     os.path.join(os.path.realpath(self.rootpath), 'requirements.txt'))
+        self.save_as(os.path.join(self.sample_path, 'gitignore.git'),
+                     os.path.join(os.path.realpath(self.rootpath), '.gitignore'))
+        self.save_as(os.path.join(self.sample_path, 'dockerignore.dock'),
+                     os.path.join(os.path.realpath(self.rootpath), '.dockerignore'))
+        self.save_as(os.path.join(self.sample_path, 'run.sh'),
+                     os.path.join(os.path.realpath(self.rootpath), 'run.sh'))
+        self.save_as(os.path.join(self.sample_path, 'run.sh'),
+                     os.path.join(os.path.realpath(self.rootpath), 'run-dev.sh'))
+        self.save_as(os.path.join(self.sample_path, 'utils.pys'),
+                     os.path.join(os.path.realpath(self.rootpath),'app', 'utils.py'))
+        self.save_as(os.path.join(self.sample_path, 'file_utils.pys'),
+                     os.path.join(os.path.realpath(self.rootpath),'app', 'file_utils.py'))
+
+        if include_model:
+            self.save_as(os.path.join(self.sample_path, 'utils.pys'),
+                         os.path.join(os.path.realpath(self.rootpath), 'app', 'utils.py'))
+            # self.save_as(os.path.join(self.sample_path, 'migrate_run.pys'),
+            #              os.path.join(os.path.realpath(self.rootpath), 'migrate_run.py'))
+            # self.save_as(os.path.join(self.sample_path, 'migrate_run.bat'),
+            #              os.path.join(os.path.realpath(self.rootpath), 'migrate_run.bat'))
+            # self.save_as(os.path.join(self.sample_path, 'migrate_run.bat'),
+            #              os.path.join(os.path.realpath(self.rootpath), 'migrate_run.sh'))
+        if include_seeds:
+            # self.save_as(os.path.join(self.sample_path, 'config.ini'),
+            #              os.path.join(os.path.realpath(self.rootpath), 'seeds', 'config.ini'))
+            self.save_as(os.path.join(self.sample_path,'seeds', 'models_rm.pys'),
+                         os.path.join(os.path.realpath(self.rootpath),'seeds', 'models_rm.py'))
+            self.save_as(os.path.join(self.sample_path,'seeds', 'seed_dev_data.pys'),
+                         os.path.join(os.path.realpath(self.rootpath),'seeds', 'seed_dev_data.py'))
+            self.save_as(os.path.join(self.sample_path,'seeds', 'seed_init.pys'),
+                         os.path.join(os.path.realpath(self.rootpath),'seeds', 'seed_init.py'))
+            self.save_as(os.path.join(self.sample_path,'seeds', 'seed_rm.pys'),
+                         os.path.join(os.path.realpath(self.rootpath),'seeds', 'seed_rm.py'))
+            # 在seeds中不能执行
+            self.save_as(os.path.join(self.sample_path,'seeds','seed_run.pys'),
+                         os.path.join(os.path.realpath(self.rootpath), 'seed_run.py'))
+            self.save_as(os.path.join(self.sample_path,'seeds', 'seed_utils.pys'),
+                         os.path.join(os.path.realpath(self.rootpath),'seeds', 'seed_utils.py'))
+
+        if include_test:
+            gen_test_code = GenTestsCodeFromUml(self.rootpath, self.modelfile, type='flask')
+            gen_test_code.gen_tests_codes()
+            # self.save_as(os.path.join(self.sample_path, 'test__init__.pys'),
+            #              os.path.join(os.path.realpath(self.rootpath),'tests', '__init__.py'))
+            # self.save_as(os.path.join(self.sample_path, 'test_run.pys'),
+            #              os.path.join(os.path.realpath(self.rootpath), 'tests', 'run.py'))
+            # saveUTF8File(os.path.join(os.path.realpath(self.rootpath),'tests', 'test_base.py'),
+            #              [self.env.get_template('test_test_base.tmp').render(service_name=gen.swager.name)],
+            #              exist_ok=False
+            #              )
+
+        if use_uwsgi:
+            saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'default.conf'),
+                         [self.env.get_template('default.conf').render(port=80)],
+                         exist_ok=False
+                         )
+            saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'supervisord.conf'),
+                         [self.env.get_template('supervisord.conf').render(swagger=gen.swager)],
+                         exist_ok=False
+                         )
+            saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'uwsgi.ini'),
+                         [self.env.get_template('uwsgi.ini').render(swagger=gen.swager,plugins=plugins)],
+                         exist_ok=False
+                         )
+            saveUTF8File(os.path.join(os.path.realpath(self.rootpath), 'uwsgi-dev.ini'),
+                         [self.env.get_template('uwsgi.ini').render(swagger=gen.swager,plugins=plugins)],
+                         exist_ok=False
+                         )
+        try:
+            k8s = upgrade.Upgrade_k8s('.',gen.swager)
+            k8s.merge_code()
+        except Exception as e:
+            print('升级 k8s 失败,error:',str(e))

gencode/gencode/export_class2swgclass.py

@@ -0,0 +1,206 @@
+from gencode.importmdj.import_uml_models import Import_uml_models
+from gencode.uml_class_models import Class,Property
+from gencode.ext import Session
+import os
+from jinja2 import FileSystemLoader, Environment
+import codecs
+import json
+import logging
+class ExportClass2SWGClass():
+    def __init__(self,source_umlfile,dest_umlfile):
+        '''
+
+        :param source_umlfile: 源umlmodel文件
+        :param dest_umlfile: 汇入swg的umlmodel文件
+        '''
+        self.source_umlfile = source_umlfile
+        self.dest_umlfile = dest_umlfile
+        self.session = Session()
+        tmp_path = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'template')
+        load = FileSystemLoader(tmp_path)
+        self.env = Environment(loader=load)
+        self.project = Import_uml_models(self.source_umlfile).import_model()
+        self.dest_model = self._load_models(self.dest_umlfile)
+        self.swg_package = self.__get_swg_package(self.dest_model)
+        self.temp = self.env.get_template('swg_package_mng.tmp')
+
+    def _load_models(self,file):
+        with codecs.open(file, encoding='utf8') as f:
+            js = f.read()
+            return json.loads(js)
+
+    def __get_swg_package(self,dmodel):
+        for tag in dmodel.get('tags',[]):
+            if tag['name']=='swagger':
+                swg_ref_id = tag['reference']['$ref']
+                for elem in dmodel.get('ownedElements', []):
+                    if elem['_id'] == swg_ref_id:
+                        return elem
+        for elem in dmodel.get('ownedElements',[]):
+            if elem['name']=='swagger':
+                return elem
+        return {}
+
+    def __get_package_mng_code_from_temp(self, bocls):
+        '''
+        从模板中载入swagger 中的package
+        :param bocls:
+        :return:
+        '''
+        cls = bocls
+        if getattr(bocls, "assign_propertys", None):
+            cls.assign_propertys(self.session)
+        codes = self.temp.render(cls_id=cls.id,
+                                 swg_pkg_id=self.swg_package['_id'],
+                                 cls_name=cls.name.replace(' ','').lower(),
+                                 cls=cls
+                                 )
+        self.session.rollback()
+        return codes
+
+    def export(self,exclude_classes:list=None,include_classes:list=None):
+        '''
+        汇入类别到swagger，当exclude_classes,include_classes同时为None时，汇入所有的类到到swagger
+        :param exclude_classes:不用汇出成swagger class的类名
+        :param include_classes:只需要汇入的swagger class的类名
+        :return:
+        '''
+        if exclude_classes is None :
+            exclude_classes = []
+        exclude_classes = [ecls.lower() for ecls in exclude_classes or []]
+        include_classes = [ecls.lower() for ecls in include_classes or []]
+        classes = self.session.query(Class). \
+            filter(Class.isswagger == False). \
+            filter(Class.type == 'UMLClass'). \
+            filter(Class.projectid == self.project.id).all()
+        classes_swg = [elm['name'] for elm in self.swg_package.get('ownedElements', []) if elm['_type'] == 'UMLClass']
+        packages_swg = [elm['name'] for elm in self.swg_package.get('ownedElements', []) if elm['_type'] == 'UMLPackage']
+        for cls in classes :
+            class_name = cls.name.replace(' ','').lower()
+            # 已产生swagger class
+            if class_name in classes_swg or  '%smng'%class_name in packages_swg :
+                continue
+            # 有排除表时，不能产生
+            if exclude_classes and  class_name in exclude_classes:
+                continue
+            # 有包含表时，只产生包含表的类别
+            if include_classes and class_name not in include_classes:
+                continue
+            codes = self.__get_package_mng_code_from_temp(cls)
+            self.swg_package.setdefault('ownedElements',[]).append(json.loads(codes,strict=False))
+        with codecs.open(self.dest_umlfile,mode='w',encoding='utf-8') as file:
+            file.write(json.dumps(self.dest_model,ensure_ascii=False))
+        # 需要删除吗？？？
+        # self.session.query(Project).filter(Project.id==self.project.id).delete()
+        logging.info('export swagger classes success.')
+
+    def __find_element(self,elemnts:list,key:str,value):
+        '''
+        查找某个元素
+        :param elemnts:
+        :param key:
+        :param value:
+        :return: obj
+        '''
+        for elem in elemnts:
+            if elem[key]==value:
+                return elem
+        return None
+
+    def export_one_swgclass(self,bocls:str):
+        '''
+        把bocls汇入到swagger，bocls不存在时会创建
+        :param bocls:
+        :return:
+        '''
+        cls = self.session.query(Class). \
+            filter(Class.isswagger == False). \
+            filter(Class.type == 'UMLClass'). \
+            filter(Class.projectid == self.project.id,Class.name==bocls).first()
+        if not cls :
+            import uuid
+            cls = Class(id=str(uuid.uuid4()),name=bocls)
+            cls.propertys = [Property(id=str(uuid.uuid4()),name='id',type='integer')]
+        # 获取swg中该类的package
+        package_cls_swg = self.__find_element(self.swg_package['ownedElements'], 'name', '%smng' % bocls)
+        # 如果swagger中没有，则把模板中的package加入swagger中
+        if not package_cls_swg:
+            package_cls_tmp = self.__get_package_mng_code_from_temp(cls)
+            self.swg_package.setdefault('ownedElements', []).append(json.loads(package_cls_tmp, strict=False))
+            with codecs.open(self.dest_umlfile, mode='w', encoding='utf-8') as file:
+                file.write(json.dumps(self.dest_model, ensure_ascii=False))
+            logging.debug('The class (%smng) has insert' % bocls)
+        else:
+            logging.debug('The class (%smng) has exist'%bocls)
+
+    def add_operation(self,swgpackage_name:str,opname:str,method:str='get'):
+        '''
+        在swagger类上增加一个方法
+        :param swgpackage_name: swagger package 名称，如：companymng
+        :param opname: companys
+        :param method: http method，如：get,put,post,delete等
+        :return:
+        '''
+        if method not in ['get','put','post','delete']:
+            logging.debug("add_operation() error, the method %s not in ['get','put','post','delete']"%method)
+        # 获取swg中该类的package
+        package_cls_swg = self.__find_element(self.swg_package['ownedElements'], 'name',swgpackage_name)
+        if not package_cls_swg:
+            logging.warning('the swagger package(%s) is not exist'%swgpackage_name)
+            return
+        obj_signal = self.__find_element(package_cls_swg['ownedElements'],'_type','UMLSignal')
+        assert obj_signal,'the swagger package(%s) has no UMLSignal class'%swgpackage_name
+        obj_cls = self.__find_element(package_cls_swg['ownedElements'],'_type','UMLClass')
+        if not obj_cls:
+            obj_cls = self.__find_element(package_cls_swg['ownedElements'], '_type', 'UMLDataType')
+        assert obj_cls, 'the swagger package(%s) has no UMLClass class' % swgpackage_name
+        # 如果方法名存在则不能产生,
+        op= self.__find_element(obj_signal['operations'],'name',opname)
+        if op:
+            logging.error('the swagger package(%s) has a opname（%s） UMLSignal class，请修改opname再试'%(swgpackage_name,opname))
+            return
+        # 从模板中拷贝 物件
+        import uuid
+        # 使用临时的类别，保证临时物件中的id是唯一的，可以直接加入的swagger中
+        cls = Class(id=str(uuid.uuid4()), name=opname)
+        package_cls_tmp = json.loads(self.__get_package_mng_code_from_temp(cls), strict=False)
+        obj_signal_tmp = self.__find_element(package_cls_tmp['ownedElements'], '_type', 'UMLSignal')
+        op_tmp = self.__find_element(obj_signal_tmp['operations'], 'stereotype', method)
+        # 与swagger建立关系
+        op_tmp['_parent']['$ref']=obj_signal['_id']
+        op_tmp['name'] = opname
+        obj_signal['operations'].append(op_tmp)
+        # 找到 in，return类
+        op_parm_in = self.__find_element(op_tmp['parameters'],'name','in')
+        obj_in = self.__find_element(package_cls_tmp['ownedElements'], '_id', op_parm_in['type']['$ref'])
+        obj_in['_parent']['$ref']=package_cls_swg['_id']
+        package_cls_swg['ownedElements'].append(obj_in)
+        op_parm_return = self.__find_element(op_tmp['parameters'],'direction','return')
+        obj_return = self.__find_element(package_cls_tmp['ownedElements'], '_id', op_parm_return['type']['$ref'])
+        obj_return['_parent']['$ref']=package_cls_swg['_id']
+        # 返回类型 用 obj_cls
+        default_value = self.__find_element(obj_return['attributes'],'name','default')
+        default_value['type']['$ref'] = obj_cls['_id']
+        package_cls_swg['ownedElements'].append(obj_return)
+        # 把in return param obj 加入到图上
+        diagram_tmp= self.__find_element(package_cls_tmp['ownedElements'], '_type', 'UMLClassDiagram')
+        diagram_main = self.__find_element(package_cls_swg['ownedElements'], '_type', 'UMLClassDiagram')
+        op_parm_in_view = self.__find_element(diagram_tmp['ownedViews'], 'model', {"$ref": obj_in['_id']})
+        op_parm_in_view['_parent']['$ref']=diagram_main['_id']
+        op_parm_in_view['left'] += 380
+        diagram_main['ownedViews'].append(op_parm_in_view)
+        op_parm_return_view = self.__find_element(diagram_tmp['ownedViews'], 'model', {"$ref": obj_return['_id']})
+        op_parm_return_view['_parent']['$ref']=diagram_main['_id']
+        op_parm_return_view['left'] += 380
+        diagram_main['ownedViews'].append(op_parm_return_view)
+        with codecs.open(self.dest_umlfile, mode='w', encoding='utf-8') as file:
+            file.write(json.dumps(self.dest_model, ensure_ascii=False))
+        logging.info("add_operation(%s,%s,%s) success" % (swgpackage_name,opname,method))
+
+
+if __name__ == '__main__':
+    exp = ExportClass2SWGClass(r'D:\mwwork\projects\mwgencode\order_system\docs\temp.json',
+                         r'D:\mwwork\projects\mwgencode\order_system\docs\temp.json')
+    # exp.export()
+    exp.export_one_swgclass('company')
+    exp.add_operation('companymng','company2','put')