pyCraft/start.py

import urllib
import urllib2
import getpass
import sys
from network import PacketSenderManager, NetworkManager
from interface import NoGUIstuff, GUI
import time
import threading
noGoooeees = False
try:
    import wx
    wx = wx
except ImportError:
    print "wxPython not found, falling back to no gui version"
    noGoooeees = True
      
class MinecraftLoginThread(threading.Thread):
    
    def __init__(self, window, rotationthread, username, password):
        threading.Thread.__init__(self)
        self.username = username
        self.password = password
        self.window = window
        self.rotationthread = rotationthread
               
    def run(self):
        url = 'https://login.minecraft.net'
        header = {'Content-Type' : 'application/x-www-form-urlencoded'}
        data = {'user' : self.username, 
                'password' : self.password,
                'version' : '13'}
        try:
            data = urllib.urlencode(data)
            req = urllib2.Request(url, data, header)
            opener = urllib2.build_opener()
            response = opener.open(req)
            response = response.read()
        except urllib2.URLError:
            self.rotationthread.Kill = True
            self.window.Status.SetForegroundColour(wx.RED)
            self.window.Status.SetLabel('Connection to minecraft.net failed')
            return
        if(response == "Bad login"):
            self.rotationthread.Kill = True
            self.window.Status.SetForegroundColour(wx.RED)
            self.window.Status.SetLabel('Incorrect username or password')
            return
        if(response == "Account migrated, use e-mail as username."):
            self.rotationthread.Kill = True
            self.window.Status.SetForegroundColour(wx.RED)
            self.window.Status.SetLabel('Account migrated, use e-mail as username.')
        response = response.split(":")
        username = response[2]
        sessionID = response[3]
        #KeepConnectionAlive(self.username, self.password, self.window).start()
        self.window.parent.username = username
        self.window.parent.sessionID = sessionID
        self.window.loggedIn = True
        self.window.handlePostLogin(self.rotationthread)
        
class KeepConnectionAlive(threading.Thread):
    
    def __init__(self, username, password, window):
        threading.Thread.__init__(self)
        self.username = username
        self.password = password
        self.window = window
     
    def run(self):
        while True:
            time.sleep(300)   
            url = 'https://login.minecraft.net'
            header = {'Content-Type' : 'application/x-www-form-urlencoded'}
            data = {'user' : self.username,
                    'password' : self.password,
                    'version' : '13'}
            try:
                data = urllib.urlencode(data)
                req = urllib2.Request(url, data, header)
                opener = urllib2.build_opener()
                response = opener.open(req)
                response = response.read()
            except urllib2.URLError:
                popup = wx.MessageBox('Keep alive to minecraft.net failed', 'Warning', 
                              wx.OK | wx.ICON_ERROR)
                popup.ShowModal()
            if(response == "Bad login"):
                popup = wx.MessageBox('Keep alive to minecraft.net failed (Incorrect username/password)', 'Warning', 
                              wx.OK | wx.ICON_ERROR)
                popup.ShowModal()
                return
            response = response.split(":")
            sessionID = response[3]
            if(self.window != None):
                self.window.parent.sessionID = sessionID
                self.window.parent.loggedIn = True            

if __name__ == "__main__":
    if (len(sys.argv) > 1):
        if(sys.argv[1] == "nogui" or noGoooeees == True):
            user = raw_input("Enter your username: ")
            passwd = getpass.getpass("Enter your password: ")
            derp = NoGUIstuff.loginToMinecraft(user, passwd)
            if(derp['Response'] == "Incorrect username/password" or derp['Response'] == "Can't connect to minecraft.net"):
                print derp['Response']
                sys.exit()                
            sessionid = derp['SessionID']
            print "Logged in as " + derp['Username'] + "! Your session id is: " + sessionid
            stuff = raw_input("Enter host and port if any: ")
            if ':' in stuff:
                StuffEnteredIntoBox = stuff.split(":")
                host = StuffEnteredIntoBox[0]
                port = int(StuffEnteredIntoBox[1])
            else:
                host = stuff
                port = 25565
            connection = NetworkManager.ServerConnection(None, derp['Username'], passwd, sessionid, host, port)
            connection.start()
            while True:
                chat = raw_input()
                PacketSenderManager.send03(connection.grabSocket(), chat)
    else:
        app = wx.PySimpleApp()
        Login = GUI.MainFrame(None, -1, "")
        app.SetTopWindow(Login)
        Login.Show()
        app.MainLoop()
        app2 = wx.PySimpleApp()
        ChatPanel = GUI.ServerChatPanel(None, -1, "pyCraft")
        app2.SetTopWindow(ChatPanel)
        ChatPanel.Show()
        app2.MainLoop()