示例#1
0
文件: client.py 项目: atmouse/P2P
  def __init__(self, host, port, usn):
    super(ClientForm, self).__init__()
    self.ui = Ui_Form()
    self.ui.setupUi(self)
    self.ui.lineEdit.setFocus()
    self.running = 1

    self.dpbar = MProgressBar(self) #download bar
    self.dpbar.setOrientation(QtCore.Qt.Vertical)
    self.dpbar.setGeometry(520, 0, 20, 189)
    self.dpbar.setValue(0)
    self.dpbar.setStyle(False)

    self.upbar = MProgressBar(self) #upload bar
    self.upbar.setOrientation(QtCore.Qt.Vertical)
    self.upbar.setGeometry(545, 0, 20, 189)
    self.upbar.setValue(0)
    self.upbar.setStyle(True)

    self.host = host
    self.port = port
    self.size = 1024
    self.socket = None
    self.username = ''

    try:
      self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
      self.socket.connect((self.host, self.port))
      self.key = int(hash(self.socket)) ^ int(time.time())
      self.key = self.key if self.key > 0 else -self.key
    except socket.error:
      print 'Not accepting connections'
      sys.exit(1)

    try:
      self.socket.send(usn)
      response = self.socket.recv(self.size)
      if response == 'ACCEPT':
        self.username = usn
      else:
        print 'Username already exists, please choose a different username'
        sys.exit(1)
    except socket.error:
      print 'Some socket error'

    self.downloader = Downloader(self.key) #make the downloader listen for incoming download requests
    #self.connect(self.downloader, QtCore.SIGNAL("update_upload_progressbar"), self.update_upload_progressbar)
    #self.connect(self.downloader, QtCore.SIGNAL("update_download_progressbar"), self.update_download_progressbar)
    self.downloader.start()

    self.receiver = Receiver(self.socket, self.key)
    self.connect(self.receiver, QtCore.SIGNAL("update_msg"), self.update_msg)
    self.connect(self.receiver, QtCore.SIGNAL("update_userlist"), self.update_userlist)
    #self.connect(self.receiver, QtCore.SIGNAL("update_download_progressbar"), self.update_download_progressbar) #tester method. should be connected to a Downloader object
    #self.connect(self.receiver, QtCore.SIGNAL("update_upload_progressbar"), self.update_upload_progressbar)
    #self.connect(self.receiver, QtCore.SIGNAL("update_download_progressbar"), self.update_download_progressbar)
    self.receiver.start() #start listening
示例#2
0
  def __init__(self, host, port):
    super(ClientForm, self).__init__()
    self.client = Client(host,port)
    self.ui = Ui_Form()
    self.ui.setupUi(self)
    self.ui.lineEdit.setFocus()
    self.icon = QtGui.QIcon('creeper.tif')
    self.auth = False
    self.colour_list = (QtCore.Qt.red, QtCore.Qt.darkRed, QtCore.Qt.blue, QtCore.Qt.darkGreen, QtCore.Qt.magenta, QtCore.Qt.darkBlue, QtCore.Qt.darkCyan,QtCore.Qt.darkMagenta, QtCore.Qt.darkYellow, QtCore.Qt.darkGray, QtGui.QColor('#00CC99'), QtGui.QColor('#0099FF'), QtGui.QColor('#005555'), QtGui.QColor('#FF6600'), QtGui.QColor('#660033'), QtGui.QColor('#9900FF'))
    self.user_colour_list = {}
    self.running = 1
    if os.path.isfile('client.log'):
      os.remove('client.log')
    self.clientLogger = Logger('client.log')

    self.createActions()
    self.createTrayIcon()
    self.trayIcon.activated.connect(self.iconActivated)
    self.trayIcon.show()
示例#3
0
文件: client.py 项目: atmouse/P2P
class ClientForm(QtGui.QWidget):
  def __init__(self, host, port, usn):
    super(ClientForm, self).__init__()
    self.ui = Ui_Form()
    self.ui.setupUi(self)
    self.ui.lineEdit.setFocus()
    self.running = 1

    self.dpbar = MProgressBar(self) #download bar
    self.dpbar.setOrientation(QtCore.Qt.Vertical)
    self.dpbar.setGeometry(520, 0, 20, 189)
    self.dpbar.setValue(0)
    self.dpbar.setStyle(False)

    self.upbar = MProgressBar(self) #upload bar
    self.upbar.setOrientation(QtCore.Qt.Vertical)
    self.upbar.setGeometry(545, 0, 20, 189)
    self.upbar.setValue(0)
    self.upbar.setStyle(True)

    self.host = host
    self.port = port
    self.size = 1024
    self.socket = None
    self.username = ''

    try:
      self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
      self.socket.connect((self.host, self.port))
      self.key = int(hash(self.socket)) ^ int(time.time())
      self.key = self.key if self.key > 0 else -self.key
    except socket.error:
      print 'Not accepting connections'
      sys.exit(1)

    try:
      self.socket.send(usn)
      response = self.socket.recv(self.size)
      if response == 'ACCEPT':
        self.username = usn
      else:
        print 'Username already exists, please choose a different username'
        sys.exit(1)
    except socket.error:
      print 'Some socket error'

    self.downloader = Downloader(self.key) #make the downloader listen for incoming download requests
    #self.connect(self.downloader, QtCore.SIGNAL("update_upload_progressbar"), self.update_upload_progressbar)
    #self.connect(self.downloader, QtCore.SIGNAL("update_download_progressbar"), self.update_download_progressbar)
    self.downloader.start()

    self.receiver = Receiver(self.socket, self.key)
    self.connect(self.receiver, QtCore.SIGNAL("update_msg"), self.update_msg)
    self.connect(self.receiver, QtCore.SIGNAL("update_userlist"), self.update_userlist)
    #self.connect(self.receiver, QtCore.SIGNAL("update_download_progressbar"), self.update_download_progressbar) #tester method. should be connected to a Downloader object
    #self.connect(self.receiver, QtCore.SIGNAL("update_upload_progressbar"), self.update_upload_progressbar)
    #self.connect(self.receiver, QtCore.SIGNAL("update_download_progressbar"), self.update_download_progressbar)
    self.receiver.start() #start listening

  def on_lineEdit_returnPressed(self):
    if self.ui.lineEdit.displayText() != '':
      stringToSend = str(self.ui.lineEdit.displayText())
      #unfortunately, we must intercept a '/download' request client side, as the server does not store search results for the clients

      if stringToSend.startswith("\download "):
        fileToDownload = stringToSend.split(' ')[1]
        for host in searchresults.keys():
          for result in searchresults[host]:
            if fileToDownload == result:
              stringToSend = "\download "+host+" "+str(self.key)+" "+fileToDownload
              self.socket.send(str(stringToSend))
              self.ui.lineEdit.setText('')
              return
      
      try:
        if not stringToSend.startswith("\download "):
          self.socket.send(str(stringToSend)) 
        else:
          self.update_msg('This file is not among your search results')
          
      except socket.error:
        self.ui.textEdit.setTextColor(QtCore.Qt.black)
        self.ui.textEdit.setText('The connection with the server has been lost, please restart the client.')
        self.ui.listWidget.clear() #delete the whole thing
              
    self.ui.lineEdit.setText('')

  def update_userlist(self, l):
    self.ui.listWidget.clear()

    for i in l:
      n = QtGui.QListWidgetItem(str(i))
      self.ui.listWidget.addItem(n)

  def update_msg(self, msg):
    self.ui.textEdit.append(msg)
    self.ui.textEdit.ensureCursorVisible()

  def update_download_progressbar(self):
    #self.dpbar.setValue(value)
    #v = value
    #if not v == 100: #XXX this is ONLY to demonstrate how it would appear. sleeping the thread is NOT a good idea; it defers other GUI events. no idea for a work around right now. its own thread would be overkill.
    #  while not v == 100:
    #    time.sleep(.005)
    #    v += 1
    #    self.dpbar.setValue(v)
    while(1):
      time.sleep(.005)
      self.dpbar.setValue(dprogress[0])
    
  def update_upload_progressbar(self):
    #self.upbar.setValue(value) 
    while(1):
      time.sleep(.005)
      self.upbar.setValue(uprogress[0])
示例#4
0
文件: client.py 项目: rvantonder/P2P
class ClientForm(QtGui.QWidget):
  def __init__(self, host, port, usn):
    super(ClientForm, self).__init__()
    self.ui = Ui_Form()
    self.ui.setupUi(self)
    self.ui.lineEdit.setFocus()
    self.running = 1

    self.dpbar = MProgressBar(self) #download bar
    self.dpbar.setOrientation(QtCore.Qt.Vertical)
    self.dpbar.setGeometry(520, 0, 20, 189)
    self.dpbar.setValue(0)
    self.dpbar.setStyle(True)

    self.dpbar_thread = Update_dpbar()
    self.connect(self.dpbar_thread, QtCore.SIGNAL("set_dpbar"), self.set_dpbar)
    self.dpbar_thread.start() 
    
    self.upbar = MProgressBar(self) #upload bar
    self.upbar.setOrientation(QtCore.Qt.Vertical)
    self.upbar.setGeometry(545, 0, 20, 189)
    self.upbar.setValue(0)
    self.upbar.setStyle(False)
    
    self.upbar_thread = Update_upbar()
    self.connect(self.upbar_thread, QtCore.SIGNAL("set_upbar"), self.set_upbar)
    self.upbar_thread.start()
    
    self.host = host
    self.port = port
    self.size = 1024
    self.socket = None
    self.username = ''

    self.downloadingFromHost = None

    try:
      self.socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
      self.socket.connect((self.host, self.port))
      self.key = int(hash(self.socket)) ^ int(time.time())
      self.key = self.key if self.key > 0 else -self.key
    except socket.error:
      print 'Not accepting connections'
      sys.exit(1)

    try:
      self.socket.send(usn)
      response = self.socket.recv(self.size)
      if response == 'ACCEPT':
        self.username = usn
      else:
        print 'Username already exists, please choose a different username'
        sys.exit(1)
    except socket.error:
      print 'Some socket error'

    self.downloader = Downloader(self.key) #make the downloader listen for incoming download requests
    self.downloader.start()

    self.receiver = Receiver(self.socket, self.key)
    self.connect(self.receiver, QtCore.SIGNAL("update_msg"), self.update_msg)
    self.connect(self.receiver, QtCore.SIGNAL("update_userlist"), self.update_userlist)
    self.receiver.start() #start listening

  def closeEvent(self, event):
    self.dpbar_thread.terminate()
    self.upbar_thread.terminate()

  def on_lineEdit_returnPressed(self):
    if self.ui.lineEdit.displayText() != '':
      stringToSend = str(self.ui.lineEdit.displayText())

      if stringToSend.startswith("\download "):
        fileToDownload = stringToSend.split(' ')[1]
        for host in searchresults.keys():
          for result in searchresults[host]:
            if fileToDownload == result:
              self.downloadingFromHost = host
              stringToSend = "\download "+host+" "+enc(str(self.key))+" "+fileToDownload #host is the hash
              self.socket.send(str(stringToSend))
              self.ui.lineEdit.setText('')
              return
      elif stringToSend.startswith("\pause"): 
        stringToSend = "\pause "+self.downloadingFromHost
        self.update_msg("Download Paused")
      elif stringToSend.startswith(r'\resume'):
        stringToSend = r'\resume '+self.downloadingFromHost
        self.update_msg("Download Resumed")
      
      try:
        if not stringToSend.startswith("\download "):
          self.socket.send(str(stringToSend)) 
        else:
          self.update_msg('This file is not among your search results')
          
      except socket.error:
        self.ui.textEdit.setTextColor(QtCore.Qt.black)
        self.ui.textEdit.setText('The connection with the server has been lost, please restart the client.')
        self.ui.listWidget.clear() #delete the whole thing
              
    self.ui.lineEdit.setText('')

  def update_userlist(self, l):
    self.ui.listWidget.clear()

    for i in l:
      n = QtGui.QListWidgetItem(str(i))
      self.ui.listWidget.addItem(n)

  def update_msg(self, msg):
    self.ui.textEdit.append(msg)
    self.ui.textEdit.ensureCursorVisible()

  def set_dpbar(self, v):
    self.dpbar.setValue(v)

  def set_upbar(self, v):
    self.upbar.setValue(v)
示例#5
0
class ClientForm(QtGui.QWidget, QtCore.QThread):
  def __init__(self, host, port):
    super(ClientForm, self).__init__()
    self.client = Client(host,port)
    self.ui = Ui_Form()
    self.ui.setupUi(self)
    self.ui.lineEdit.setFocus()
    self.icon = QtGui.QIcon('creeper.tif')
    self.auth = False
    self.colour_list = (QtCore.Qt.red, QtCore.Qt.darkRed, QtCore.Qt.blue, QtCore.Qt.darkGreen, QtCore.Qt.magenta, QtCore.Qt.darkBlue, QtCore.Qt.darkCyan,QtCore.Qt.darkMagenta, QtCore.Qt.darkYellow, QtCore.Qt.darkGray, QtGui.QColor('#00CC99'), QtGui.QColor('#0099FF'), QtGui.QColor('#005555'), QtGui.QColor('#FF6600'), QtGui.QColor('#660033'), QtGui.QColor('#9900FF'))
    self.user_colour_list = {}
    self.running = 1
    if os.path.isfile('client.log'):
      os.remove('client.log')
    self.clientLogger = Logger('client.log')

    self.createActions()
    self.createTrayIcon()
    self.trayIcon.activated.connect(self.iconActivated)
    self.trayIcon.show()
    
  def iconActivated(self, event):
      if event in (QtGui.QSystemTrayIcon.Trigger, QtGui.QSystemTrayIcon.DoubleClick):
          self.showNormal()
          #self.previewWindow.show() #normal, if wanted
    
  def createTrayIcon(self):
      self.trayIconMenu = QtGui.QMenu(self)
      self.trayIconMenu.addAction(self.minimizeAction)
      self.trayIconMenu.addAction(self.maximizeAction)
      self.trayIconMenu.addAction(self.restoreAction)
      self.trayIconMenu.addSeparator()
      self.trayIconMenu.addAction(self.quitAction)
        

      self.trayIcon = QtGui.QSystemTrayIcon(self)
      self.trayIcon.setIcon(QtGui.QIcon('tray.png'))
      self.trayIcon.setContextMenu(self.trayIconMenu)
         
         
  def createActions(self):
    self.minimizeAction = QtGui.QAction("Mi&nimize", self,
            triggered=self.hide)

    self.maximizeAction = QtGui.QAction("Ma&ximize", self,
            triggered=self.showMaximized)

    self.restoreAction = QtGui.QAction("&Restore", self,
            triggered=self.showNormal)

    self.quitAction = QtGui.QAction("&Quit", self,
            triggered=QtGui.qApp.quit)
         
  def closeEvent(self, event):
      if self.trayIcon.isVisible():
          #QtGui.QMessageBox.information(self, "Systray",
              #"Program continues running in tray")
          self.hide()
          event.ignore()


  def parse_msg(self, msg): 
  
    try:
      user,msg = msg.split(': ',1) 
      try:
        if msg.startswith('(l'): #then it is a pickle, if not just display message TODO
          userlist = pickle.loads(msg)

          userlist.sort()

          for i in userlist:
            if not (str(i) in self.user_colour_list):
              self.user_colour_list[str(i)] = self.colour_list[len(self.user_colour_list)%len(self.colour_list)]
          
          self.ui.listWidget.clear() #delete the whole thing

          for user in userlist: 
            n = QtGui.QListWidgetItem(self.icon, str(user))
            n.setTextColor(self.user_colour_list[str(user)])
            self.ui.listWidget.addItem(n)
        elif msg == 'TERMINATE':
          self.running = 0
          self.clientLogger.logger.info("Server Termination")
          return
        else: 
          self.ui.textEdit.setTextColor(self.user_colour_list.get(str(user), QtCore.Qt.black))
          self.ui.textEdit.append(user+': '+msg)
          self.ui.textEdit.ensureCursorVisible() #this will scroll it down

      except EOFError: #if the user sends data that seems like a pickle
        self.ui.textEdit.append(user+': '+msg) #just do this
    except ValueError:
      self.ui.textEdit.setTextColor(QtCore.Qt.black)
      self.ui.textEdit.append(msg)
      self.ui.textEdit.ensureCursorVisible() #whisper

  def request_username(self):
    try:
      self.client.open_socket()
    except socket.error:
      self.ui.textEdit.append("[ERROR] Server is not accepting connections") 
      self.clientLogger.logger.warn("Server is not accepting connections")
      return

    while 1: #loop until accepted
        response = self.client.socket.recv(self.client.size)
        if response == 'REJECT':
          self.ui.textEdit.setTextColor(QtCore.Qt.black)
          self.ui.textEdit.append('Username rejected -- already in use') #TODO server should say this nicely
          self.ui.lineEdit.setText('Enter username') #sometimes it 'skips' over this. no idea why; try not having focus here when settin
          self.client.close_socket()
          self.client.open_socket()
        elif response == 'ACCEPT':
          self.clientLogger.logger.info('Username accepted')
          break #get out
          
    self.auth = True #if accepted
    self.clientLogger.logger.info('Authorized')

  def run(self):       
    self.clientLogger.logger.info('Running')

    self.request_username() #a loop until an acceptable username is given

    while self.running:
      try:
        response = self.client.socket.recv(self.client.size)
      except socket.error:
        self.clientLogger.logger.warn('Response error from server')
        self.client.close_socket()
        self.clientLogger.logger.info("Socket closed")
        return
      try:
        self.parse_msg(response)
      except UnboundLocalError:
        pass  
          
    self.client.close_socket()
    self.clientLogger.logger.info("Socket closed")

  def on_lineEdit_returnPressed(self):
    if self.ui.lineEdit.displayText() != '':
      stringToSend = self.ui.lineEdit.displayText()

      self.clientLogger.logger.info('Would like to send '+stringToSend)
      if not self.auth:
        try:
          self.client.socket.send('request:'+str(stringToSend)) #its a username request
        except socket.error:
          self.ui.textEdit.setText('[ERROR] Server is not accepting connections')
          self.clientLogger.logger.info('Sending request for username '+stringToSend)
      else:
        try:
          self.client.socket.send(str(stringToSend)) #cast qt string
        except socket.error:
          self.ui.textEdit.setTextColor(QtCore.Qt.black)
          self.ui.textEdit.setText('The connection with the server has been lost, please restart the client.')
          self.clientLogger.logger.info('Connection with the server has been lost.')
          self.ui.listWidget.clear() #delete the whole thing
              
    self.ui.lineEdit.setText('')