Update 'server.py'

server.py

@@ -1,49 +1,49 @@
-#!/usr/bin/env python
+#!/usr/bin/env python
-
+
-from socket import socket, AF_INET, SOCK_STREAM
+from socket import socket, AF_INET, SOCK_STREAM
-from threading import Thread
+from threading import Thread
-from time import sleep
+from time import sleep
-from subprocess import check_call
+from subprocess import check_call
-from sys import argv, executable
+from sys import argv, executable
-try:
+try:
-    from cryptography.fernet import Fernet
+    from cryptography.fernet import Fernet
-except:
+except:
-    check_call((executable, '-m', 'pip', 'install', 'cryptography'))
+    check_call((executable, '-m', 'pip', 'install', 'cryptography'))
-    from cryptography.fernet import Fernet
+    from cryptography.fernet import Fernet
-
+
-class Server:
+class Server:
-    s = socket(AF_INET, SOCK_STREAM)
+    s = socket(AF_INET, SOCK_STREAM)
-
+
-    def __init__(self):
+    def __init__(self):
-        if len(argv) == 4: k = argv[3].encode()
+        if len(argv) == 4: k = argv[3].encode()
-        else:
+        else:
-            k = Fernet.generate_key()
+            k = Fernet.generate_key()
-            print(k)
+            print(k).decode()
-        self.f = Fernet(k)
+        self.f = Fernet(k)
-        self.s.bind((argv[1], int(argv[2])))
+        self.s.bind((argv[1], int(argv[2])))
-        self.s.listen(3)
+        self.s.listen(3)
-        self.client, self.addr = self.s.accept()
+        self.client, self.addr = self.s.accept()
-
+
-    def rMsg(self):
+    def rMsg(self):
-        while True:
+        while True:
-            d = ''
+            d = ''
-            d = self.f.decrypt(self.client.recv(1024)).decode()
+            d = self.f.decrypt(self.client.recv(1024)).decode()
-            print(end=d)
+            print(end=d)
-            sleep(0.1)
+            sleep(0.1)
-
+
-    def chat(self):
+    def chat(self):
-        self.receiving = Thread(target=self.rMsg)
+        self.receiving = Thread(target=self.rMsg)
-        self.receiving.daemon = True
+        self.receiving.daemon = True
-        self.receiving.start()
+        self.receiving.start()
-        while True:
+        while True:
-            msg = input()
+            msg = input()
-            msg = f"\nADMIN: {msg}\n"
+            msg = f"\nADMIN: {msg}\n"
-            print(msg)
+            print(msg)
-            self.client.send(self.f.encrypt(msg.encode()))
+            self.client.send(self.f.encrypt(msg.encode()))
-
+
-if __name__ == '__main__':
+if __name__ == '__main__':
-    check_call('clear')
+    check_call('clear')
-    Server_m = Server()
+    Server_m = Server()
-    Server_m.chat()
+    Server_m.chat()
-    Server_m.receiving.join()
+    Server_m.receiving.join()
-    Server_m.s.close()
+    Server_m.s.close()