Shageenderan Sapai commited on
Commit
f34f359
·
1 Parent(s): 16186a1

bug fix for user_id in upload

Browse files
Files changed (1) hide show
  1. app/cache.py +2 -3
app/cache.py CHANGED
@@ -17,9 +17,8 @@ AWS_SECRET_KEY = os.getenv('AWS_SECRET_KEY')
17
  REGION = os.getenv('AWS_REGION')
18
 
19
  def upload_file_to_s3(filename):
20
- # user_id = users\to_upload\da43883b-7e12-425f-b498-ac8665dd23fe.pkl extract everything before the first dot and the last \
21
- # might be '/' on linux so extract everything after to_upload?
22
- user_id = re.search(r'users\\to_upload\\(.*).pkl', filename).group(1)
23
  # user_id = filename.split('.')[0]
24
  print(user_id)
25
  function_name = upload_file_to_s3.__name__
 
17
  REGION = os.getenv('AWS_REGION')
18
 
19
  def upload_file_to_s3(filename):
20
+ # Extract user_id from the filename
21
+ user_id = os.path.basename(filename).split('.')[0]
 
22
  # user_id = filename.split('.')[0]
23
  print(user_id)
24
  function_name = upload_file_to_s3.__name__