diff --git a/models/interview_zsgc.py b/models/interview_zsgc.py index 342287e..58b15ff 100644 --- a/models/interview_zsgc.py +++ b/models/interview_zsgc.py @@ -40,10 +40,11 @@ class InterviewDo: insert_data = [self.data_in] if isinstance(self.data_in, list): insert_data = self.data_in - - # sql = f"insert into test.interview(auth, state, name, phone, position, teacher_name, boss_name, stage, " \ - # f"interview_type,interview_location, interview_time) values" - sql = f"insert into HR.resumes(name, kardid) values" + keys = () + for data in insert_data: + keys = tuple(data.keys()) + break + sql = f"insert into HR.resumes{keys} values" print(sql) return {'sql': sql, 'insert_data': insert_data @@ -53,19 +54,37 @@ class InterviewDo: updateStr = '' whereStr = '' for key, value in self.data_in.items(): - updateStr += str(key) + ' = ' + str(value) + ' ' + if isinstance(value, str): + updateStr += str(key) + ' = ' + "'" + value + "'" + ' ' + continue + updateStr += str(key) + ' = ' + value + ' ' for key, value in self.where.items(): - whereStr += str(key) + ' = ' + str(value) + ' ' - updateStr = updateStr.strip() - whereStr = whereStr.strip() - sql = f"alter table HR.resumes update `{updateStr}` where `{whereStr}`" + if isinstance(value, str): + whereStr += str(key) + ' = ' + "'" + value + "'" + ' ' + continue + whereStr += str(key) + ' = ' + value + ' ' + updateStr.strip() + whereStr.strip() + sql = f"alter table HR.resumes update {updateStr} where {whereStr}" print(sql) return {'sql': sql, } def find_interview_sql(self): - sql = f"select `{self.find_column}` from HR.resumes where {self.where}" + whereStr = '' + findStr = '' + for fstr in self.find_column: + findStr += fstr + ', ' + for key, value in self.where.items(): + if isinstance(value, str): + whereStr += str(key) + ' = ' + "'" + value + "'" + ' ' + continue + whereStr += str(key) + ' = ' + value + ' ' + whereStr = whereStr.strip() + findStr = findStr.strip().strip(',') + # sql = f"select {findStr} from HR.resumes where {whereStr}" + sql = f"select * from HR.resumes where {whereStr}" print(sql) return {'sql': sql, }