diff --git a/Main/management/commands/receive.py b/Main/management/commands/receive.py index 71a912e..a602181 100644 --- a/Main/management/commands/receive.py +++ b/Main/management/commands/receive.py @@ -1,3 +1,6 @@ +from os.path import join, exists +from shutil import rmtree + import pika from django.core.management.base import BaseCommand @@ -27,6 +30,10 @@ class Command(BaseCommand): print(e) solution.result = "TE" solution.save() + finally: + path = join("solutions", str(id)) + if exists(path): + rmtree(path) channel.basic_consume(queue="test", on_message_callback=callback, auto_ack=True) channel.start_consuming() diff --git a/SprintLib/testers/BaseTester.py b/SprintLib/testers/BaseTester.py index ff1447b..f0b9150 100644 --- a/SprintLib/testers/BaseTester.py +++ b/SprintLib/testers/BaseTester.py @@ -91,6 +91,8 @@ class BaseTester: join("solutions/" + str(self.solution.id), file.filename), "wb" ) as fs: fs.write(get_bytes(file.fs_id)) + for file in listdir("solutions/" + str(self.solution.id)): + call("chmod 777 " + "solutions/" + str(self.solution.id) + "/" + file) print("Files copied") try: self.before_test()