Added an option to provide a custom output directory

Author: anatolykopyl

main.py

@@ -115,10 +115,11 @@ def _quit():
 parser.add_argument('-j', '--json', help='save commands output as JSON file', action='store_true')
 parser.add_argument('-f', '--file', help='save output in a file', action='store_true')
 parser.add_argument('-c', '--cli', help='command line mode', action='store')
+parser.add_argument('-o', '--output', help='where to store photos', action='store')
 
 args = parser.parse_args()
 
-api = Osintgram(args.id, args.file, args.json, args.cli)
+api = Osintgram(args.id, args.file, args.json, args.cli, args.output)
 
 
 commands = {

output/dont_delete_this_folder.txt

@@ -1 +1 @@
-Please don't deleate this folder.
+Please don't delete this folder.

src/Osintgram.py

@@ -31,7 +31,7 @@ class Osintgram:
     writeFile = False
     jsonDump = False
 
-    def __init__(self, target, is_file, is_json, is_cli):
+    def __init__(self, target, is_file, is_json, is_cli, output_dir):
         u = config.getUsername()
         p = config.getPassword()
         if not is_cli:
@@ -150,14 +150,14 @@ def get_addrs(self):
                 i = i + 1
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_addrs.txt"
+                file_name = output_dir + "/" + self.target + "_addrs.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['address'] = addrs_list
-                json_file_name = "output/" + self.target + "_addrs.json"
+                json_file_name = output_dir + "/" + self.target + "_addrs.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -200,7 +200,7 @@ def get_captions(self):
             file = None
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_captions.txt"
+                file_name = output_dir + "/" + self.target + "_captions.txt"
                 file = open(file_name, "w")
 
             for s in captions:
@@ -211,7 +211,7 @@ def get_captions(self):
 
             if self.jsonDump:
                 json_data['captions'] = captions
-                json_file_name = "output/" + self.target + "_followings.json"
+                json_file_name = output_dir + "/" + self.target + "_followings.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -239,7 +239,7 @@ def get_total_comments(self):
             posts += 1
 
         if self.writeFile:
-            file_name = "output/" + self.target + "_comments.txt"
+            file_name = output_dir + "/" + self.target + "_comments.txt"
             file = open(file_name, "w")
             file.write(str(comments_counter) + " comments in " + str(posts) + " posts\n")
             file.close()
@@ -249,7 +249,7 @@ def get_total_comments(self):
                 'comment_counter': comments_counter,
                 'posts': posts
             }
-            json_file_name = "output/" + self.target + "_comments.json"
+            json_file_name = output_dir + "/" + self.target + "_comments.json"
             with open(json_file_name, 'w') as f:
                 json.dump(json_data, f)
 
@@ -285,13 +285,13 @@ def get_comment_data(self):
 
         print(t)
         if self.writeFile:
-            file_name = "output/" + self.target + "_comment_data.txt"
+            file_name = output_dir + "/" + self.target + "_comment_data.txt"
             with open(file_name, 'w') as f:
                 f.write(str(t))
                 f.close()
 
         if self.jsonDump:
-            file_name_json = "output/" + self.target + "_comment_data.json"
+            file_name_json = output_dir + "/" + self.target + "_comment_data.json"
             with open(file_name_json, 'w') as f:
                 f.write("{ \"Comments\":[ \n")
                 f.write('\n'.join(json.dumps(comment) for comment in _comments) + ',\n')
@@ -351,14 +351,14 @@ def get_followers(self):
                 followings_list.append(follow)
 
         if self.writeFile:
-            file_name = "output/" + self.target + "_followers.txt"
+            file_name = output_dir + "/" + self.target + "_followers.txt"
             file = open(file_name, "w")
             file.write(str(t))
             file.close()
 
         if self.jsonDump:
             json_data['followers'] = followers
-            json_file_name = "output/" + self.target + "_followers.json"
+            json_file_name = output_dir + "/" + self.target + "_followers.json"
             with open(json_file_name, 'w') as f:
                 json.dump(json_data, f)
 
@@ -416,14 +416,14 @@ def get_followings(self):
                 followings_list.append(follow)
 
         if self.writeFile:
-            file_name = "output/" + self.target + "_followings.txt"
+            file_name = output_dir + "/" + self.target + "_followings.txt"
             file = open(file_name, "w")
             file.write(str(t))
             file.close()
 
         if self.jsonDump:
             json_data['followings'] = followings_list
-            json_file_name = "output/" + self.target + "_followings.json"
+            json_file_name = output_dir + "/" + self.target + "_followings.json"
             with open(json_file_name, 'w') as f:
                 json.dump(json_data, f)
 
@@ -472,7 +472,7 @@ def get_hashtags(self):
             hashtags_list = []
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_hashtags.txt"
+                file_name = output_dir + "/" + self.target + "_hashtags.txt"
                 file = open(file_name, "w")
 
             for k, v in ssort:
@@ -488,7 +488,7 @@ def get_hashtags(self):
 
             if self.jsonDump:
                 json_data['hashtags'] = hashtags_list
-                json_file_name = "output/" + self.target + "_hashtags.json"
+                json_file_name = output_dir + "/" + self.target + "_hashtags.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
         else:
@@ -564,7 +564,7 @@ def get_user_info(self):
                 if 'contact_phone_number' in data and data['contact_phone_number']:
                     user['contact_phone_number'] = data['contact_phone_number']
 
-                json_file_name = "output/" + self.target + "_info.json"
+                json_file_name = output_dir + "/" + self.target + "_info.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(user, f)
 
@@ -590,7 +590,7 @@ def get_total_likes(self):
             posts += 1
 
         if self.writeFile:
-            file_name = "output/" + self.target + "_likes.txt"
+            file_name = output_dir + "/" + self.target + "_likes.txt"
             file = open(file_name, "w")
             file.write(str(like_counter) + " likes in " + str(like_counter) + " posts\n")
             file.close()
@@ -600,7 +600,7 @@ def get_total_likes(self):
                 'like_counter': like_counter,
                 'posts': like_counter
             }
-            json_file_name = "output/" + self.target + "_likes.json"
+            json_file_name = output_dir + "/" + self.target + "_likes.json"
             with open(json_file_name, 'w') as f:
                 json.dump(json_data, f)
 
@@ -635,7 +635,7 @@ def get_media_type(self):
         if counter > 0:
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_mediatype.txt"
+                file_name = output_dir + "/" + self.target + "_mediatype.txt"
                 file = open(file_name, "w")
                 file.write(str(photo_counter) + " photos and " + str(video_counter) + " video posted by target\n")
                 file.close()
@@ -648,7 +648,7 @@ def get_media_type(self):
                     "photos": photo_counter,
                     "videos": video_counter
                 }
-                json_file_name = "output/" + self.target + "_mediatype.json"
+                json_file_name = output_dir + "/" + self.target + "_mediatype.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -700,14 +700,14 @@ def get_people_who_commented(self):
             print(t)
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_users_who_commented.txt"
+                file_name = output_dir + "/" + self.target + "_users_who_commented.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['users_who_commented'] = ssort
-                json_file_name = "output/" + self.target + "_users_who_commented.json"
+                json_file_name = output_dir + "/" + self.target + "_users_who_commented.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
         else:
@@ -768,14 +768,14 @@ def get_people_who_tagged(self):
             print(t)
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_users_who_tagged.txt"
+                file_name = output_dir + "/" + self.target + "_users_who_tagged.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['users_who_tagged'] = ssort
-                json_file_name = "output/" + self.target + "_users_who_tagged.json"
+                json_file_name = output_dir + "/" + self.target + "_users_who_tagged.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
         else:
@@ -816,14 +816,14 @@ def get_photo_description(self):
                 count += 1
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_photodes.txt"
+                file_name = output_dir + "/" + self.target + "_photodes.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['descriptions'] = descriptions_list
-                json_file_name = "output/" + self.target + "_descriptions.json"
+                json_file_name = output_dir + "/" + self.target + "_descriptions.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -869,7 +869,7 @@ def get_user_photo(self):
                     counter = counter + 1
                     url = item["image_versions2"]["candidates"][0]["url"]
                     photo_id = item["id"]
-                    end = "output/" + self.target + "_" + photo_id + ".jpg"
+                    end = output_dir + "/" + self.target + "_" + photo_id + ".jpg"
                     urllib.request.urlretrieve(url, end)
                     sys.stdout.write("\rDownloaded %i" % counter)
                     sys.stdout.flush()
@@ -881,7 +881,7 @@ def get_user_photo(self):
                         counter = counter + 1
                         url = i["image_versions2"]["candidates"][0]["url"]
                         photo_id = i["id"]
-                        end = "output/" + self.target + "_" + photo_id + ".jpg"
+                        end = output_dir + "/" + self.target + "_" + photo_id + ".jpg"
                         urllib.request.urlretrieve(url, end)
                         sys.stdout.write("\rDownloaded %i" % counter)
                         sys.stdout.flush()
@@ -913,7 +913,7 @@ def get_user_propic(self):
                 URL = data["hd_profile_pic_versions"][items-1]['url']
 
             if URL != "":
-                end = "output/" + self.target + "_propic.jpg"
+                end = output_dir + "/" + self.target + "_propic.jpg"
                 urllib.request.urlretrieve(URL, end)
                 pc.printout("Target propic saved in output folder\n", pc.GREEN)
 
@@ -942,12 +942,12 @@ def get_user_stories(self):
                 story_id = i["id"]
                 if i["media_type"] == 1:  # it's a photo
                     url = i['image_versions2']['candidates'][0]['url']
-                    end = "output/" + self.target + "_" + story_id + ".jpg"
+                    end = output_dir + "/" + self.target + "_" + story_id + ".jpg"
                     urllib.request.urlretrieve(url, end)
 
                 elif i["media_type"] == 2:  # it's a gif or video
                     url = i['video_versions'][0]['url']
-                    end = "output/" + self.target + "_" + story_id + ".mp4"
+                    end = output_dir + "/" + self.target + "_" + story_id + ".mp4"
                     urllib.request.urlretrieve(url, end)
 
         if counter > 0:
@@ -1013,14 +1013,14 @@ def get_people_tagged_by_user(self):
                     tagged_list.append(tag)
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_tagged.txt"
+                file_name = output_dir + "/" + self.target + "_tagged.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['tagged'] = tagged_list
-                json_file_name = "output/" + self.target + "_tagged.json"
+                json_file_name = output_dir + "/" + self.target + "_tagged.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -1032,7 +1032,7 @@ def get_user(self, username):
         try:
             content = self.api.username_info(username)
             if self.writeFile:
-                file_name = "output/" + self.target + "_user_id.txt"
+                file_name = output_dir + "/" + self.target + "_user_id.txt"
                 file = open(file_name, "w")
                 file.write(str(content['user']['pk']))
                 file.close()
@@ -1219,14 +1219,14 @@ def get_fwersemail(self):
                 t.add_row([str(node['id']), node['username'], node['full_name'], node['email']])
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_fwersemail.txt"
+                file_name = output_dir + "/" + self.target + "_fwersemail.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['followers_email'] = results
-                json_file_name = "output/" + self.target + "_fwersemail.json"
+                json_file_name = output_dir + "/" + self.target + "_fwersemail.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -1300,14 +1300,14 @@ def get_fwingsemail(self):
                 t.add_row([str(node['id']), node['username'], node['full_name'], node['email']])
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_fwingsemail.txt"
+                file_name = output_dir + "/" + self.target + "_fwingsemail.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['followings_email'] = results
-                json_file_name = "output/" + self.target + "_fwingsemail.json"
+                json_file_name = output_dir + "/" + self.target + "_fwingsemail.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -1381,14 +1381,14 @@ def get_fwingsnumber(self):
                 t.add_row([str(node['id']), node['username'], node['full_name'], node['contact_phone_number']])
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_fwingsnumber.txt"
+                file_name = output_dir + "/" + self.target + "_fwingsnumber.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['followings_phone_numbers'] = results
-                json_file_name = "output/" + self.target + "_fwingsnumber.json"
+                json_file_name = output_dir + "/" + self.target + "_fwingsnumber.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -1463,14 +1463,14 @@ def get_fwersnumber(self):
                 t.add_row([str(node['id']), node['username'], node['full_name'], node['contact_phone_number']])
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_fwersnumber.txt"
+                file_name = output_dir + "/" + self.target + "_fwersnumber.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['followings_phone_numbers'] = results
-                json_file_name = "output/" + self.target + "_fwerssnumber.json"
+                json_file_name = output_dir + "/" + self.target + "_fwerssnumber.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
 
@@ -1525,14 +1525,14 @@ def get_comments(self):
             print(t)
 
             if self.writeFile:
-                file_name = "output/" + self.target + "_users_who_commented.txt"
+                file_name = output_dir + "/" + self.target + "_users_who_commented.txt"
                 file = open(file_name, "w")
                 file.write(str(t))
                 file.close()
 
             if self.jsonDump:
                 json_data['users_who_commented'] = ssort
-                json_file_name = "output/" + self.target + "_users_who_commented.json"
+                json_file_name = output_dir + "/" + self.target + "_users_who_commented.json"
                 with open(json_file_name, 'w') as f:
                     json.dump(json_data, f)
         else: