def gen_proto(args): '''Generate proto from sql file''' print '-'*50 print 'Generating .proto\n' if len(args) < 2: print '''Using 'config/server_database.sql' for sql file, 'shared/man_db_msg.proto' for proto file. If you don't want it: Usage: gen_proto sql_filename proto_filename''' dir = os.path.dirname(os.path.abspath(__file__)) + '/' sql_filename = dir + 'config/server_database.sql' proto_filename = 'shared/man_db_msg.proto' else: sql_filename = args[0] proto_filename = args[1] db = sql_sparser.get_db_struct(sql_filename) p = db2proto_gen.protogen() p.from_db(db, proto_filename) print '-'*50 + '\n'
def gen_orm(args): '''Generate orm from sql file''' print '-'*50 print 'Generating orm\n' dir = os.path.dirname(os.path.abspath(__file__)) + '/' if len(args) < 2: print '''Using 'config/server_database.sql' for sql file, 'shared/orm.py' for orm file. If you don't want it: Usage: gen_orm sql_filename orm_filename''' orm_filename = dir + 'shared/orm.py' sql_filename = dir + 'config/server_database.sql' else: sql_filename = args[0] orm_filename = args[1] orm_file = open(orm_filename, 'w') db = sql_sparser.get_db_struct(sql_filename) res_str = db2orm_gen.generate_classes(db) orm_file.write(res_str) orm_file.close() print '-'*50 + '\n'