Advertisement
renix1

Simple chat with socket

May 27th, 2017
152
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.99 KB | None | 0 0
  1. import socket
  2. from getpass import getuser
  3.  
  4. def receive():
  5.     HOST = '127.0.0.1'
  6.     PORT = 1337
  7.     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  8.     s.bind((HOST, PORT))
  9.     s.listen(1)
  10.     while 1:
  11.         conn, addr = s.accept()
  12.         while 1:
  13.             msg = conn.recv(1024)
  14.             if msg:
  15.                 print("%s" % (str(msg)))
  16.             break
  17.         print("Ending connection with client: ", addr)
  18.  
  19. def send(msg='Nothing'):
  20.     msg = bytes("%s: %s" % (getuser(), msg), 'utf-8')
  21.     HOST = '127.0.0.1'
  22.     PORT = 1337
  23.     s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
  24.     s.connect((HOST, PORT))
  25.     s.send(msg)
  26.  
  27. def main():
  28.     choose = int(input('[1] - Receiver\n[2] - Sender\nChoose a option: '))
  29.     if choose == 1:
  30.         receive()
  31.     elif choose == 2:
  32.         while 1:
  33.             msg = input("Write a message: ")
  34.             send(msg)
  35.     else:
  36.         print('Failed to get response from your choose.\n')
  37.         quit(0)
  38.  
  39. main()
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement