From 8f3ebbe78a3cfc31cfad33f90bed01e4403d0d18 Mon Sep 17 00:00:00 2001
From: "Iury O. G. Figueiredo" <tau@localhost.localdomain>
Date: Sun, 1 Nov 2020 14:35:01 -0300
Subject: [PATCH] Code improvement.

---
 setup.py    |  2 +-
 websnake.py | 13 ++-----------
 2 files changed, 3 insertions(+), 12 deletions(-)

diff --git a/setup.py b/setup.py
index c9ec707..60e71a3 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@
 
 from distutils.core import setup
 setup(name="websnake",
-      version="2.0.0",
+      version="2.0.1",
       py_modules=['websnake'],
       author="Iury O. G. Figueiredo",
       author_email="ioliveira@id.uff.br",
diff --git a/websnake.py b/websnake.py
index 1c01609..c260105 100644
--- a/websnake.py
+++ b/websnake.py
@@ -6,12 +6,12 @@
 from untwisted.network import Spin, xmap, spawn, SSL
 from untwisted.dispatcher import Dispatcher
 from untwisted.event import get_event
+from base64 import encodebytes
 from untwisted import core
 from tempfile import TemporaryFile 
 from socket import getservbyname
 import sys
 
-
 class Headers(dict):
     def __init__(self, data):
         for ind in data:
@@ -196,22 +196,13 @@ def post(addr, payload=b'', version='HTTP/1.1', headers={},  auth=()):
     if url.scheme == 'https' else create_con(url.hostname, port, request)
 
 def build_auth(username, password):
-    from base64 import encodestring
-    
     # The headers will be encoded as utf8.
     username = username.encode('utf8')
     password = password.encode('utf8')
 
-    base = encodestring(b'%s:%s' % (username, password))
+    base = encodebytes(b'%s:%s' % (username, password))
     base = base.replace(b'\n', b'').decode('utf8')
     return "Basic %s" % base
 
 
 
-
-
-
-
-
-
-