main.py

# Imports
import os
import base64
import datetime


def main():

    # Prompt user for sorting by day and max tokens per file
    sort_by_day = input("Do you want to sort by day? (y/n): ").lower() == "y"
    if sort_by_day:
        limit_tokens = input("Do you want to limit each of the day files to 10 tokens? (y/n): ").lower() == "y"
    else:
        limit_tokens = False

    # For limit_tokens, create a dict to store each date and the amount 
    if limit_tokens:
        day_file_count = {}
    
    # Output folder
    name = str(datetime.datetime.now()).split(".")[0].replace(":", "-")
    dirs = [
        f"output/{name}",
        f"output/{name}/sorted_by_month",
        f"output/{name}/sorted_by_year",
        f"output/{name}/sorted_by_relative_time_year",
        f"output/{name}/sorted_by_relative_time_month"
    ]
    if sort_by_day:
        dirs.append(f"output/{name}/sorted_by_day")
    for dir in dirs:
        if not os.path.exists(dir):
            os.makedirs(dir)

    # Iterate through file
    with open("input.txt") as f:
        tokens = f.read().splitlines()

    # Starting
    start_time = datetime.datetime.utcnow()
    print("Starting...")
    i = 0
    previous_percent = 0
    for token in tokens:
        i += 1
        try:
            # Splitting token to remove email/pass
            full_token = token
            if ":" in token:
                token = token.split(":")[2]

            # Getting userid
            userid = base64.b64decode(token.split(".")[0] + "==").decode("utf-8")

            # Getting timestamp
            creationdate_unix = int(bin(int(userid))[:-22], 2) + 1420070400000

            # Get date
            year, month = datetime.datetime.fromtimestamp(creationdate_unix / 1000).strftime("%Y"), datetime.datetime.fromtimestamp(creationdate_unix / 1000).strftime("%m")
            if sort_by_day:
                day = datetime.datetime.fromtimestamp(creationdate_unix / 1000).strftime("%d")
                if limit_tokens:
                    if day_file_count.get(f"{year}-{month}-{day}") != None:
                        day_file_count[f"{year}-{month}-{day}"] = day_file_count[f"{year}-{month}-{day}"] + 1
                    else:
                        day_file_count[f"{year}-{month}-{day}"] = 0
                    day = f"{day} #{int(day_file_count[f'{year}-{month}-{day}']/10)}"
            difference = datetime.datetime.utcnow() - datetime.datetime.fromtimestamp(creationdate_unix / 1000)
            years, months = difference.days // 365, (difference.days % 365) // 30

            # Write token
            for dir in dirs:
                if "by_year" in dir:
                    with open(f"{dir}/{year}.txt", "a") as f:
                        f.write(full_token + "\n")
                elif "by_month" in dir:
                    with open(f"{dir}/{year}-{month}.txt", "a") as f:
                        f.write(full_token + "\n")
                elif sort_by_day and "by_day" in dir:
                    with open(f"{dir}/{year}-{month}-{day}.txt", "a") as f:
                        f.write(full_token + "\n")
                elif "relative_time_year" in dir:
                    with open(f"{dir}/{years} year(s).txt", "a") as f:
                        f.write(full_token + "\n")
                elif "relative_time_month" in dir:
                    with open(f"{dir}/{years} year(s) {months % 12} month(s).txt", "a") as f:
                        f.write(full_token + "\n")

        # Error
        except Exception as e:
            print(f"Error - {token} - {str(e)}")
            with open(f"output/{name}/failed.txt", "a") as f:
                f.write(full_token + "\n")
            continue

        # Print percent complete every 10% of progress
        percent = 100 * i / len(tokens)
        if int(percent) % 10 == 0 and int(percent) != previous_percent:
            print(f"{int(percent)}% done...")
            previous_percent = int(percent)

    # Finishing
    print(f"Finished sorting {len(tokens)} tokens in {(datetime.datetime.utcnow() - start_time).total_seconds()} seconds!")


if __name__ == "__main__":
    os.system('cls' if os.name == 'nt' else 'clear')
    main()