def __download_file(self, download):
# Make sure the directory exists
save_directory = os.path.dirname(download.save_location)
if not os.path.isdir(save_directory):
os.makedirs(save_directory)
# Fail if the file already exists
if os.path.isdir(download.save_location):
raise IsADirectoryError("{} is a directory".format(download.save_location))
# Resume the previous download if possible
start_point = 0
download_mode = 'wb'
if os.path.isfile(download.save_location):
if self.__is_same_download_as_before(download):
print("Resuming download {}".format(download.save_location))
download_mode = 'ab'
start_point = os.stat(download.save_location).st_size
else:
os.remove(download.save_location)
# Download the file
resume_header = {'Range': 'bytes={}-'.format(start_point)}
download_request = SESSION.get(download.url, headers=resume_header, stream=True)
downloaded_size = start_point
file_size = int(download_request.headers.get('content-length'))
if downloaded_size < file_size:
with open(download.save_location, download_mode) as save_file:
for chunk in download_request.iter_content(chunk_size=DOWNLOAD_CHUNK_SIZE):
# Pause if needed
while self.__paused:
time.sleep(0.1)
save_file.write(chunk)
downloaded_size += len(chunk)
if self.__cancel:
self.__cancel = False
save_file.close()
download.cancel()
self.__current_download = None
return
if file_size > 0:
progress = int(downloaded_size / file_size * 100)
download.set_progress(progress)
save_file.close()
if download.number == download.out_of_amount:
finish_thread = threading.Thread(target=download.finish)
finish_thread.start()
if self.__queue.empty():
Config.unset("current_download")
def __refresh_token(self, refresh_token: str) -> str:
request_url = "https://auth.gog.com/token"
params = {
'client_id': self.client_id,
'client_secret': self.client_secret,
'grant_type': 'refresh_token',
'refresh_token': refresh_token,
}
response = SESSION.get(request_url, params=params)
response_params = response.json()
self.active_token = response_params['access_token']
expires_in = response_params["expires_in"]
self.active_token_expiration_time = time.time() + int(expires_in)
return response_params['refresh_token']
def __get_xml_checksum(url):
result = {}
try:
response = SESSION.get(url)
if response.status_code == http.HTTPStatus.OK and len(response.text) > 0:
response_object = ET.fromstring(response.text)
if response_object and response_object.attrib:
result = response_object.attrib
else:
print("Couldn't read xml data. Response with code {} received with the following content: {}".format(
response.status_code, response.text
))
except requests.exceptions.RequestException as e:
print("Couldn't read xml data. Received RequestException : {}".format(e))
finally:
return result
def __request(self, url: str = None, params: dict = None) -> dict:
# Refresh the token if needed
if self.active_token_expiration_time < time.time():
print("Refreshing token")
refresh_token = Config.get("refresh_token")
Config.set("refresh_token", self.__refresh_token(refresh_token))
# Make the request
headers = {
'Authorization': "Bearer {}".format(str(self.active_token)),
}
response = SESSION.get(url, headers=headers, params=params)
if self.debug:
print("Request: {}".format(url))
print("Return code: {}".format(response.status_code))
print("Response body: {}".format(response.text))
print("")
return response.json()
def download_operation(self, download, start_point, download_mode):
# Download the file
resume_header = {'Range': 'bytes={}-'.format(start_point)}
download_request = SESSION.get(download.url, headers=resume_header, stream=True, timeout=30)
downloaded_size = start_point
file_size = int(download_request.headers.get('content-length'))
result = True
if downloaded_size < file_size:
with open(download.save_location, download_mode) as save_file:
for chunk in download_request.iter_content(chunk_size=DOWNLOAD_CHUNK_SIZE):
# Pause if needed
while self.__paused:
time.sleep(0.1)
save_file.write(chunk)
downloaded_size += len(chunk)
if self.__cancel:
result = False
break
if file_size > 0:
progress = int(downloaded_size / file_size * 100)
download.set_progress(progress)
save_file.close()
return result
def get_file_size(self, url):
xml_link = self.__request(url)['checksum']
xml_string = SESSION.get(xml_link).text
root = ET.fromstring(xml_string)
return root.attrib["total_size"]
def get_download_file_md5(self, url):
xml_link = self.__request(url)['checksum']
xml_string = SESSION.get(xml_link).text
root = ET.fromstring(xml_string)
return root.attrib['md5']
