gyao: "fix" temp directory

Amani-so · Jan 5, 2020 · e5c834e · e5c834e
1 parent c8af7b1
commit e5c834e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/yuu/ext/gyao.py b/yuu/ext/gyao.py
@@ -21,15 +21,16 @@ def __init__(self, files, key, iv, url, session):
         self.merge = True
 
         if os.name == "nt":
-            yuu_folder = os.path.join(os.getenv('LOCALAPPDATA'), 'yuu_data')
+            self.yuu_folder = os.path.join(os.getenv('LOCALAPPDATA'), 'yuu_data')
             sffx = '\\'
         else:
             self.yuu_folder = os.path.join(os.getenv('HOME'), '.yuu_data')
             sffx = '/'
         if not os.path.isdir(self.yuu_folder):
             os.mkdir(self.yuu_folder)
 
-        self.temporary_folder = tempfile.mkdtemp(suffix=sffx, dir=self.yuu_folder)
+        self.temporary_folder = tempfile.mkdtemp(dir=self.yuu_folder)
+        self.temporary_folder = self.temporary_folder + sffx
 
 
     def download_chunk(self):
@@ -42,12 +43,12 @@ def download_chunk(self):
                             vid = self.session.get(tsf)
                             outf.write(vid.content)
                         except Exception as err:
-                            print('[ERROR] Problem occured\nreason: {}'.format(err))
+                            yuu_log.error('Problem occured\nreason: {}'.format(err))
                             return None, self.temporary_folder
                     pbar.update()
                     self.downloaded_files.append(outputtemp)
         except KeyboardInterrupt:
-            print('[WARN] User pressed CTRL+C, cleaning up...')
+            yuu_log.warn('User pressed CTRL+C, cleaning up...')
             return None, self.temporary_folder
         return self.downloaded_files, self.temporary_folder