用100多行python代碼開發java代碼生成器

不少時候，咱們須要對某個表進行業務開發，生成對應的controller,service,dao,entity等操做類，分別編寫增、刪、改、查業務操做方法 ，而這些細活每每比較簡單且沒有挑戰性，純粹苦力活，又累又耗時，程序的生命是有限，不能浪費，這些方法徹底能夠有跡可尋的能夠採用某些方法快速產生，所以想到python這個利器，採用 flask服務基於html定製代碼模板，因而有了代碼生成器的開發，目前有部份功能還未完成，不過仍是能夠知足基本的開發應用；

先說明，首先文中所用到的模板，是基於咱們如今有項目定製的，像其中的BaseMarketMongoServiceImpl等類是咱們框架底層的基類，若是你須要使用請自行到templates目錄更改；

開發工具版本:python 3.5 + flask + pycharm

上代碼：

# -*- coding: UTF-8 -*-
'''
Created on 2018年7月24日
經過模板自動生成java文件,可生成controller,service,dao,entity層等類,提供簡單的list,findById,delete,save,update,page方法
@author: JL
'''

import os
import json
import time
import tarfile
from flask import Flask, render_template, send_from_directory, request

app = Flask(__name__)


@app.route('/index')
def index():
    return render_template('create_class.html')


@app.route("/download/<filename>", methods=['GET'])
def downloader(filename):
    # 指定文件下載目錄，默認爲當前項目根路徑
    dirpath = os.path.join(app.root_path, '')
    # as_attachment=True 表示下載
    return send_from_directory(dirpath, filename, as_attachment=True)


@app.route('/createClass', methods=['GET', 'POST'])
def create_class():
    file_name = msg = None
    # {'column': {'age': 'int', 'id': 'String', 'address': 'String', 'name': 'String'}, 'table': 'cc_user'}
    fields = request.form['fields']
    if len(fields) <= 0:
        msg='request data json is null!'
    print(fields)
    j = json.loads(fields, encoding='utf-8')
    class_name = j['class']
    package = j['package']
    db_type = j['type']
    if len(class_name) <= 0 :
        msg = 'className is null!'
    if len(package) <= 0:
        msg = 'package is null'
    if len(db_type) <= 0:
        msg = 'type is null'
    print(class_name + '\n' + package)
    if not msg or len(msg) <= 0:
        d = time.strftime("%Y-%m-%d", time.localtime())
        entity = request.form.get('entity')
        if entity and len(entity) >= 1:
            print('--- create entity class')
            create_entity(class_name, package, j['table'], j['column'], db_type, d)
        dao = request.form.get('dao')
        if dao and len(dao) >= 1:
            print('--- create dao class')
            create_dao(class_name, package, d)
        service = request.form.get('service')
        if service and len(service) >= 1:
            print('--- create service class')
            create_service(class_name, package, d)
        controller = request.form.get('controller')
        if controller and len(controller) >= 1:
            print('--- create controller class')
            create_controller(class_name, package, d)
            file_name = make_targz()
    return render_template('create_class.html', msg=msg, file_name=file_name)


# 建立entity
def create_entity(class_name, package, table_name, columns, db_type, date):
    propertys = ''
    methods = ''
    if columns:
        for key in columns.keys():
            propertys += 'private %s %s;' % (columns[key], key) + '\n'
            p1 = 'public void get%s() {\nreturn this.%s;\n}' % (key.title(), key)
            p2 = 'public %s set%s(%s %s) {\nthis.%s = %s;\n}' % (columns[key], key.title(), columns[key], key, key, key)
            methods += p1+'\n'+p2
    c = {'package': package + '.entity',
         'entity_package': package + '.entity.' + class_name,
         'class_name': class_name,
         'table_name': table_name,
         'propertys': propertys,
         'methods': methods,
         'date': date}
    if db_type == 'mongodb':
        s = render_template('entity_mongodb_templates.html', **c)
        create_java_file(class_name, package + '.entity', s)
    elif db_type == 'mysql':
        s = render_template('entity_mysql_templates.html', **c)
        create_java_file(class_name, package + '.entity', s)
        s = render_template('entity_mysql_mapper_templates.html', **c)
        create_java_file(class_name, package + '.entity', s, 'Mapper.xml')


# 建立Dao
def create_dao(class_name, package, date):
    c = {'package': package + '.dao',
         'class_name': class_name,
         'entity_package': package + '.entity.' + class_name,
         'date': date}
    s = render_template('dao_templates.html', **c)
    create_java_file(class_name + 'Dao', package + '.dao', s)


# 建立Service
def create_service(class_name, package, date):
    c = {'package': package + '.service',
         'class_name': class_name,
         'small_class_name': small_str(class_name),
         'entity_package': package + '.entity.' + class_name,
         'dao_package': package + '.dao.' + class_name + 'Dao',
         'date': date}
    s = render_template('service_templates.html', **c)
    create_java_file(class_name + 'Service', package + '.service', s)


# 建立controller
def create_controller(class_name, package, date):
    c = {'package': package + '.entity',
         'class_name': class_name,
         'small_class_name': small_str(class_name),
         'entity_package': package + '.entity.' + class_name,
         'dao_package': package + '.dao.' + class_name + 'Dao',
         'service_package': package + '.dao.' + class_name + 'Service',
         'date': date}
    s = render_template('controller_templates.html', **c)
    # print(s)
    create_java_file(class_name + 'Controller', package + '.controller', s)


# 將首字母轉換爲小寫
def small_str(s):
    if len(s) <= 1:
        return s
    return (s[0:1]).lower() + s[1:]


# 建立java文件
def create_java_file(class_name, package, text, suffix='.java'):
    dirs = 'D:/temp/python/'+package.replace('.', '/')+'/'
    if not os.path.exists(dirs):
        os.makedirs(dirs, 0o777)
    fd = os.open(dirs + class_name + suffix, os.O_WRONLY | os.O_CREAT)
    os.write(fd, text.encode(encoding="utf-8", errors="strict"))
    os.close(fd)


#生成tar.gz壓縮包
def make_targz():
    file_name = 'com.tar.gz'
    source_dir = 'D:/temp/python/'
    with tarfile.open(file_name, "w:gz") as tar:
        tar.add(source_dir, arcname=os.path.basename(source_dir))
    return file_name


if __name__ == '__main__':
    app.run()

模板定製：java結構代碼

package {{ package }};

import org.springframework.data.mongodb.core.mapping.Document;

/**
 * @author admin
 * @version 1.0
 * @Description 實體類
 * @date {{ date }}
 */
@Document(collection = "{{ table_name }}")
public class {{ class_name }} {

{{ propertys }}

{{ methods }}
}

其它就不一一貼上代碼，文末會貼上項目下載碼雲連接；

運行服務，在瀏覽器中輸入：http://127.0.0.1:5000/index

輸入數據結構json串，選擇生成的類，結構比較簡單，都看的懂，提交後，頁腳有下載打包的連接；

下載解壓即有了java代碼

打開目錄下java文件

完畢，固然功能比較簡單，後續再完善。

最後貼上碼雲中項目的地址：https://gitee.com/omsgit/auto-create-java