Skip to content

Commit 644c38e

Browse files
huyuwei1996林武
authored andcommitted
Fix: prevent overwriting existing init files in code generator
1 parent 19ef52d commit 644c38e

File tree

2 files changed

+11
-2
lines changed

2 files changed

+11
-2
lines changed

backend/plugin/code_generator/service/gen_service.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -175,8 +175,9 @@ async def generate(self, *, pk: int) -> None:
175175
code_folder.mkdir(parents=True, exist_ok=True)
176176

177177
init_filepath = code_folder.joinpath('__init__.py')
178-
async with aiofiles.open(init_filepath, 'w', encoding='utf-8') as f:
179-
await f.write(gen_template.init_content)
178+
if not os.path.exists(init_filepath):
179+
async with aiofiles.open(init_filepath, 'w', encoding='utf-8') as f:
180+
await f.write(gen_template.init_content)
180181

181182
# api __init__.py
182183
if 'api' in str_code_filepath:

backend/sql/postgresql/init_test_data.sql

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -36,3 +36,11 @@ values (1, 'af4c804f-3966-4949-ace2-3bb7416ea926', 'admin', '用户88888', '$2b
3636

3737
insert into sys_user_role (id, user_id, role_id)
3838
values (1, 1, 1);
39+
40+
-- Reset auto-increment values for each table based on max id
41+
SELECT setval(pg_get_serial_sequence('sys_dept', 'id'),COALESCE(MAX(id), 0) + 1, true) FROM sys_dept;
42+
SELECT setval(pg_get_serial_sequence('sys_menu', 'id'),COALESCE(MAX(id), 0) + 1, true) FROM sys_menu;
43+
SELECT setval(pg_get_serial_sequence('sys_role', 'id'),COALESCE(MAX(id), 0) + 1, true) FROM sys_role;
44+
SELECT setval(pg_get_serial_sequence('sys_role_menu', 'id'),COALESCE(MAX(id), 0) + 1, true) FROM sys_role_menu;
45+
SELECT setval(pg_get_serial_sequence('sys_user', 'id'),COALESCE(MAX(id), 0) + 1, true) FROM sys_user;
46+
SELECT setval(pg_get_serial_sequence('sys_user_role', 'id'),COALESCE(MAX(id), 0) + 1, true) FROM sys_user_role;

0 commit comments

Comments
 (0)