Advertisement
Mochinov

Untitled

May 15th, 2023
1,140
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 1.51 KB | None | 0 0
  1.  
  2.     def update_input(self, file, input_type: str, validator) -> bool:
  3.  
  4.         if (
  5.             instance := self.type_instance.get(input_type)
  6.         ):
  7.             repositories = self.repositories_instance.get(input_type)
  8.             instance = instance.objects.filter(
  9.                 scenario=self.scenario,
  10.                 initial_file_name=input_type,
  11.             ).first()
  12.  
  13.             if instance:
  14.                 instance.data_file = file
  15.                 instance.data_uploaded = True
  16.                 instance.save()
  17.                 data_frame = pd.read_csv(
  18.                     instance.data_file.path,
  19.                 )
  20.                 data_frame.columns = repositories.get("fields")
  21.  
  22.                 data_frame['id'] = [uuid.uuid4() for _ in range(len(data_frame.index))]
  23.                 data_frame["scenario_id"] = data_frame['art_code_nm'].map(lambda _: self.scenario.id)
  24.                 # data_frame["stock"] =  [int(data_frame['stock'][x]) for x in range(len(data_frame['stock'].index))]
  25.                 # print(len(data_frame['stock'].index), len(data_frame.index))
  26.                 if not data_frame.empty:
  27.                     with get_session() as session:
  28.                         data_creator = repositories.get("creator")(
  29.                             session=session,
  30.                             df=data_frame,
  31.                             result_id=instance.id,
  32.                         )
  33.                         data_creator.execute()
  34.  
  35.                     return True
  36.  
  37.         return False
  38.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement