cli.py

import requests,os
BASE_URL = 'https://api.github.com/users/peme969'
following_url = f'{BASE_URL}/following'
followers_url = f'{BASE_URL}/followers'
exceptions = {'following':[],'followers':[]}
exc_4f = os.environ['Exceptions_follow'] # for multiple users seperate by a comma (eg. user1, user2, user3)
exc_f = os.environ['Exceptions_following'] # for multiple users seperate by a comma (eg. user1, user2, user3)
exceptions['followers'].extend([user.strip() for user in exc_f.split(',')])
exceptions['followering'].extend([user.strip() for user in exc_4f.split(',')])
not_ = []
yes = []
GITHUB_TOKEN = os.environ['Graphql_Token']
HEADERS = {
    'Authorization': f'token {GITHUB_TOKEN}',
    'Accept': 'application/vnd.github.v3+json',
}
def check_exception(which,data):                              
    if which == 'follower':
        if data in exceptions['followers']:
            return False
        else:
            return True
     elif which == 'following':
        if data in exceptions['following']:
            return False
        else:
            return True
     else:
        print('Sorry, this option doesnt exist :(')
def fetch_all(url):
    results = []
    while url:
        try:
            response = requests.get(url, headers=HEADERS)
            response.raise_for_status()
            results.extend(response.json())
            url = response.links.get('next', {}).get('url')
        except requests.exceptions.RequestException as e:
            print(f"Error fetching data from {url}: {e}")
            break
    return results
def follow_user(username):
    url = f'https://api.github.com/user/following/{username}'
    try:
        response = requests.put(url, headers=HEADERS)
        if response.status_code == 204:
            if check_exceptions('following',username):
                print(f'\033[1;32mSuccessfully followed {username}.\033[0m')
            else:
                print(f'\033[1;31mSeems\033[0m like \033[1;31m{username}\033[0m was in your exceptions list.')
                yes.append(username)
        else:
            print(f'\033[1;31mFailed to follow {username}. {response.json()}\033[0m')
    except requests.exceptions.RequestException as e:
        print(f'\033[1;31mGithub API error. Error following {username}: {e}\033[0m')
def unfollow_user(username):
    url = f'https://api.github.com/user/following/{username}'
    try:
        response = requests.delete(url, headers=HEADERS)
        if response.status_code == 204:
            if check_exceptions('followers',username):
                print(f'\033[1;32mSuccessfully unfollowed {username}.\033[0m')
            else:
                print(f'\033[1;31mSeems\033[0m like \033[1;31m{username}\033[0m was in your exceptions list.')
                not_.append(username)
        else:
            print(f'\033[1;31mGithub API error. Failed to unfollow {username}. {response.json()}\033[0m')
    except requests.exceptions.RequestException as e:
        print(f'\033[1;31mError unfollowing {username}: {e}\033[0m')
following_data = fetch_all(following_url)
followers_data = fetch_all(followers_url)
following = [user['login'].lower() for user in following_data]  
followers = [user['login'].lower() for user in followers_data] 
not_followers = [user for user in following if user not in followers]
not_following_back = [user for user in followers if user not in following]
for user in not_followers:
    print(f'\033[1;33mYou follow {user}, but they don\'t follow you back.\033[0m')
    consent = input(f"Do you want to unfollow {user}? (yes/no): ").strip().lower()
    if consent == "yes":
        unfollow_user(user)
    else:
        print(f'Okay! Didn\'t unfollow {user}...')
for user in not_following_back:
    print(f'\033[1;36m{user} follows you, but you don\'t follow them back.\033[0m')
    consent = input(f"Do you want to follow {user}? (yes/no): ").strip().lower()
    if consent == "yes":
        follow_user(user)
    else:
        print(f'Okay! Didn\'t follow {user}...')
print("\033[1;34mProcessing complete!\033[0m")
print(f'\033[1;34m📊 Summary:\033[0m')
print(f'\033[1;33m🔴 {len(not_followers)} user(s) you follow but don\'t follow you back.\033[0m')
print(f'\033[1;36m🟢 {len(not_following_back)} user(s) follow you but you don\'t follow back.\033[0m')
print('\033[1mUsers that were in your \033[3mfollowing\033[0m\033[1m exception list that were\033[3mNOT\033[0m followed back:\033[0m')
print('\n', *yes, sep='\n\033[33;1m  •\033[0m ')
print('\033[1mUsers that were in your \033[3mfollowers\033[0m\033[1m exceptions list that were\033[3mNOT\033[0m unfollowed: \033[0m')
print('\n', *not_, sep='\n\033[33;1m  •\033[0m ')