sebastian/RandomDBGenerator
1
0
Fork 0
Code Issues 3 Pull Requests Packages Projects Releases Wiki Activity

Generate inserts command considering date columns

Browse Source
This commit is contained in:
Sebastian Böckelmann 2024-04-17 10:55:01 +02:00
parent e1c76efc03
commit 4c175a49e7
4 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pythonProject/schema.sql
View File

@ -2,6 +2,7 @@ CREATE TABLE `test` (
`id` bigint(20) NOT NULL, `id` bigint(20) NOT NULL,
`test_value` int(11) NOT NULL, `test_value` int(11) NOT NULL,
`test_string` varchar(255) DEFAULT NULL, `test_string` varchar(255) DEFAULT NULL,
`local_date` date DEFAULT NULL,
PRIMARY KEY (`id`) PRIMARY KEY (`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci

18
pythonProject/sqlgenerator/SQLGenerator.py
View File

@ -1,4 +1,6 @@
import datetime
import random import random
import time
from sqlgenerator.ColumnValue import ValueType, ColumnValue from sqlgenerator.ColumnValue import ValueType, ColumnValue
from sqlparse.SQLColumn import SQLColumn, SQLColumnType from sqlparse.SQLColumn import SQLColumn, SQLColumnType
@ -11,8 +13,12 @@ max_bigint = 9223372036854775807
min_int = -2147483648 min_int = -2147483648
max_int = 2147483647 max_int = 2147483647
min_date = datetime.date(2023, 1, 1)
max_date = datetime.date(2024, 12, 31)
def generate_random_insert(sql_table: SQLTable): def generate_random_insert(sql_table: SQLTable):
columnValues = [] columnValues = []
for column in sql_table.columns: for column in sql_table.columns:
columnValues.append(generate_random_column(column)) columnValues.append(generate_random_column(column))
@ -21,8 +27,7 @@ def generate_random_insert(sql_table: SQLTable):
if columnValue.valueType == ValueType.INT: if columnValue.valueType == ValueType.INT:
sql_insert_command = sql_insert_command.replace("${i}", str(columnValue.value) + ", ${i}") sql_insert_command = sql_insert_command.replace("${i}", str(columnValue.value) + ", ${i}")
else: else:
sql_insert_command = sql_insert_command.replace("${i}", "'"+columnValue.value+"', ${i}") sql_insert_command = sql_insert_command.replace("${i}", "'" + columnValue.value + "', ${i}")
sql_insert_command = sql_insert_command.replace(", ${i}", "") sql_insert_command = sql_insert_command.replace(", ${i}", "")
return sql_insert_command return sql_insert_command
@ -30,10 +35,15 @@ def generate_random_insert(sql_table: SQLTable):
def generate_random_column(sql_column: SQLColumn): def generate_random_column(sql_column: SQLColumn):
if sql_column.columnType == SQLColumnType.BIGINT: if sql_column.columnType == SQLColumnType.BIGINT:
return ColumnValue( random.randint(min_bigint, max_bigint), ValueType.INT) return ColumnValue(random.randint(min_bigint, max_bigint), ValueType.INT)
elif sql_column.columnType == SQLColumnType.INT: elif sql_column.columnType == SQLColumnType.INT:
return ColumnValue(random.randint(min_int, max_int), ValueType.INT) return ColumnValue(random.randint(min_int, max_int), ValueType.INT)
elif sql_column.columnType == SQLColumnType.VARCHAR: elif sql_column.columnType == SQLColumnType.VARCHAR:
text_legnth = random.randint(1, 255) text_legnth = random.randint(1, 255)
lorem_text = lorem.sentence() lorem_text = lorem.sentence()
return ColumnValue(lorem_text[:text_legnth], ValueType.STRING) return ColumnValue(lorem_text[:text_legnth], ValueType.STRING)
elif sql_column.columnType == SQLColumnType.DATE:
delta = max_date - min_date
random_days = random.randrange(delta.days + 1)
random_date = min_date + datetime.timedelta(days=random_days)
return ColumnValue(str(random_date), ValueType.STRING)

2
pythonProject/sqlparse/SQLColumn.py
View File

@ -5,7 +5,7 @@ class SQLColumnType(Enum):
BIGINT = 'bigint(20)' BIGINT = 'bigint(20)'
INT = 'int(11)' INT = 'int(11)'
VARCHAR = 'varchar(255)' VARCHAR = 'varchar(255)'
DATE = 'date'
class SQLColumn: class SQLColumn:

2
pythonProject/sqlparse/SQLParser.py
View File

@ -70,6 +70,8 @@ def get_column_type(column_definition: str):
return SQLColumn.SQLColumnType.INT return SQLColumn.SQLColumnType.INT
elif column_definition_characteristic == 'varchar(255)': elif column_definition_characteristic == 'varchar(255)':
return SQLColumn.SQLColumnType.VARCHAR return SQLColumn.SQLColumnType.VARCHAR
elif column_definition_characteristic == 'date':
return SQLColumn.SQLColumnType.DATE
def get_column_name(column_definition: str): def get_column_name(column_definition: str):