#!/usr/bin/env python

# -*- coding: cp936 -*-

'''

功能说明：

实现从supermap后台的mssql里点数据集到AutoCAD里文本型的转换。

要是改为多线程的效果可能会更好。但现在的处理量很小，用不着给自己找麻烦

2011-10-4~2011-10-5

TODO:程度里的try...except太多，有些可以改成with语句。

'''

__author__ = 'shifeng'

import pyodbc

import logging

import comtypes.client

import numbers

import array

import re

import ConfigParser

def getconnect(db_type=None,db_arg=None):

'''

sql server:

connector=pyodbc.connect('DRIVER={SQL SERVER};SERVER=LUOBO\SQLEXPRESS;DATABASE=chengguuo')

'''

if db_type in ('MSSQL','JET'):

try:

connector = pyodbc.connect(db_arg)

except :

logging.error("数据库连接错误！")

raise

return connector

def getdrawscript(f=None):

'''

如果给定文件名则打开该CAD文件。

如果没有给定文件名则认为待编辑的CAD文件已经打开。

'''

if not f:

try:

cad=comtypes.client.GetActiveObject('AutoCAD.Application') #这个可能有点问题

cad.Visible=False

drawingscript=cad.ActiveDocument

except :

logging.error("获得活动cad文件失败。请查看草稿文件是否已经打开，或请关闭除了草稿文件以外的所有cad文件。")

raise

else:

try:

cad=comtypes.client.CreateObject('AutoCAD.Application')

except :

logging.error("请确定本机上安装了autocad")

raise

try:

drawingscript=cad.Documents.Open(f)

except :

logging.error("cad文件打开失败。")

cad.Quit()

raise

return drawingscript

def getrows(sqlstring,connect=None,*args,**kargs):

'''

当返回大量记录时可以要占用大量内存空间

>>> sqlstring = 'SELECT SmX,SmY,name FROM SMDTV_93'

>>> connect = pyodbc.connect('DRIVER={SQL SERVER};SERVER=LUOBO\SQLEXPRESS;DATABASE=chengguuo')

>>> rows = getrows(sqlstring,connect)

>>> isinstance(rows,list)

True

>>> row=rows[1]

>>> isinstance(row[0],numbers.Real)

True

>>> isinstance(row[1],numbers.Real)

True

>>> isinstance(row[2],str)

True

'''

cursor=connect.cursor()

try:

cursor.execute(sqlstring)

rows = cursor.fetchall()

except :

logging.error("数据库进行sql查寻时出错")

raise

finally:

connect.close()

return rows # 返回引用，注意防止修改

def drawtext(x=None,y=None,text='什么都没有',draw=None,size=50):

'''

在cad里输入文字。

>>> drawtext(x=36395514.9324,y=3250445.33498,text=r'A1;T{H0.7x;S^3;}x',draw=getdrawscript('D:\Drawing11.dwg')) == 0

True

'''

if isinstance(x,numbers.Real) and isinstance(y,numbers.Real) and text: #无文字则不绘

point=array.array('d',[x,y,0])

try:

ms=draw.ModelSpace #draw必须没有关闭

#draw.ModelSpace.AddText(text,point,size) #单行文本

ms.AddMText(point,size,text) #多行文本

except:

logging.error("绘制文字出错！")

logging.error("nx:%sny:%sntext:%snpoint:%sndraw:%snms:s"%(x,y,text,point,draw,ms))

draw.Application.Quit()

raise

return 0

return 1

def removesharp(text):

'''

去掉text中的表示supermap标签专题图上下标的特殊符号

>>> removesharp("T#-3#=x")

'T3x'

>>> removesharp("J#-2#=S#+2")

'J2S2'

>>> removesharp("")

''

>>> removesharp(None)

''

'''

if text:

sharppattern=re.compile(r'#[+-=]')

modifiedtext=sharppattern.sub(r"",text,count=0)

else:

modifiedtext=""

return modifiedtext

def addchinesenote(test):

pass

def sharp2mtext(text):

'''

>>> text1 = 'T#-3#=x'

>>> text2 = 'J#-2#=S#+2'

>>> text3 = None

>>> sharp2mtext(text1) == '\A1;T{H0.7x;\\S^3;}x'

True

>>> #注意这是用的是repr()函数输出。

>>> print sharp2mtext(text1)

A1;T{H0.7x;\S^3;}x

>>> sharp2m

2000

text(text2) == '\A1;J{H0.7x;\S^2;}S{H0.7x;\S2^;}'

True

>>> sharp2mtext(text3) == ''

True

'''

if not text:

return ''

else:

ws=text.split('#')

logging.info(ws)

ws_conved=[]

for w in ws[1:]:

w_tem=''

if w[0] == '+':

w_tem = r'{H0.7x;S' + w[1:] + r'^;}'

elif w[0] == '-':

w_tem = r'{H0.7x;S^' + w[1:] + r';}'

elif w[0] == '=':

w_tem = w[1:]

ws_conved.append(w_tem)

text_conved = r'A1;'+ws[0]+''.join(ws_conved)

return text_conved

def main(*args,**kargs):

'''

>>> #main(filename="D:\Drawing1.dwg",sqlstring="SELECT SmX,SmY,name FROM SMDTV_93")

0

>>> # main(filename="D:\Drawing2.dwg",sqlstring="SELECT SmX,SmY,name FROM SMDTV_93") 失败，由于Drawing2.dwg还不存在

>>> # 0

>>> config = ConfigParser.ConfigParser()

>>> config.read('dot2text.cfg')

['dot2text.cfg']

>>> connectstring = config.get('APPLICATION','connectstring')

>>> connectstring='DRIVER={SQL SERVER};SERVER=LUOBO\SQLEXPRESS;DATABASE=chengguuo'

>>> filename = config.get('APPLICATION','outputfile')

>>> sqlstring = config.get('APPLICATION','sqlstring')

>>> dbtype =config.get('APPLICATION','dbtype')

>>> main(filename=filename,sqlstring=sqlstring,dbtype=dbtype,connectstring=connectstring)

0

'''

try:

filename=kargs['filename']

sqlstring=kargs['sqlstring']

dbtype=kargs['dbtype']

connectstring=kargs['connectstring']

except:

logging.info("请给出正确的关键字参数")

return 1

if not filename:

draw=getdrawscript()

else:

draw=getdrawscript(filename)

try:

layer=draw.Layers.Add("DOT_2_TEXT") #新增图层

draw.ActiveLayer=layer

except :

logging.error("添加新图层出错")

draw.Application.Quit()

raise

# 这里出过大bug，主要是异常的处理

#try:

# dbconnect=getconnect(db_type=dbtype,db_arg=connectstring)

#except:

# raise

#finally:

# draw.Application.Quit()

try:

dbconnect=getconnect(db_type=dbtype,db_arg=connectstring)

except:

draw.Application.Quit()

raise

try:

rows=getrows(sqlstring,dbconnect)

except:

return 1

try:

for row in rows:

#drawtext(x=row[0],y=row[1],text=removesharp(row[2]),draw=draw)

drawtext(x=row[0],y=row[1],text=sharp2mtext(row[2]),draw=draw)

draw.save()

except:

logging.error("主函数中的drawtext操作出错")

logging.error("drawtext:nrow[0]:%snrow[1]:%snrow[2]:%sn"%(row[0],row[1],row[2]))

return 1

draw.Application.Quit()

return 0

if __name__ == "__main__":

#filename = raw_input("请输入文件名n") # D:Drawing1.dwg

#sqlstring = raw_input("***请不要误操作！***n请输入查询语句。n") # select SmX,SmY,name from SMDTV93

config = ConfigParser.ConfigParser()

config.read('dot2text.cfg')

connectstring = config.get('APPLICATION','connectstring')

filename = config.get('APPLICATION','outputfile')

sqlstring = config.get('APPLICATION','sqlstring')

dbtype =config.get('APPLICATION','dbtype')

main(filename=filename,sqlstring=sqlstring,dbtype=dbtype,connectstring=connectstring)

[APPLICATION]

connectstring =DRIVER={SQL SERVER};SERVER=LUOBOSQLEXPRESS;DATABASE=chengguo

outputfile =D:Drawing1.dwg

dbtype =MSSQL

sqlstring =SELECT SmX,SmY,name from SMDTV_93