main.py

import API.Presence
import argparse
import math
import time
from pypresence import Presence

parser = argparse.ArgumentParser()

parser.add_argument("-u", "--user", type=int, help = "User ID", required=True)
parser.add_argument("-r", "--refresh", type=int, default=20, help = "Refresh period")
parser.add_argument("-o", "--online", type=bool, default=False, help = "Show when you are just Online")
parser.add_argument("-s", "--security", type=str, default="", help = "A .ROBLOSECURITY token, necessary for the API to gather game info for some players")
 
args = parser.parse_args()

ClientID = 874664899035402283
UserID = args.user
ShowOnline = args.online
RefreshPeriod = args.refresh
RobloSecurity = args.security

ThisPresence = API.Presence.PresenceAPI(roblosecurity = RobloSecurity)

RPC = None

Connected = False
StartTime = 0

def InitRPC():
    global RPC
    RPC = Presence(ClientID, pipe=0)
    
def Connect():
    global Connected
    global RPC

    if Connected == False:
        try:
            RPC.connect()
            Connected = True
            return True
        except:
            RPC = None
            return False
    return False
def Disconnect():
    global Connected
    global StartTime
    global RPC

    if Connected == True:
        try:
            RPC.close()
            StartTime = 0
            Connected = False
            return True
        except:
            RPC = None
            StartTime = 0
            Connected = False
            return False
    return False

print("Started!")

while True:
    if RPC is None:
        try:
            InitRPC()
            print("Connected to Discord!")
        except:
            print("Failed to connect to Discord, is it running?")
    
    CurrentPresence = None
    try:
        CurrentPresence = ThisPresence.GetUserPresence(UserID)
    except:
        print("Failed to get presence data!")

    if CurrentPresence != None:
        GameName = CurrentPresence["lastLocation"]
        RootPlaceID = CurrentPresence["rootPlaceId"]
        if GameName == None or GameName == "":
            GameName = ""

        Buttons = [
            {"label": "Profile", "url": "https://www.roblox.com/users/" + str(UserID) + "/profile"}
        ]
        if RootPlaceID  != "" and RootPlaceID != None:
            Buttons.append({"label": "Experience", "url": "https://www.roblox.com/games/" + str(RootPlaceID)})
    
        if CurrentPresence["userPresenceType"] == 1 and ShowOnline == True:
            if StartTime == 0:
                StartTime = math.floor(time.time())
            if Connect() == True:
                print("User went Online!")
            if GameName != "":
                if Connected == True:
                    try:
                        RPC.update(state=GameName, details="Online", large_image="player", small_image="online", start=StartTime, buttons=Buttons)
                    except:
                        Disconnect()
            else:
                if Connected == True:
                    try:
                        RPC.update(details="Online", large_image="player", small_image="online", start=StartTime, buttons=Buttons)
                    except:
                        Disconnect()
        elif CurrentPresence["userPresenceType"] == 2:
            if StartTime == 0:
                StartTime = math.floor(time.time())
            if Connect() == True:
                print("User opened the Player!")
        
            if GameName != "":
                if Connected == True:
                    try:
                        RPC.update(state=GameName, details="Playing", large_image="player", small_image="playing", start=StartTime, buttons=Buttons)
                    except:
                        Disconnect()
            else:
                if Connected == True:
                    try:
                        RPC.update(details="Playing", large_image="player", small_image="playing", start=StartTime, buttons=Buttons)
                    except:
                        Disconnect()
        elif CurrentPresence["userPresenceType"] == 3:
            if StartTime == 0:
                StartTime = math.floor(time.time())
            if Connect() == True:
                print("User opened Studio!")
        
            if GameName != "":
                if Connected == True:
                    try:
                        RPC.update(state=GameName, details="Building", large_image="studio", small_image="building", start=StartTime, buttons=Buttons)
                    except:
                        Disconnect()
            else:
                if Connected == True:
                    try:
                        RPC.update(details="Building", large_image="studio", small_image="building", start=StartTime, buttons=Buttons)
                    except:
                        Disconnect()
        else:
            if Disconnect() == True:
                print("User went Offline!")
    
    time.sleep(RefreshPeriod)