table_info.py 2.0 KB
#author:        4N
#createtime:    2021/1/27
#email:         nheweijun@sina.com

from ..models import Table,Columns,DES


from app.util.component.ApiTemplate import ApiTemplate
from app.util.component.ModelVisitor import ModelVisitor
from app.util.component.PGUtil import PGUtil
from osgeo.ogr import DataSource,Layer
from osgeo.osr import SpatialReference
class Api(ApiTemplate):
    api_name = "表信息"
    def process(self):
        res = {}
        try:
            table_guid = self.para.get("guid")
            table = Table.query.filter_by(guid=table_guid).one_or_none()
            if not table:
                raise Exception("数据不存在!")
            pg_ds = PGUtil.open_pg_data_source(0,DES.decode(table.relate_database.sqlalchemy_uri))
            layer:Layer = pg_ds.GetLayerByName(table.name)
            append_dict ={"srid":None,"sr_wkt":None,"sr_proj4":None,"exist":1}
            if layer:
                sr:SpatialReference = layer.GetSpatialRef()
                if sr:
                    append_dict["srid"] = sr.GetAuthorityCode(None)
                    append_dict["sr_wkt"] = sr.ExportToWkt()
                    append_dict["sr_proj4"] = sr.ExportToProj4()
            else:
                append_dict["exist"]=0
            columns = table.relate_columns.order_by(Columns.name).all()
            res["data"]=ModelVisitor.table_to_json(table)
            res["data"]["columns"] = ModelVisitor.objects_to_jsonarray(columns)
            res["data"].update(append_dict)
            res["result"] = True
    
        except Exception as e:
            raise e
        return res
    
    api_doc={
    "tags":["管理接口"],
    "parameters":[
        {"name": "guid",
         "in": "formData",
         "type": "string",
         "description": "表guid","required":"true"},
    ],
    "responses":{
        200:{
            "schema":{
                "properties":{
                }
            }
            }
        }
    }