summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--omegle.py2
-rw-r--r--proxy.py14
2 files changed, 11 insertions, 5 deletions
diff --git a/omegle.py b/omegle.py
index 2f70e21..85d87c3 100644
--- a/omegle.py
+++ b/omegle.py
@@ -83,7 +83,7 @@ class OmegleChat(object):
conn.close()
def send(self,msg):
- msg = unicode(msg)
+ msg = unicode(msg).encode("utf8")
self.conn_lock.acquire()
self.conn.request("POST",
"/send",
diff --git a/proxy.py b/proxy.py
index 3501e72..28d7d16 100644
--- a/proxy.py
+++ b/proxy.py
@@ -11,14 +11,17 @@ class OmegleProxyChat(OmegleChat):
OmegleChat.__init__(self)
self.name = name
self.color = color
- def c(self,str):
- return "\033[%sm\033[1m%s\033[0m\033[0m" %(self.color, str)
+ def c(self,str,bold=False):
+ if bold:
+ return "\033[1m\033[%sm%s\033[0m\033[0m" %(self.color, str)
+ else:
+ return "\033[%sm%s\033[0m" %(self.color, str)
def set_partner(self,partner):
self.partner = partner
def on_message(self,message):
- print self.c("[MSG] %s: %s" %(self.name, message))
+ print self.c("[MSG] %s: %s" %(self.name, message),bold=True)
self.partner.send(message)
def on_connect(self):
print self.c("[EVT] %s Connection confirmed" % self.name)
@@ -44,7 +47,10 @@ A.start()
B.start()
print "press enter to abort"
-sys.stdin.readline()
+try:
+ sys.stdin.readline()
+except KeyboardInterrupt:
+ pass
A.disconnect()
B.disconnect()