ChatWidget.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-

'''
Created on 2017年12月20日
@author: Irony."[讽刺]
@site: http://alyl.vip, http://orzorz.vip, http://coding.net/u/892768447, http://github.com/892768447
@email: 892768447@qq.com
@file: ChatWidget
@description: 
'''
from collections import OrderedDict
import json
import os
import sys

from PyQt5.Qsci import QsciScintilla, QsciLexerJSON
from PyQt5.QtChart import QChartView
from PyQt5.QtCore import Qt, pyqtSignal
from PyQt5.QtGui import QColor, QFont, QFontMetrics, QKeySequence, QMovie
from PyQt5.QtWidgets import QSplitter, QTreeWidget, QApplication, QWidget,\
    QVBoxLayout, QPushButton, QTreeWidgetItem, QMessageBox, QShortcut, QLabel
import chardet

from ChartView import ChartView  # @UnresolvedImport


__version__ = "0.0.1"


class LoadingWidget(QLabel):

    def __init__(self, *args, **kwargs):
        super(LoadingWidget, self).__init__(*args, **kwargs)
        self.setAlignment(Qt.AlignHCenter | Qt.AlignVCenter)
        self._movie = QMovie("loading.gif")
        self.setMovie(self._movie)

    def deleteLater(self):
        self._movie.stop()
        self._movie.deleteLater()
        del self._movie
        super(LoadingWidget, self).deleteLater()

    def show(self):
        self.setVisible(True)
        super(LoadingWidget, self).show()
        self._movie.start()

    def closeEvent(self, event):
        self._movie.stop()
        self.setVisible(False)
        super(LoadingWidget, self).closeEvent(event)


class ClassifyWidget(QTreeWidget):

    fileSelected = pyqtSignal(str)

    def __init__(self, *args, **kwargs):
        super(ClassifyWidget, self).__init__(*args, **kwargs)
        self.setHeaderHidden(True)
        self.setColumnCount(1)
        self.itemDoubleClicked.connect(self.onItemDoubleClicked)
        baseDir = "分类"
        for name in os.listdir(baseDir):
            path = os.path.join(baseDir, name)
            if os.path.isdir(path):
                item = QTreeWidgetItem(self)
                item.setText(0, name)
                for file in os.listdir(path):
                    path = os.path.join(path, file)
                    if os.path.isfile(path) and file.endswith(".json"):
                        item = QTreeWidgetItem(item)
                        item.setText(0, os.path.splitext(file)[0])
                        item.setToolTip(0, path)
        self.expandAll()

    def onItemDoubleClicked(self, item):
        file = item.toolTip(0)
        if file:
            self.fileSelected.emit(file)


class CodeScintilla(QsciScintilla):

    def __init__(self, *args, **kwargs):
        super(CodeScintilla, self).__init__(*args, **kwargs)
        self.init()
        self.linesChanged.connect(self.onLinesChanged)

    def onLinesChanged(self):
        self.setMarginWidth(0, self.fontMetrics().width(str(self.lines())) + 6)

    def init(self):
        self.setUtf8(True)
        lexer = QsciLexerJSON(self)
        self.setLexer(lexer)
        self.setAutoCompletionCaseSensitivity(False)  # 忽略大小写
        self.setAutoCompletionSource(self.AcsAll)
        self.setAutoCompletionThreshold(1)  # 一个字符就弹出补全
        self.setAutoIndent(True)  # 自动缩进
        self.setBackspaceUnindents(True)
        self.setBraceMatching(self.StrictBraceMatch)
        self.setIndentationGuides(True)
        self.setIndentationsUseTabs(False)
        self.setIndentationWidth(4)
        self.setTabIndents(True)
        self.setTabWidth(4)
        self.setWhitespaceSize(1)
        self.setWhitespaceVisibility(self.WsVisible)
        self.setWhitespaceForegroundColor(Qt.gray)
        self.setWrapIndentMode(self.WrapIndentFixed)
        self.setWrapMode(self.WrapWord)
        # 折叠
        self.setFolding(self.BoxedTreeFoldStyle, 2)
        self.setFoldMarginColors(QColor("#676A6C"), QColor("#676A6D"))
        font = self.font() or QFont()
        font.setFamily("Consolas")
        font.setFixedPitch(True)
        font.setPointSize(13)
        self.setFont(font)
        self.setMarginsFont(font)
        self.fontmetrics = QFontMetrics(font)
        lexer.setFont(font)
        self.setMarginWidth(0, self.fontmetrics.width(str(self.lines())) + 6)
        self.setMarginLineNumbers(0, True)
        self.setMarginsBackgroundColor(QColor("gainsboro"))
        self.setMarginWidth(1, 0)
        self.setMarginWidth(2, 14)  # 折叠区域
        # 绑定自动补齐热键Alt+/
        completeKey = QShortcut(QKeySequence(Qt.ALT + Qt.Key_Slash), self)
        completeKey.setContext(Qt.WidgetShortcut)
        completeKey.activated.connect(self.autoCompleteFromAll)


class CodeWidget(QWidget):

    runSignal = pyqtSignal(str)

    def __init__(self, *args, **kwargs):
        super(CodeWidget, self).__init__(*args, **kwargs)
        layout = QVBoxLayout(self, spacing=0)
        layout.setContentsMargins(0, 0, 0, 0)

        self.runButton = QPushButton(
            "运行", self, objectName="RunButton", clicked=self.onRunButton)
        self.codeScintilla = CodeScintilla(self)
        layout.addWidget(self.runButton)
        layout.addWidget(self.codeScintilla)

    def onRunButton(self):
        text = self.text()
        if not text:
            return QMessageBox.information(self, "提示", "json数据不能为空")
        self.runSignal.emit(text)

    def openFile(self, file):
        try:
            with open(file, "rb") as fp:
                text = fp.read()
                encoding = chardet.detect(text) or {}
                encoding = encoding.get(
                    "encoding", "utf-8") or "utf-8"
                text = text.decode(encoding)
                try:
                    text = json.dumps(
                        json.loads(text, encoding=encoding,
                                   object_pairs_hook=OrderedDict),
                        ensure_ascii=False, indent=4)
                except Exception as e:
                    print(e)
                self.setText(text)
        except Exception as e:
            QMessageBox.critical(self, "错误", str(e))

    def setText(self, text):
        self.codeScintilla.setText(text)

    def text(self):
        return self.codeScintilla.text().strip()


class ChartWidget(QWidget):

    def __init__(self, *args, **kwargs):
        super(ChartWidget, self).__init__(*args, **kwargs)
        self.resize(800, 600)
        layout = QVBoxLayout(self)
        layout.setContentsMargins(0, 0, 0, 0)
        self.splitter = QSplitter(self)
        layout.addWidget(self.splitter)
        self.splitter.setOrientation(Qt.Horizontal)
        self.splitter.setStretchFactor(0, 2)
        self.splitter.setStretchFactor(1, 3)
        self.splitter.setStretchFactor(2, 5)
        self.splitter.setAutoFillBackground(True)
        self.splitter.setChildrenCollapsible(False)
        self.splitter.setHandleWidth(2)
        # 分类
        self.classifyWidget = ClassifyWidget(self)
        self.splitter.addWidget(self.classifyWidget)
        # 代码
        self.codeWidget = CodeWidget(self)
        self.splitter.addWidget(self.codeWidget)

        # 等待界面
        self.loadingWidget = LoadingWidget(self, visible=False)
        self.loadingWidget.resize(self.size())

        # 绑定信号槽
        self.classifyWidget.fileSelected.connect(self.codeWidget.openFile)
        self.codeWidget.runSignal.connect(self.onRun)

    def onRun(self, text):
        self.setEnabled(False)
        self.loadingWidget.show()
        # 解析json生成view并添加
        if hasattr(self, "previewView"):
            # 删除旧view
            self.previewView.setParent(None)
            self.previewView.setVisible(False)
            self.previewView.hide()
            self.previewView.deleteLater()
            del self.previewView
        self.previewView = self.getChartView(text)
        if isinstance(self.previewView, QChartView):
            self.splitter.addWidget(self.previewView)
        else:
            QMessageBox.warning(self, "提示", self.previewView)
            del self.previewView
        self.setEnabled(True)
        self.loadingWidget.close()

    def getChartView(self, text):
        try:
            return ChartView(text)
        except Exception as e:
            return str(e)

    def resizeEvent(self, event):
        super(ChartWidget, self).resizeEvent(event)
        self.loadingWidget.resize(self.size())

    def closeEvent(self, event):
        self.loadingWidget.close()
        self.loadingWidget.deleteLater()
        del self.loadingWidget
        super(ChartWidget, self).closeEvent(event)


if __name__ == "__main__":
    app = QApplication(sys.argv)
    w = ChartWidget()
    w.show()
    sys.exit(app.exec_())