def main(): # os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3' # # with open(args.zipfile) as file: # print(file) # # if args.zipfile[-4:] == '.zip': # with zipfile.ZipFile(args.zipfile,"r") as zip_ref: # zip_ref.extractall('extracted') # elif args.zipfile[-3:] == '.7z': # archive = py7zr.SevenZipFile(args.zipfile, mode='r') # archive.extractall('extracted') # archive.close() # else: # print("Cannot read the zip file") dirName = 'extracted' listOfFiles = getListOfFiles(dirName) reader = Reader() for elem in listOfFiles: print(elem) reader.predict(elem) files = glob.glob('extracted/*') print('---------------------NOW DELETE------------------------') for elem in listOfFiles: os.remove(elem) for f in files: print(f) try: shutil.rmtree(f) except OSError as e: print("Error: %s - %s." % (e.filename, e.strerror))