From a6a343fde4b205fd2b3a8eac713813498d6acb21 Mon Sep 17 00:00:00 2001
From: samerbam <samerbam@users.noreply.github.com>
Date: Sun, 11 Jun 2023 18:13:07 -0400
Subject: [PATCH] add in temp files

---
 root/app/fanficdownload.py | 20 ++++++++++++--------
 1 file changed, 12 insertions(+), 8 deletions(-)

diff --git a/root/app/fanficdownload.py b/root/app/fanficdownload.py
index c4b2fb2..d19710f 100644
--- a/root/app/fanficdownload.py
+++ b/root/app/fanficdownload.py
@@ -7,7 +7,7 @@ from optparse import OptionParser
 import re
 from configparser import ConfigParser
 from tempfile import mkdtemp
-from shutil import rmtree, copyfile
+from shutil import rmtree, copyfile, copy2
 import socket
 from time import strftime, localtime
 import os
@@ -124,9 +124,10 @@ def check_regexes(output):
 
 def downloader(args):
     url, inout_file, path, live = args
-    # loc = mkdtemp() 
+    loc = mkdtemp()
+    file_backup = '/config/downloads'
+    # loc = '/config/downloads'
     # loc = '/config/downloads'
-    loc = '/config/downloads'
     if not exists(loc):
         makedirs(loc)
         copyfile("/config/personal.ini", "{}/personal.ini".format(loc))
@@ -213,6 +214,8 @@ def downloader(args):
                     'WARNING',
                     live)
                 output += log("Added /Story-file to library with id 0", 'GREEN', live)
+            # copy2(cur, f"/config/downloads/{get_files(loc, '.epub', False)[0]}")
+            copy2(cur, f"{file_backup}/{get_files(loc, '.epub', False)[0]}")
             remove(cur)
         else:
             res = check_output(
@@ -221,6 +224,7 @@ def downloader(args):
             check_regexes(res)
             cur = get_files(loc, '.epub', True)[0]
             name = get_files(loc, '.epub', False)[0]
+            copy2(cur, f"{file_backup}/{name}")
             rename(cur, name)
             output += log(
                 "Downloaded story {} to {}".format(
@@ -230,15 +234,15 @@ def downloader(args):
                 live)
         if not live:
             print(output.strip())
-        # rmtree(loc)
+        rmtree(loc)
     except Exception as e:
         output += log("Exception: {}".format(e), 'FAIL', live)
         if not live:
             print(output.strip())
-        # try:
-            # rmtree(loc)
-        # except BaseException:
-            # pass
+        try:
+            rmtree(loc)
+        except BaseException:
+            pass
         with open(inout_file, "a") as fp:
             fp.write("{}\n".format(url))