gmclcore/core.py

#------------------#
import os,requests,json,time,sys,psutil,string,platform,glob,getpass,shutil,hashlib,random,subprocess,math,zipfile,threading
#------------------#

class MultiThreadDownload(threading.Thread):
    '''The Class for Multi-Thread Download'''
    '''Get from Internet and improved by Ghink Network Studio'''
    def __init__(self,url,startpos,endpos,f,UA):
        super(MultiThreadDownload,self).__init__()
        self.url=url
        self.startpos=startpos
        self.endpos=endpos
        self.fd=f
        self.UA=UA
    def download(self):
        headers=self.UA.update({"Range":"bytes=%s-%s"%(self.startpos,self.endpos)})
        res=requests.get(self.url,headers=headers)
        self.fd.seek(self.startpos)
        self.fd.write(res.content)
    def run(self):
        self.download()

class GMCLCore(object):
    '''The Main Class of the Launcher Core'''
    def __init__(self,LauncherName="GMCLCore",LauncherVersion="A0.2.0",LogOutput=False):
        '''The global variable set function'''
        ''':LauncherName The name of your launcher,the default value is "GMCLCore"'''
        ''':LauncherVersion The version of your launcher,the default value is the version of core'''
        ''':LogOutput The switch of the log output function,the default value is False'''
        self.__LauncherName=LauncherName
        self.__LauncherVersion=LauncherVersion
        self.__LogOutput=LogOutput
        self.__UserAgent={'User-Agent':LauncherName+'/'+LauncherVersion+' ((GMCL Core Alpha 0.2.0;Alpha))'}
        self.__Log=[]
        self.__AuthWay=""
        self.Log("info","Successful.","__init__")
    def __del__(self):
        self.Log("info","Successful.","__del__")
    def Log(self,Type,Text,Function="Anonymous Function",ErrorType="TypeError"):
        '''The function which was used to manager the log output system'''
        ''':Type Type of the log info'''
        ''':Text The main text of log info'''
        ''':Function The function that cause the log,the default value is "Anonymous Function"'''
        ''':Type The type of the error,the default value is "TypeError",Support "TypeError","ValueError","UserWarning"'''
        if(Text!="" or None):
            if(Type=="info"):
                self.__Log.append(("INFO",time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),Function,Text))
                return ("INFO",Text)
            elif(Type=="warn"):
                self.__Log.append(("WARN",time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),Function,Text))
                return ("WARN",Text)
            elif(Type=="error"):
                self.__Log.append(("ERROR",time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),Function,Text,ErrorType))
                if(ErrorType=="TypeError"):
                    raise TypeError(Text)
                elif(ErrorType=="ValueError"):
                    raise ValueError(Text)
                else:
                    raise UserWarning(Text)
            else:
                self.__Log.append(("OTHER",time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),Function,Text))
                return ("OTHER",Text)
    def ReturnLog(self):
        '''The function which was used to return the list of program log'''
        self.Log("info","Successful.","ReturnLog")
        return self.__Log
    def PrintLog(self):
        '''The function which was used to print the list of program log'''
        self.Log("info","Successful.","PrintLog")
        print(self.__Log)
    def MakeDir(self,Path):
        '''The function which was used to create a dir with determination'''
        ''':Path The path of the dir you want to create'''
        if(os.path.isdir(Path)==False):
            os.mkdir(Path)
            self.Log("info","Successful.","MakeDir")
        else:
            return self.Log("warn","Exist.","MakeDir")
    def SearchFile(self,Path,Tag,Num=1):
        '''The function which was used to search files'''
        ''':Path The root path you want to search'''
        ''':Tag The name keyword of the file you want to search'''
        ''':Num How much result you want to get,you should set as "all" if you want to return all result,the default value is 1'''
        List=[]
        i=0
        for root,dirs,files in os.walk(Path,topdown=True):
            if Tag in files:
                i+=1
                List.append(root+'\\'+Tag)
                if(i==Num):
                    break
            else:
                pass
        self.Log("info","Success.","SearchFile")
        return List
    def GetUUID(self,Account="Steve"):
        '''The function which was used to generate UUID'''
        ''':Account Your account,the default value is "Steve"'''
        if(self.__AuthWay=="offline"):
            result="OfflinePlayer:"+Account
            md5=hashlib.md5()
            md5.update(result.encode(encoding='utf-8'))
            self.Log("info","Success.","GetUUID")
            return md5.hexdigest()
        else:
            return self.Log("warn","Wrong auth way","GetUUID")
    def GetJavaPath(self,Num=1):
        '''The function which was used to search java'''
        ''':Num How much result you want to get,you should set as "all" if you want to return all result,the default value is 1'''
        if(platform.system()=="Windows"):
            self.Log("info","Success.","GetJavaPath")
            return self.SearchFile('C:\\','javaw.exe',Num)
        elif(platform.system()=="Linux"):
            self.Log("info","Success.","GetJavaPath")
            return self.SearchFile('\\','java',Num)
        else:
            self.Log("info","Success.","GetJavaPath")
            return self.SearchFile('\\','java',Num)
    def SetAccount(self,Account="Steve",Password=""):
        '''The function which was used to set game account data'''
        ''':Account Your account,the default value is "Steve"'''
        ''':Password Your password'''
        if(self.__AuthWay=="online"):
            if(Account.count("@")==1):
                if(json.loads(requests.get("https://authserver.mojang.com/",headers=self.__UserAgent).text)['status']=="OK"):
                    result=json.loads(requests.post("https://authserver.mojang.com/authenticate",'{"agent":{"name":"Minecraft","Version":1},"username":"'+Account+'","password":"'+Password+'"}',headers=self.__UserAgent).text)
                    self.__Token=result['accessToken']
                    self.__UUID=result['selectedProfile']['id']
                    self.__ID=result['selectedProfile']['name']
                    self.Log("info","Success.","GetAccount")
                else:
                    return self.Log("warn","Fail to request Mojang\'s auth server.","SetAccount")
            else:
                return self.Log("warn","Wrong Account.","SetAccount")
        elif(self.__AuthWay=="offline"):
            if(Account.count("@")==1):
                return self.Log("warn","Wrong Account.","SetAccount")
            else:
                self.__ID=Account
                self.__UUID=self.GetUUID(Account)
                self.__Token=""
                self.Log("info","Success.","GetAccount")
        else:
            pass
    def SetAuthWay(self,AuthWay):
        '''The auth way set function'''
        ''':AuthWay Your auth way,must be "offline" or "online"'''
        if(AuthWay=="offline"):
            self.__AuthWay="offline"
            self.Log("info","Success.","SetAuthWay")
        elif(AuthWay=="online"):
            self.__AuthWay="online"
            self.Log("info","Success.","SetAuthWay")
        else:
            self.Log("warn","Wrong Auth Way.","SetAccount")
    def SetRecomMem(self):
        '''The recommendation memory value set function'''
        self.__Memory = float(psutil.virtual_memory().free / 1024 ** 3) * 0.8
        self.Log("info","Success.","SetRecomMem")
    def SetJavaPath(self,path):
        '''The java path set function'''
        ''':Path Your java path'''
        if(path==None):
            self.Log("warn","Wrong local address for java.","SetJavaPath")
        else:
            if(os.path.exists(path)):
                self.__JAVA=path
                self.Log("info","Success.","SetJavaPath")
            else:
                self.Log("warn","Wrong local address for java.","SetJavaPath")
    def SetGameMem(self,Memory):
        '''The game memory value set function'''
        ''':Memory The memory of the game,the unit is GiBytes'''
        try:
            self.__Memory=float(Memory)
            self.Log("info","Success.","SetGameMem")
        except:
            self.Log("warn","Wrong parameter for game memory.","SetGameMem")
    def Download(self,DownloadFrom,DownloadTo,ThreadNum=3):
        '''The multi-thread download function'''
        ''':DownloadFrom The online url for the file'''
        ''':DownloadTo The local path for the file'''
        ''':ThreadNum The number of the thread,the default value is 3'''
        if(DownloadFrom=="" or DownloadTo==""):
            self.Log("warn","Wrong online address or local address for download.","Download")
        else:
            url=DownloadFrom
            filename=DownloadTo
            filesize=int(requests.head(url,headers=self.__UserAgent).headers['Content-Length'])
            threadnum=ThreadNum
            threading.BoundedSemaphore(threadnum)
            step=filesize // threadnum
            mtd_list=[]
            start=0
            end=-1
            tempf=open(filename,'w')
            tempf.close()
            with open(filename,'rb+') as  f:
                fileno=f.fileno()
                while end < filesize -1:
                    start=end +1
                    end=start + step -1
                    if end > filesize:
                        end=filesize
                    dup=os.dup(fileno)
                    fd=os.fdopen(dup,'rb+',-1)
                    t=MultiThreadDownload(url,start,end,fd,self.__UserAgent)
                    t.start()
                    mtd_list.append(t)
                for i in mtd_list:
                    i.join()
            self.Log("info","Success.","Download")