GilbertClaus commited on
Commit
4265ecd
·
1 Parent(s): 12c8076

User-Agent

Browse files
Files changed (1) hide show
  1. others.py +7 -3
others.py CHANGED
@@ -85,14 +85,18 @@ def add_space(name):
85
  return new_name
86
 
87
  def download_file(url, new_name, directory):
88
- response = requests.get(url, stream=True)
 
 
 
89
  content_type = response.headers.get("Content-Type")
90
  extension = mimetypes.guess_extension(content_type)
 
91
 
92
  if not os.path.exists(directory):
93
  os.makedirs(directory)
94
 
95
- filename = f"{directory}/{new_name}{extension}"
96
  with open(filename, 'wb') as file:
97
  total_size = int(response.headers.get("Content-Length", 0))
98
  progress_bar = tqdm(total=total_size, unit="B", unit_scale=True, ncols=80)
@@ -104,7 +108,7 @@ def download_file(url, new_name, directory):
104
 
105
  progress_bar.close()
106
 
107
- return filename
108
 
109
  def convert_japanese_to_romaji(text):
110
  translator = Translator(service_urls=['translate.google.com'])
 
85
  return new_name
86
 
87
  def download_file(url, new_name, directory):
88
+ headers = {
89
+ "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
90
+ }
91
+ response = requests.get(url, stream=True, headers=headers)
92
  content_type = response.headers.get("Content-Type")
93
  extension = mimetypes.guess_extension(content_type)
94
+ file_name = new_name + "." + extension
95
 
96
  if not os.path.exists(directory):
97
  os.makedirs(directory)
98
 
99
+ filename = f"{directory}/{file_name}"
100
  with open(filename, 'wb') as file:
101
  total_size = int(response.headers.get("Content-Length", 0))
102
  progress_bar = tqdm(total=total_size, unit="B", unit_scale=True, ncols=80)
 
108
 
109
  progress_bar.close()
110
 
111
+ return f'File berhasil diunduh dan disimpan di direktori: {filename}'
112
 
113
  def convert_japanese_to_romaji(text):
114
  translator = Translator(service_urls=['translate.google.com'])