From 4633484606d10975571a2e2473f351b5191549b6 Mon Sep 17 00:00:00 2001 From: Rishi Date: Tue, 12 Jul 2022 13:19:40 +0530 Subject: [PATCH] Add heirarchy in output folder (based on target) --- src/Osintgram.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/src/Osintgram.py b/src/Osintgram.py index 7abe11d6..1db7ac0f 100644 --- a/src/Osintgram.py +++ b/src/Osintgram.py @@ -61,6 +61,13 @@ def setTarget(self, target, show_output): self.following = self.check_following() if(show_output): self.__printTargetBanner__() + + def create_target_output_directory(self): + # Create directory for target (if absent) + user_output_dir = self.output_dir + "/" + self.target + if not os.path.exists(user_output_dir): + os.mkdir(user_output_dir) + return user_output_dir def __get_feed__(self): data = [] @@ -883,6 +890,8 @@ def get_user_photo(self): next_max_id = results.get('next_max_id') try: + target_output_directory = self.create_target_output_directory() + for item in data: if counter == limit: break @@ -890,7 +899,7 @@ def get_user_photo(self): counter = counter + 1 url = item["image_versions2"]["candidates"][0]["url"] photo_id = item["id"] - end = self.output_dir + "/" + self.target + "_" + photo_id + ".jpg" + end = target_output_directory + "/" + photo_id + ".jpg" urllib.request.urlretrieve(url, end) sys.stdout.write("\rDownloaded %i" % counter) sys.stdout.flush() @@ -902,7 +911,7 @@ def get_user_photo(self): counter = counter + 1 url = i["image_versions2"]["candidates"][0]["url"] photo_id = i["id"] - end = self.output_dir + "/" + self.target + "_" + photo_id + ".jpg" + end = target_output_directory + "/" + photo_id + ".jpg" urllib.request.urlretrieve(url, end) sys.stdout.write("\rDownloaded %i" % counter) sys.stdout.flush()