mwgencode 1.4.1__py3-none-any.whl

gencode/importxmi/import_states.py

@@ -0,0 +1,150 @@
+try:  # 导入模块
+    import xml.etree.cElementTree as ET
+except ImportError:
+    import xml.etree.ElementTree as ET
+
+uml="http://schema.omg.org/spec/UML/2.0"
+xmi="{http://schema.omg.org/spec/XMI/2.1}"
+_id = xmi+"id"
+_type = xmi+"type"
+_extension = xmi+'Extension'
+
+class ImportStates():
+    def __init__(self):
+        self._objects = {}
+        self.states = []
+        self.trans = []
+
+    def _getDoc(self,attr):
+        extens = attr.find(_extension)
+        if extens is not None:
+            # print('     extens:',extens.attrib)
+            cls_doc = extens.find('documentation')
+            if cls_doc is not None:
+                return cls_doc.attrib.get('value')
+        return ''
+
+    def _handle_subv(self, subv, state):
+        # print('subv:',subv.attrib)
+        # if subv.attrib.get(_type,None) != 'uml:State':
+        #     return None
+        region = subv.find('region', None)
+        if region:
+            self._handle_region(region, subv.attrib.get('name') + '_')
+        def getEnter(subv):
+            enters = subv.findall('entry')
+            result = []
+            for enter in enters:
+                result.append({'name':enter.attrib.get('name'),'doc':self._getDoc(enter)})
+            return result
+
+        def getExit(subv):
+            enters = subv.findall('exit')
+            result = []
+            for enter in enters:
+                result.append({'name': enter.attrib.get('name'), 'doc': self._getDoc(enter)})
+            return result
+        return {'type':'state', 'id':subv.attrib.get(_id),
+                            'name':subv.attrib.get('name'),'doc':self._getDoc(subv),
+                            'onenter':getEnter(subv),'onexit':getExit(subv)}
+
+    def _handle_tran(self, tran):
+        def getTriger(tran):
+            trgs = tran.findall('trigger')
+            for trg in trgs:
+                if trg.attrib.get(_type)=='uml:Trigger':
+                    if trg.attrib.get('name') != '':
+                        return trg.attrib.get('name')
+            # 如果没有trigger则直接返回tran的name
+            result = tran.attrib.get('name').split('(')[0]
+            if result:
+                return result
+
+        def getGuard(tran):
+            guard =  tran.find('guard')
+            if guard is not None:
+                result = guard.attrib.get('specification')
+                if result is not None:
+                    return result
+            return ''
+
+        def getBefore(tran):
+            result  =  tran.attrib.get('name',None)
+            if result:
+                return result
+            return ''
+
+        def getEffect(tran):
+            effe = tran.find('effect')
+            # print(tran.attrib)
+            if effe is not None:
+                result = effe.attrib.get('name')
+                if result is not None:
+                    return result
+            return ''
+        # print(tran.attrib)
+        try:
+            return {'trigger': getTriger(tran),
+                    'doc': self._getDoc(tran),
+                    'source': self._objects[tran.attrib.get('source')]['name'],
+                    'target': self._objects[tran.attrib.get('target')]['name'],
+                    'guard': getGuard(tran),
+                    'before':getBefore(tran),
+                    'after': getEffect(tran)}
+        except Exception as e:
+            # print('--------------',tran.attrib)
+            # print('--------------',e)
+            return None
+
+            # print('tran:',tran.attrib)
+
+    def _handle_region(self, region, state=''):
+        state_d = {}
+        for subv in region.findall('subvertex'):
+            if subv.attrib.get(_type, None) != 'uml:State':
+                continue
+            state_o = self._handle_subv(subv, state)
+            if state_o is not None:
+                self._objects[state_o['id']] = state_o
+                state_d.setdefault(state + 'states',[]).append(state_o)
+        if state_d:
+            self.states.append(state_d)
+
+        for tran in region.findall('transition'):
+            if tran.attrib.get(_type) != 'uml:Transition':
+                continue
+            tran_o = self._handle_tran(tran)
+            if tran_o:
+                self.trans.append(tran_o)
+
+
+    def _handle_pkgElm(self, pkgElms):
+        for pkg in pkgElms:
+            if pkg.attrib.get(_type, None) != 'uml:StateMachine':
+                continue
+            region =  pkg.find('region',None)
+            if region is None :
+                continue
+            self._handle_region(region)
+
+    def impxml(self,file):
+        self._objects = {}
+        self.states = []
+        self.trans = []
+        tree = ET.parse(file)  # 分析XML文件
+        root = tree.getroot()
+        # print('0', root)
+        for idx0,r in enumerate(root):
+            # print('%d'%idx0, r.attrib)
+            if r.attrib.get('name', None) != 'RootModel':
+                continue
+            self._handle_pkgElm(r.findall('packagedElement'))
+
+
+if __name__ == '__main__':
+    i = ImportStates()
+    i.impxml(r"D:\mwwork\projects\iparking\docs-parking\parking-state.xml")
+    # print(i._objects)
+    # print(i.states)
+    print(i.trans)
+    # i.impxml(r"test.xml")

gencode/importxmi/import_swagger.py

@@ -0,0 +1,151 @@
+try:  # 导入模块
+    import xml.etree.cElementTree as ET
+except ImportError:
+    import xml.etree.ElementTree as ET
+
+from mwutils.utils import none2default
+from gencode import (ImportBase,uml,xmi,_id,_type,_extension
+             )
+class ImportSwagger(ImportBase):
+    def __init__(self):
+        self.swagger= {'title':'swagger','version':'v1.0','doc':'please add swagger package to uml'}
+        self.packages = []
+        self.objects_bo = {}
+        self.datatypes = []
+        self.singals = []
+        self.pts = []
+        self.classes = []
+
+    def _handle_swagger(self, pkg_swg):
+        # 记录swagger package 信息
+        self.swagger['title'], self.swagger['version'],*_ = \
+           none2default(self.get_stereotype(pkg_swg), ['swagger', 'v1.0'])
+        self.swagger['resp_xml'] = 'xml' in self.get_stereotype(pkg_swg)
+        self.swagger['doc'] = self.get_doc(pkg_swg)
+        pkgs = pkg_swg.findall('packagedElement')
+        def set_lk_flag(op):
+            '''
+            给lk关联的datatype打上lk的标志，便于产生lk的datatype
+            :param id:
+            :return:
+            '''
+            def set_flg(obj,type):
+                for attr in obj['attrs']:
+                    if attr['type'] == 'object':
+                        obj_attr = self.objects_bo[attr['typename']]
+                        if obj_attr['type'] not in ('uml:DataType', 'uml:Class'):
+                            raise Exception('[%s.%s]的类型必须为：DataType 或Class'
+                                             % (obj['name'],attr['name']))
+                        # 返回的类型只能允许result和return的attr才要做lookup
+                        if type=='return' and attr['name'] not in ('result','return'):
+                            continue
+                        obj_attr['haslkobj'] = True
+                        for dt in self.datatypes:
+                            if dt['id']==attr['typename']:
+                                dt['haslkobj'] = True
+                                break;
+
+            try:
+                in_obj = self.objects_bo[op['in']['typename']]
+                set_flg(in_obj,'in')
+            except Exception as e:
+                raise Exception('[%s.%s.%s] 发生错误， error:%s'%(op['pkg_name'],op['name'],in_obj['name'],e))
+            try:
+                return_obj = self.objects_bo[op['return']['typename']]
+                set_flg(return_obj,'return')
+            except Exception as e:
+                raise Exception('[%s.%s.%s] 发生错误， error:%s' % (op['pkg_name'],op['name'], return_obj['name'], e))
+
+        def add_datatype(pkg_dt):
+            if pkg_dt.attrib.get(_type, None) == 'uml:DataType':
+                datatype = self.get_bo_class(pkg_dt)
+                datatype['haslkobj'] = self.objects_bo[pkg_dt.attrib.get(_id)].get('haslkobj', False)
+                self.datatypes.append(datatype)
+            elif pkg_dt.attrib.get(_type, None) == 'uml:Signal':
+                signal = self.get_bo_class(pkg_dt)
+                signal['pkg_name'] = pkg.attrib.get('name')
+                ops_lk = []
+                for op in signal['ops']:
+                    op['pkg_name'] = signal['pkg_name']
+                    if 'lk' in op['stereotype']:
+                        op_cp = op.copy()
+                        op['stereotype'].remove('lk')
+                        op_cp['stereotype'] = ['lk']
+                        op_cp['name']='%s_%s'%(op['name'],'lk')
+                        ops_lk.append(op_cp)
+                        # 给op关联的物件设为lk
+                        set_lk_flag(op)
+                signal['ops'].extend(ops_lk)
+                self.singals.append(signal)
+            elif pkg_dt.attrib.get(_type, None) == 'uml:PrimitiveType':
+                # self.objects_bo['AAAAAAFeCiNYUToFjYA=']
+                self.pts.append(self.get_bo_class(pkg_dt))
+            elif pkg_dt.attrib.get(_type, None) == 'uml:Class':
+                datatype = self.get_bo_class(pkg_dt)
+                datatype['haslkobj'] = self.objects_bo[pkg_dt.attrib.get(_id)].get('haslkobj',False)
+                self.datatypes.append(datatype)
+                # self.pts.append(self.get_bo_class(pkg_dt))
+        # def add_datatype_from_pts():
+        #     '''
+        #     把不在swagger下的class加到datatypes中，以便能正确的产生swagger类
+        #     :return:
+        #     '''
+        #     for pt in self.pts:
+        #         for attr in pt.get('attrs',[]):
+        #             if attr['type']=='object':
+        #                 ref_class = self.objects_bo[attr['typename']]
+        #                 for datatype in self.datatypes:
+        #                     if datatype['id']==ref_class['id']:
+        #                         break
+        #                 else:
+        #                     self.datatypes.append(ref_class)
+        #
+
+
+        # swagger下的每个package是一个tag
+        for pkg in pkgs:
+            package = {'name':pkg.attrib.get('name'), 'doc':self.get_doc(pkg)}
+            # 只允许容器类才能加入
+            if pkg.attrib.get(_type, None) in['uml:Package','uml:Model','uml:Component']:
+                self.packages.append(package)
+            else:
+                add_datatype(pkg)
+            pkgs_datatype = pkg.findall('packagedElement')
+            for pkg_dt in pkgs_datatype:
+                add_datatype(pkg_dt)
+        # 增加不在swagger package中的类
+        # add_datatype_from_pts()
+
+    def _handle_pkgElm(self, pkgElms):
+        for pkg in pkgElms:
+            # 记录swagger信息
+            if pkg.attrib.get('name', None) != 'swagger' :
+                continue
+            self.swagger['title'],self.swagger['version'],*_ = \
+                       self.get_stereotype(pkg)
+            self.swagger['doc'] = self.get_doc(pkg)
+            self._handle_swagger(pkg)
+
+    def impxml(self,file):
+        tree = ET.parse(file)  # 分析XML文件
+        root = tree.getroot()
+        for idx0,r in enumerate(root):
+            if r.attrib.get('name', None) != 'RootModel':
+                continue
+            # 加入所有的数据类型的class
+            for bo_cls in self.find_bo_classes(r):
+                self.objects_bo[bo_cls['id']]=(self.get_bo_class(bo_cls['obj']))
+            # 处理所有package
+            self._handle_pkgElm(r.findall('packagedElement'))
+
+if __name__ == '__main__':
+    i = ImportSwagger()
+    i.impxml(r"D:\mwwork\projects\its\mobile_gateway_server\docs\uml analyse.xml")
+    # print(i.trans)
+    # print(i.objects_pkg)
+    print('swagger',i.swagger)
+    print('datatypes',i.datatypes)
+    print('singals',i.singals)
+    print('p', i.pts)
+    # i.impxml(r"test.xml")
+

gencode/swg2_class_models.py

@@ -0,0 +1,179 @@
+########################################
+# create by :cxh-pc
+# create time :2018-04-26 20:44:57.895581
+########################################
+from sqlalchemy import Table,Column, Integer, String, ForeignKey, Boolean,\
+                DateTime,Time,Date,Float
+from sqlalchemy.orm import relationship,aliased
+from sqlalchemy import or_
+from .ext import Base as Model
+import json
+from enum import Enum
+class Swagger2(Model):
+    __tablename__ = 'swagger2'
+    name = Column(String(50), unique= True)
+    host = Column(String(50))
+    version = Column(String(50))
+    id = Column(String(50), primary_key=True)
+    resp_xml = Column(Boolean)
+    resp_html = Column(Boolean)
+    doc = Column(String(50))
+    auth = Column(String(50), default= "jwt")
+    tags = relationship("Tag2", back_populates="swagger", foreign_keys="Tag2.swaggerid", cascade="all, delete-orphan")
+    defines = relationship("Define2", back_populates="swagger", foreign_keys="Define2.swaggerid", cascade="all, delete-orphan")
+    def __repr__(self):
+        return json.dumps(self.to_json())
+    def to_json(self):
+        return {key: getattr(self, key) for key in self.__table__.columns.keys()
+                   if hasattr(self,key)
+               }
+class Tag2(Model):
+    __tablename__ = 'tag2'
+    name = Column(String(50), unique= True)
+    doc = Column(String(50))
+    id = Column(String(50), primary_key=True)
+    swaggerid = Column(String(50),ForeignKey("swagger2.id"), nullable= False)
+    swagger = relationship("Swagger2", back_populates="tags", foreign_keys="Tag2.swaggerid")
+    paths = relationship("Path2", back_populates="tag", foreign_keys="Path2.tagid", cascade="all, delete-orphan")
+    def call_path(self):
+        return self.name
+    def __repr__(self):
+        return json.dumps(self.to_json())
+    def to_json(self):
+        return {key: getattr(self, key) for key in self.__table__.columns.keys()
+                   if hasattr(self,key)
+               }
+class Action2(Model):
+    """
+    把path归于tag下，更易于产生代码
+    """
+    __tablename__ = 'action2'
+    action = Column(String(50))
+    summary = Column(String(50))
+    doc = Column(String(50))
+    ver_str = Column(String(50))
+    has_formdata = Column(Boolean)
+    id = Column(String(50), primary_key=True)
+    is_lk = Column(Boolean, default= False)
+    # stereotype 包含 lk ，如：get/lk
+    include_lk = Column(Boolean, default= False)
+    # 有page过滤条件
+    has_page = Column(Boolean, default= False)
+    in_param_name = Column(String(50))
+    return_param_name = Column(String(50))
+    # 需要认证
+    is_auth = Column(Boolean, default= False)
+    auth_type = Column(String(50))
+    pathid = Column(Integer,ForeignKey("path2.id"), nullable= False)
+    default_respid = Column(Integer,ForeignKey("resp2.id"))
+    path = relationship("Path2", back_populates="actions", foreign_keys="Action2.pathid")
+    default_resp = relationship("Resp2", foreign_keys="Action2.default_respid")
+    params = relationship("Param2", back_populates="action", foreign_keys="Param2.actionid", cascade="all, delete-orphan")
+    resps = relationship("Resp2", back_populates="action", foreign_keys="Resp2.actionid", cascade="all, delete-orphan")
+    def call_path(self):
+        return '%s.%s'%(self.path.call_path(),self.action)
+    def __repr__(self):
+        return json.dumps(self.to_json())
+    def to_json(self):
+        return {key: getattr(self, key) for key in self.__table__.columns.keys()
+                   if hasattr(self,key)
+               }
+class Param2(Model):
+    __tablename__ = 'param2'
+    in_ = Column(String(50))
+    name = Column(String(50))
+    desc = Column(String(50))
+    required = Column(Boolean)
+    is_array = Column(Boolean)
+    type_attr = Column(String(50))
+    type = Column(String(50))
+    format = Column(String(50))
+    ref_name = Column(String(50))
+    id = Column(Integer, primary_key=True)
+    is_path = Column(Boolean, default= False)
+    is_header = Column(Boolean, default= False)
+    is_body = Column(Boolean, default= False)
+    is_query = Column(Boolean, default= False)
+    is_formdata = Column(Boolean, default= False)
+    actionid = Column(String(50),ForeignKey("action2.id"), nullable= False)
+    action = relationship("Action2", back_populates="params", foreign_keys="Param2.actionid")
+    def call_path(self):
+        return '%s.%s'%(self.action.call_path(),self.name)
+    def __repr__(self):
+        return json.dumps(self.to_json())
+    def to_json(self):
+        return {key: getattr(self, key) for key in self.__table__.columns.keys()
+                   if hasattr(self,key)
+               }
+class Resp2(Model):
+    __tablename__ = 'resp2'
+    code = Column(String(50))
+    desc = Column(String(50))
+    type_attr = Column(String(50))
+    is_array = Column(Boolean)
+    ref_name = Column(String(50), default= '')
+    type = Column(String(50))
+    id = Column(Integer, primary_key=True)
+    format = Column(String(50))
+    # 记录引用物件的类型
+    umlcls_type = Column(String(50))
+    name = Column(String(50), default= '')
+    actionid = Column(String(50),ForeignKey("action2.id"), nullable= False)
+    action = relationship("Action2", back_populates="resps", foreign_keys="Resp2.actionid")
+    def call_path(self):
+        return '%s.%s'%(self.action.call_path(),self.name)
+    def __repr__(self):
+        return json.dumps(self.to_json())
+    def to_json(self):
+        return {key: getattr(self, key) for key in self.__table__.columns.keys()
+                   if hasattr(self,key)
+               }
+class Define2(Model):
+    __tablename__ = 'define2'
+    name = Column(String(50), unique= True)
+    desc = Column(String(50))
+    id = Column(String(50), primary_key=True)
+    is_lk = Column(Boolean, default= False)
+    swaggerid = Column(String(50),ForeignKey("swagger2.id"), nullable= False)
+    swagger = relationship("Swagger2", back_populates="defines", foreign_keys="Define2.swaggerid")
+    attrs = relationship("Attr2", back_populates="define", foreign_keys="Attr2.defineid", cascade="all, delete-orphan")
+    def __repr__(self):
+        return json.dumps(self.to_json())
+    def to_json(self):
+        return {key: getattr(self, key) for key in self.__table__.columns.keys()
+                   if hasattr(self,key)
+               }
+class Attr2(Model):
+    __tablename__ = 'attr2'
+    name = Column(String(50))
+    is_array = Column(Boolean)
+    desc = Column(String(50))
+    type_attr = Column(String(50))
+    type = Column(String(50))
+    format = Column(String(50))
+    id = Column(Integer, primary_key=True)
+    ref_name = Column(String(50))
+    defineid = Column(String(50),ForeignKey("define2.id"), nullable= False)
+    define = relationship("Define2", back_populates="attrs", foreign_keys="Attr2.defineid")
+    def __repr__(self):
+        return json.dumps(self.to_json())
+    def to_json(self):
+        return {key: getattr(self, key) for key in self.__table__.columns.keys()
+                   if hasattr(self,key)
+               }
+class Path2(Model):
+    __tablename__ = 'path2'
+    name = Column(String(50))
+    id = Column(Integer, primary_key=True)
+    is_lk = Column(Boolean, default= False)
+    tagid = Column(String(50),ForeignKey("tag2.id"), nullable= False)
+    tag = relationship("Tag2", back_populates="paths", foreign_keys="Path2.tagid")
+    actions = relationship("Action2", back_populates="path", foreign_keys="Action2.pathid", cascade="all, delete-orphan")
+    def call_path(self):
+        return '%s.%s'%(self.tag.call_path(),self.name)
+    def __repr__(self):
+        return json.dumps(self.to_json())
+    def to_json(self):
+        return {key: getattr(self, key) for key in self.__table__.columns.keys()
+                   if hasattr(self,key)
+               }