ChatClient.py

import ast
import json
import os
import platform

from tkinter import *
from emoji import demojize
import requests

from ws4py.client.threadedclient import WebSocketClient

connected = False
channel_ = '公共聊天室'
my_trip = ''
var_text = None
my_text = None

client_name = '[PIG Chat](https://github.com/MuRongPIG/CrossStreet-Python)'
client_key = 'HXjURea2vReYejA'


class Settings:
    def __init__(self):
        self.save_filename = 'chat.json'
        self.username = 'undefined'
        self.password = ''
        self.channel = channel_
        self.load()

    def load(self):
        if not os.path.exists(self.save_filename):
            self.new()
        with open(self.save_filename, 'r') as f:
            js = json.loads(f.read())
            self.username = js['username']
            self.password = js['password']
            self.channel = js['channel']

    def save(self):
        with open(self.save_filename, 'w') as f:
            js = json.dumps({
                'username': self.username,
                'password': self.password,
                'channel': self.channel
            })
            f.write(js)

    def new(self):
        with open(self.save_filename, 'w') as f:
            js = json.dumps({
                'username': self.username,
                'password': self.password,
                'channel': self.channel
            })
            f.write(js)


def parse_cmds(data):
    global my_trip
    m_settings = Settings()
    if 'cmd' not in data:
        return ''
    cmd = data['cmd']
    if cmd == 'onlineAdd':
        if data['nick'] == m_settings.username:
            my_trip = data['trip']
            if my_trip == 'null':
                my_trip = ''
        if api == 0:
            return "%s 加入聊天室" % data['nick'] + '\n' + '来自 %s' % data['client']
        if api == 1:
            return "%s 加入聊天室" % data['nick']

    if cmd == 'chat':
        if 'trip' in data:
            return "%s %s\n%s" % (data['trip'], data['nick'], data['text'])
        return "%s %s\n%s" % (data['nick'], '', data['text'])
    if cmd == 'onlineSet':
        users = ''
        for u in data['nicks']:
            users = users + u + ','
        users += m_settings.username
        if api == 0:
            values = (data['ver'], data['cookie'])
            return '欢迎来到十字街！请保证您已经阅读并同意了服务协议。\n当前聊天室版本： %s \n您的cookie：%s \n如果您所在的聊天室没有在线的用户，' \
                   '可以尝试加入聊天室 ?公共聊天室\n在线的用户: ' % values + users
        if api == 1:
            return 'Users online:' + users
    if cmd == 'info':
        return "信息：" + data['text']
    if cmd == 'warn':
        return "警告：" + data['text']
    if cmd == 'onlineRemove':
        return '%s 离开了聊天室' % data['nick']
    return '(%s)' % json.dumps(data)


class MyClient(WebSocketClient):
    def opened(self):
        def update():
            try:
                update_api = 'https://api.github.com/repos/MuRongPIG/CrossStreet-python/releases/latest'
                js = json.loads(requests.get(update_api).text)
                tag_name = js.get('tag_name')
                print(tag_name)
                if tag_name != 'v1.5.1':
                    var_append(var_text,'有新版本可用：' + tag_name + '\n' + '请前往https://github.com/MuRongPIG/CrossStreet-python/releases/latest 查看')
                else:
                    var_append(var_text,'CrossStreet-Python 已是最新版本')
            except:
                    var_append(var_text,'检测更新失败')
        global connected
        settings = Settings()
        channel = settings.channel
        if channel == '':
            channel = channel_
        # print(self.channel)
        if len(settings.password) > 0 :
            user = settings.username
            pswd = settings.password
            req = json.dumps({"cmd": "join", "channel": channel, "nick": user, "password": pswd, "clientName": client_name, "clientKey": client_key,})
        else:
            user = settings.username
            pswd = ''
            req = json.dumps({"cmd": "join", "channel": channel, "nick": user, "clientName": client_name, "clientKey": client_key,})
        self.send(req)
        update()
        connected = True

    def closed(self, code, reason=None):
        global connected
        #var_append("Closed down:"+code+reason)
        connected = False

    def send_text(self, text):
        data = json.dumps({"cmd": "chat", "text": text})
        print('<-', data,type(data))
        self.send(data)

    def send_all(self, text):
        json_str = ast.literal_eval(text)
        json_str_ = json.dumps(json_str)
        print('<-', json_str_)
        self.send(json_str_)

    def received_message(self, resp):
        global var_text
        print('->', resp)
        try:
            resp = json.loads(str(resp))
        except Exception as e:
            print(resp, e)
            return
        data = resp
        result = parse_cmds(data) + '\n'
        result = demojize(result)
        # print(result)
        var_append(var_text, result)


class ChatClientSetup:
    def __init__(self, root=None):
        global api
        api = 0
        from tkinter import ttk
        self.root = root
        if self.root is None:
            self.root = Tk()
        top = self.root
        self.top = self.root
        if platform.system() == 'Windows':
            top.attributes("-toolwindow", 1)
            top.attributes("-topmost", 1)
        top.resizable(width=True, height=True)
        top.title("登录-新版十字街 By @MuRongPIG")
        # top.overrideredirect(True)
        # self.root.iconify()

        self.setup_user = StringVar()
        self.setup_password = StringVar()
        self.setup_channel = StringVar()
        self.setup_channel.set(channel_)

        self.settings = Settings()

        frame = Frame(top)

        self.setup_user.set(str(self.settings.username))
        self.setup_password.set(str(self.settings.password))
        self.setup_channel.set(str(self.settings.channel))

        Label(frame, text='用户名').grid(row=1, column=1)
        Entry(frame, textvariable=self.setup_user).grid(row=1, column=2)
        Label(frame, text='密码').grid(row=2, column=1)
        Entry(frame, textvariable=self.setup_password,show='*').grid(row=2, column=2)
        Label(frame, text='房间(默认%s)' % channel_).grid(row=3, column=1)
        Entry(frame, textvariable=self.setup_channel).grid(row=3, column=2)


        def go(*args):
            global api
            if comboxlist.get() == "crosst.chat":
                api = 0
            if comboxlist.get() == "hack.chat":
                api = 1

        comboxlist=ttk.Combobox(frame) #初始化
        comboxlist["values"]=("crosst.chat","hack.chat")
        comboxlist.current(0)  #选择第一个
        comboxlist.grid(row=4,column=2)
        comboxlist.bind("<<ComboboxSelected>>",go)


        frame.pack(side=TOP)
        btnn = Button(top, text='登录', command=self.setup_confirm)
        btnn.pack(side=BOTTOM, fill=X, expand=1)

        top.mainloop()

    def setup_confirm(self):
        self.settings.username = self.setup_user.get()
        self.settings.password = self.setup_password.get()
        self.settings.channel = self.setup_channel.get()
        self.settings.save()
        self.top.destroy()

        client = ChatClient()
        client.loop()


    def loop(self):
        self.root.mainloop()


def var_append(var, data):
    my_text.configure(state='normal')
    my_text.mark_set('insert', END)
    # var.set(var.get() + '\n' + data)
    my_text.insert(INSERT, '\n' + data)
    my_text.mark_set('insert', END)
    my_text.see(END)
    my_text.configure(state='disabled')


class ChatClient:
    def __init__(self, root=None):
        if api == 0:
            self.api = 'wss://ws.crosst.chat:35197/'
        if api == 1:
            self.api = 'wss://hack.chat/chat-ws'
        self.settings = Settings()

        self.username = self.settings.username
        self.channel = self.settings.channel
        self.password = self.settings.password

        self.root = root
        if self.root is None:
            self.root = Tk()
        self.title = '聊天室 - %s' % self.channel
        self.root.title(self.title)

        self.var_send = StringVar()
        self.entry = Entry(self.root, textvariable=self.var_send)
        self.entry.grid(column=0, columnspan=5, row=1, sticky=N + E +W)
        self.entry.bind('<Return>', self.send_message)
        global var_text
        var_text = StringVar()

        self.scroll = Scrollbar()
        global my_text
        my_text = Text(self.root)
        self.scroll.config(command=my_text.yview)
        my_text.config(yscrollcommand=self.scroll.set)
        my_text.grid(column=0, row=0)
        self.scroll.grid(column=5, row=0, sticky=N + S + W)

        self.button = Button(self.root, text="SEND", command=self.send_message)
        self.button.grid(column=5, row=1)

        self.frame = Frame(self.root)


        self.var = IntVar()
        self.var.set(1)

        self.Radiobutton1 = Radiobutton(self.frame, text="发送信息/命令(默认)", value=1, variable=self.var)
        self.Radiobutton1.grid(column=0, row=0, sticky=W)

        self.Radiobutton1 = Radiobutton(self.frame, text="发送websocket信息", value=2, variable=self.var)
        self.Radiobutton1.grid(column=1, row=0, sticky=W)
        self.frame.grid(column=0, row=2)


        self.ws = None
        self.init_ws()

    def init_ws(self):
        self.ws = MyClient(self.api)
        self.ws.connect()

    def send_message(self, event=None):
        type_ = self.var.get()
        message = self.var_send.get()
        if type_ == 1:
            self.ws.send_text(message)
            self.var_send.set("")
        if type_ == 2:
            self.ws.send_all(message)
            self.var_send.set("")


    def loop(self):
        self.root.mainloop()


if __name__ == '__main__':
    _settings = Settings()
    _setup = ChatClientSetup()
    _setup.loop()