diff --git a/.idea/misc.xml b/.idea/misc.xml
index 1fd5465..3c29c38 100644
--- a/.idea/misc.xml
+++ b/.idea/misc.xml
@@ -1,4 +1,4 @@
-
+
\ No newline at end of file
diff --git a/config.py b/config.py
index 8c1bd8a..6c979ea 100644
--- a/config.py
+++ b/config.py
@@ -44,3 +44,5 @@ test_user_id = 10
# 随机数种子
RANDOM_SEED = 42
+
+generate_fake_data_enable=False
\ No newline at end of file
diff --git a/project_evaluation_analysis/generate_history_data.py b/project_evaluation_analysis/generate_history_data.py
index 3397830..cc82700 100644
--- a/project_evaluation_analysis/generate_history_data.py
+++ b/project_evaluation_analysis/generate_history_data.py
@@ -60,6 +60,9 @@ def generate_history_data():
repository
"""
else:
+ if config.generate_fake_data_enable==False:
+ config.logger.info(f"因generate_fake_data_enable=False,故取消插入日期{date}的repository数据")
+ continue
config.logger.info(f"插入日期{date}的repository数据")
sql = f"""
INSERT INTO
diff --git a/project_evaluation_analysis/generate_milestone_history_data.py b/project_evaluation_analysis/generate_milestone_history_data.py
index 3908def..57fafdc 100644
--- a/project_evaluation_analysis/generate_milestone_history_data.py
+++ b/project_evaluation_analysis/generate_milestone_history_data.py
@@ -62,6 +62,9 @@ SELECT date from ai_milestone_history where date>'{begin_date_str}' GROUP BY dat
def insert_test_milestone(generate_day:datetime):
+ if config.generate_fake_data_enable == False:
+ config.logger.info(f"因generate_fake_data_enable=False,故取消插入日期{generate_day}的milestone数据")
+ return
generate_day_str=generate_day.strftime('%Y-%m-%d')
generate_sql=f"""
insert into ai_milestone_history(date,
diff --git a/project_evaluation_analysis/generate_statistics_data.py b/project_evaluation_analysis/generate_statistics_data.py
index 308771d..314abe0 100644
--- a/project_evaluation_analysis/generate_statistics_data.py
+++ b/project_evaluation_analysis/generate_statistics_data.py
@@ -44,6 +44,12 @@ def generate_statistics_data():
if len(date_list) > 0:
for date in date_list:
+ if config.generate_fake_data_enable == False:
+ if date == current_date:
+ generate_statistics_data_single(conn, date)
+ else:
+ config.logger.info(f"因generate_fake_data_enable=False,故取消插入日期{date}的statistics数据")
+ continue
generate_statistics_data_single(conn, date)
config.logger.info(f"统计个人画像数据完成")
diff --git a/project_evaluation_analysis/generate_task_static_history.py b/project_evaluation_analysis/generate_task_static_history.py
index 074b8c0..1b5e5d5 100644
--- a/project_evaluation_analysis/generate_task_static_history.py
+++ b/project_evaluation_analysis/generate_task_static_history.py
@@ -53,6 +53,7 @@ SELECT date from ai_task_static_history where date>'{begin_date_str}' GROUP BY d
generate_day=last_date+delta
config.logger.info(f"开始生成{generate_day}到{today}的数据")
while generate_day'{begin_date_str}' GROUP BY d
def insert_test_task(generate_day:datetime):
+ if config.generate_fake_data_enable == False:
+ config.logger.info(f"因generate_fake_data_enable=False,故取消插入日期{generate_day}的task_static数据")
+ return
generate_day_str=generate_day.strftime('%Y-%m-%d')
generate_sql=f"""
insert into ai_task_static_history (repo_id,closed_count,open_count,date,is_test)