@@ -50,6 +50,7 @@ def parse_cmd_arags() -> argparse.ArgumentParser:
50
50
parser.add_argument('--ints',type=int,default=0,help="int column counts")
51
51
parser.add_argument('--floats',type=int,default=0,help="float column counts")
52
52
parser.add_argument('--doubles',type=int,default=0,help='double column counts')
53
+ parser.add_argument('--texts',type=int,default=0,help="text column counts")
53
54
parser.add_argument('--varchars',type=int,default=0,help="varchar column counts")
54
55
parser.add_argument('--varchar-length',type=int,default=128,help="varchar column length default 128")
55
56
parser.add_argument('--decimals',type=int,default=0,help="decimal column counts")
@@ -110,8 +111,12 @@ def create(args:argparse.ArgumentParser):
110
111
logger.error("decimal-precision argument must big then 0")
111
112
sys.exit(3)
112
113
for d in range(args.decimals):
113
- columns.append(f"dm{d} decimal({args.decimal_precision},{args.decimal_scale})")
114
-
114
+ columns.append(f"dm{d} decimal({args.decimal_precision},{args.decimal_scale}) not null")
115
+ # 检查 text 字段的数量
116
+ if args.texts >=1:
117
+ for t in range(args.texts):
118
+ columns.append(f"t{t} text not null")
119
+
115
120
# 如果没有指定任何类型的列,那么直接退出
116
121
if len(columns) == 1 and args.auto_primary_key==True:
117
122
logger.error(f"do not have any columns in table {args.database}.{args.table}")
@@ -204,6 +209,10 @@ def insert_sql(args:argparse.ArgumentParser) -> tuple:
204
209
if args.decimals >= 1:
205
210
for dm in range(args.decimals):
206
211
columns.append( (f"dm{dm}",get_float_value) )
212
+
213
+ if args.texts >= 1:
214
+ for t in range(args.texts):
215
+ columns.append( (f"t{t}",get_str_value) )
207
216
208
217
if len(columns) == 0:
209
218
logger.error(f"columns counts equal 0")
@@ -257,6 +266,7 @@ def drop(args):
257
266
logger.error(str(err))
258
267
sys.exit(3)
259
268
269
+
260
270
261
271
262
272
def main():
0 commit comments