import argparse
import os
import sys
import pymongo

sys.path.append(os.path.join(sys.path[0], '../'))
from instabot import Bot

myclient = pymongo.MongoClient("mongodb://localhost:27017/")
mydb = myclient["instagate"]
mycol = mydb["users"]

parser = argparse.ArgumentParser(add_help=True)
parser.add_argument('-u', type=str, help="username")
parser.add_argument('-p', type=str, help="password")
parser.add_argument('-proxy', type=str, help="proxy")
parser.add_argument('users', type=str, nargs='+', help='users')
args = parser.parse_args()

bot = Bot()
bot.login(username=args.u, password=args.p,
          proxy=args.proxy)

for username in args.users:
    flwrs = bot.get_user_followers(username)
    for flwr_id in flwrs:
        follower = mycol.find_one({"user_id": flwr_id})
        if follower:
            print("Warning: Skipping duplicate entry!")
            continue

        nickname = bot.get_username_from_user_id(flwr_id)
        print (nickname)

        followers = bot.get_user_followers(flwr_id)
        following = bot.get_user_following(flwr_id)
        payload = {"username": nickname, "following": following, "followers": followers, "user_id": flwr_id}
        try:
            mycol.insert_one(payload)
        except:
            print("Warning: Skipping duplicate entry!")