首页 > 代码库 > 使用Python生成源文件的两种方法
使用Python生成源文件的两种方法
利用Python的字符串处理模块,开发人员可以编写脚本用来生成那些格式相同的C、C++、JAVA源程序、头文件和测试文件,从而避免大量的重复工作。本文概述两种利用Python string类生成java源代码的方法。
1,String Template
Template是一个好东西,可以将字符串的格式固定下来,重复利用。Template也可以让开发人员可以分别考虑字符串的格式和其内容了,无形中减轻了开发人员的压力。Template属于string中的一个类,有两个重要的方法:substitute和safe_substitute。替换时使用substitute(),若未能提供模板所需的全部参数值时会发生异常。使用safe_substitute() 则会替换存在的字典值,保留未存在的替换符号。要使用的话可用以下方式调用:
from
string
import
Template
Template有个特殊标示符$,它具有以下的规则:
(1)主要实现方式为$xxx,其中xxx是满足python命名规则的字符串,即不能以数字开头、不能为关键字等;
(2)如果$xxx需要和其他字符串接触时,用{}将xxx包裹起来;
开发人员通过编写template文件,并通过Template方法创建模板、substitute方法替换字符串即可快捷的生成所需的文件。编写template文件时一定要注意“$”的使用,因为Python会将以“$”开头的字符串理解成需要替换的变量。
2,replace
str.replace(old, new[, max])
Python replace() 方法把字符串中的 old(旧字符串) 替换成 new(新字符串),如果指定第三个参数max,则替换不超过 max 次。
模板文件tenplate.java(自:基于模板的简易代码生成器Python源码)
/** * created since ${now} */ package com.alipay.mspcore.common.dal.ibatis; import java.util.Date; import junit.framework.Assert; import com.alipay.mspcore.common.dal.daointerface.${testObject}DAO; import com.alipay.mspcore.common.dal.dataobject.${testObject}; import com.alipay.sofa.runtime.test.AnnotatedAutowireSofaTestCase; import com.iwallet.biz.common.util.money.Money; /** * @author ${author} * @version ${version}: MBIM_Service${testObject}_Device.java, ${now} ${author} */ public class Ibatis${testObject}DAOTester extends AnnotatedAutowireSofaTestCase { @Override public String[] getConfigurationLocations() { return new String[] { "META-INF/spring/common-dal-db.xml", "META-INF/spring/mobilespcore-common-dal-dao.xml", "META-INF/spring/common-dal.xml" }; } @Override public String[] getResourceFilterNames() { return new String[] { "META-INF/spring/common-dal-db.xml" }; } @Override public String[] getWebServiceConfigurationLocations() { return new String[] {}; } private ${testObject}DAO get${testObject}DAO() { ${testObject}DAO dao = (${testObject}DAO) this.getBean("${testObjVarName}DAO", ${testObject}DAO.class, null); return dao; } public void test${testObject}DAO() { ${testObject}DAO configDAO = get${testObject}DAO(); Assert.assertNotNull(configDAO); } public void test() { ${testObject}DO ${testObjVarName}DO = new ${testObject}DO(); ${testObjVarName}DO.setGmtCreate(new Date()); ${testObjVarName}DO.setGmtModified(new Date()); ${testObjVarName}DO.setId(10000); ${testObject}DAO ${testObjVarName}DAO = get${testObject}DAO(); long result = ${testObjVarName}DAO.insert(${testObjVarName}DO); Assert.assertTrue(result > 0); } }
Python代码
import os import datetime from string import Template tplFilePath = 'D:\\Project\\Python\\code_gen\\template.java' path = 'D:\\Project\\Python\\code_gen\\' testObjList = ['Basic_connect', 'Sms', 'Phonebook', ] for testObj in testObjList: testObjVarName = testObj[0].lower() + testObj[1:] filename = 'MBIM_Service_' + testObj +'_device.java' author = 'AidanZhang' version='V0.1' now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S') tplFile = open(tplFilePath) gFile = open(path+filename ,"w") #method 1 with Template and substitute(safe_substitute) lines=[] tmp=Template(tplFile.read()) lines.append(tmp.substitute( author = author, now = now, testObject = testObj, testObjVarName = testObjVarName, version = version)) gFile.writelines(lines) ''' #Method 2, with replace fileList = tplFile.readlines() for fileLine in fileList: line = fileLine.replace('${author}',author) .replace('${now}',now) .replace('${testObject}',testObj) .replace('${version}',version) .replace('${testObjVarName}',testObjVarName) print line gFile.writelines(line) ''' tplFile.close() gFile.close() print 'generate %s over. ~ ~' % (path+filename)
运行结果
generate D:\Project\Python\code_gen\MBIM_Service_Basic_connect_device.java over. ~ ~ generate D:\Project\Python\code_gen\MBIM_Service_Sms_device.java over. ~ ~ generate D:\Project\Python\code_gen\MBIM_Service_Phonebook_device.java over. ~ ~
使用Python生成源文件的两种方法