Example #1
0
    def __init__(self, app, logger):
        # 同時に1起動制限
        self.firstApp = False

        t = threading.Thread(target=self.checkSemaphore)
        t.start()
        t.join(0.2)

        if not self.firstApp:
            sys.exit(1)

        # 初期化処理
        QtGui.QDialog.__init__(self)
        self.app = app
        self.logger = logger

        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # load settings
        try:
            f = open(self.SETTINGS_FILE_NAME, "rb")
            self.settings = pickle.load(f)
            f.close()
            for k, v in self.INIT_SETTINGS.items():
                if not self.settings.has_key(k):
                    self.settings[k] = v
        except:
            self.settings = self.INIT_SETTINGS

        # models
        self.dicTableModel = NicoDicTableModel(self)
        self.liveTableModel = NicoLiveTableModel(self)
        self.watchListTableModel = WatchListTableModel(self)
        self.communityListTableModel = CommunityTableModel(self)

        # tab widget
        self.tabWidget = DraggableTabWidget(self.ui.centralwidget)
        self.tabWidget.setLayoutDirection(QtCore.Qt.LeftToRight)
        self.ui.gridLayout.addWidget(self.tabWidget, 0, 0, 1, 1)
        self.connect(self.tabWidget, QtCore.SIGNAL("currentChanged(int)"), self.tabWidgetChangedHandler)

        # initial tabs
        DicUserTabWidget(self)
        LiveUserTabWidget(self)
        WatchListUserTabWidget(self)
        CommunityListUserTabWidget(self)
        self.tabWidget.setCurrentIndex(0)

        # trayIcon/trayIconMenu/trayIconImg
        self.trayIconImg = QtGui.QIcon(self.tr(":/dic.ico"))
        self.trayIconMenu = QtGui.QMenu(self)
        self.trayIconMenu.addAction(u"表示/非表示", lambda: self.toggleWindowVisibility())
        self.trayIconMenu.addAction(u"終了", lambda: self.app.quit())
        self.trayIcon = QtGui.QSystemTrayIcon(self)
        self.trayIcon.setContextMenu(self.trayIconMenu)
        self.trayIcon.setIcon(self.trayIconImg)
        self.trayIcon.show()
        self.connect(self.trayIcon, QtCore.SIGNAL("activated(QSystemTrayIcon::ActivationReason)"), self.trayIconHandler)

        # window style
        self.setWindowFlags(self.windowFlags() & ~QtCore.Qt.WindowMinimizeButtonHint)

        # first data fetch
        self.nicopoll = NicoPoll(self.dicTableModel, self.liveTableModel)
        self.nicopoll.fetch(self)

        # menu
        self.fileMenu = QtGui.QMenu(self.ui.menubar)
        self.fileMenu.addAction(u"設定", lambda: self.showSettingsDialog())
        self.fileMenu.addSeparator()
        self.fileMenu.addAction(u"トレイアイコンに収納", lambda: self.hide())
        self.fileMenu.addAction(u"終了", lambda: self.app.quit())
        self.fileMenu.setTitle(self.trUtf8("ファイル"))
        self.ui.menubar.addAction(self.fileMenu.menuAction())
Example #2
0
class MainWindow(QtGui.QMainWindow):
    POLLING_DURATION = 10000  # 10000msec = 10sec
    SETTINGS_FILE_NAME = "nicopealert.dat"
    INIT_SETTINGS = {
        "version": "0.0.1",  # 設定ファイルが互換性がなくなるときに変更
        "watchList": {},
        "communityList": {},
        "tabList": [],
        "browserOpenMode": 0,
    }

    def checkSemaphore(self):
        self.sem = QtCore.QSystemSemaphore("nicopealert-app", 1)
        self.sem.acquire()
        self.firstApp = True
        self.inited = False

    def __init__(self, app, logger):
        # 同時に1起動制限
        self.firstApp = False

        t = threading.Thread(target=self.checkSemaphore)
        t.start()
        t.join(0.2)

        if not self.firstApp:
            sys.exit(1)

        # 初期化処理
        QtGui.QDialog.__init__(self)
        self.app = app
        self.logger = logger

        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        # load settings
        try:
            f = open(self.SETTINGS_FILE_NAME, "rb")
            self.settings = pickle.load(f)
            f.close()
            for k, v in self.INIT_SETTINGS.items():
                if not self.settings.has_key(k):
                    self.settings[k] = v
        except:
            self.settings = self.INIT_SETTINGS

        # models
        self.dicTableModel = NicoDicTableModel(self)
        self.liveTableModel = NicoLiveTableModel(self)
        self.watchListTableModel = WatchListTableModel(self)
        self.communityListTableModel = CommunityTableModel(self)

        # tab widget
        self.tabWidget = DraggableTabWidget(self.ui.centralwidget)
        self.tabWidget.setLayoutDirection(QtCore.Qt.LeftToRight)
        self.ui.gridLayout.addWidget(self.tabWidget, 0, 0, 1, 1)
        self.connect(self.tabWidget, QtCore.SIGNAL("currentChanged(int)"), self.tabWidgetChangedHandler)

        # initial tabs
        DicUserTabWidget(self)
        LiveUserTabWidget(self)
        WatchListUserTabWidget(self)
        CommunityListUserTabWidget(self)
        self.tabWidget.setCurrentIndex(0)

        # trayIcon/trayIconMenu/trayIconImg
        self.trayIconImg = QtGui.QIcon(self.tr(":/dic.ico"))
        self.trayIconMenu = QtGui.QMenu(self)
        self.trayIconMenu.addAction(u"表示/非表示", lambda: self.toggleWindowVisibility())
        self.trayIconMenu.addAction(u"終了", lambda: self.app.quit())
        self.trayIcon = QtGui.QSystemTrayIcon(self)
        self.trayIcon.setContextMenu(self.trayIconMenu)
        self.trayIcon.setIcon(self.trayIconImg)
        self.trayIcon.show()
        self.connect(self.trayIcon, QtCore.SIGNAL("activated(QSystemTrayIcon::ActivationReason)"), self.trayIconHandler)

        # window style
        self.setWindowFlags(self.windowFlags() & ~QtCore.Qt.WindowMinimizeButtonHint)

        # first data fetch
        self.nicopoll = NicoPoll(self.dicTableModel, self.liveTableModel)
        self.nicopoll.fetch(self)

        # menu
        self.fileMenu = QtGui.QMenu(self.ui.menubar)
        self.fileMenu.addAction(u"設定", lambda: self.showSettingsDialog())
        self.fileMenu.addSeparator()
        self.fileMenu.addAction(u"トレイアイコンに収納", lambda: self.hide())
        self.fileMenu.addAction(u"終了", lambda: self.app.quit())
        self.fileMenu.setTitle(self.trUtf8("ファイル"))
        self.ui.menubar.addAction(self.fileMenu.menuAction())

    def show(self):
        QtGui.QMainWindow.show(self)
        self.activateWindow()

        if not self.inited:
            # データ挿入
            self.watchListTableModel.appendItems(self.settings["watchList"])
            self.communityListTableModel.appendItems(self.settings["communityList"])
            for tabcond in self.settings["tabList"]:
                if tabcond["class"] == u"DicUserTabWidget":
                    t = DicUserTabWidget(self, False)
                    t.setCond(tabcond)
                elif tabcond["class"] == u"LiveUserTabWidget":
                    t = LiveUserTabWidget(self, False)
                    t.setCond(tabcond)

            # タブの見た目関係初期化
            for i in xrange(0, self.tabWidget.count()):
                w = self.tabWidget.widget(i)
                w.init_after_show()

            self.inited = True

        # set timer for polling
        self.timer = QtCore.QTimer(self)
        self.connect(self.timer, QtCore.SIGNAL("timeout()"), self.timer_handler)
        self.timer.setInterval(self.POLLING_DURATION)
        self.timer.start()

    def timer_handler(self):
        self.nicopoll.fetch(self)

        #

    def showVersionUpDialog(self):
        # クライアントバージョンが古い
        msg = QtGui.QMessageBox.critical(
            self,
            self.trUtf8("バージョンアップのお知らせ"),
            self.trUtf8("ニコ百アラートがバージョンアップしました。お使いのバージョンは今後利用できなくなります。お手数ですが、バージョンアップお願いいたします。"),
            self.trUtf8("ダウンロードページを開く"),
            self.trUtf8("終了する"),
        )
        if msg == 0:
            import webbrowser

            webbrowser.open("http://dic.nicovideo.jp/nicopealert/")
        sys.exit(2)

    def toggleWindowVisibility(self):
        if self.isVisible():
            self.hide()
        else:
            self.show()

    def trayIconHandler(self, reason):
        # NOTE: Macではダメっぽい
        if reason == QtGui.QSystemTrayIcon.DoubleClick:
            self.toggleWindowVisibility()

    def tabWidgetChangedHandler(self, index):
        self.tabWidget.currentWidget().setTabNotify(False)

    def appendWatchList(self, category, title, view_title):
        key = u"%s%s" % (category, title)
        i = {"category": category, "title": title, "view_title": view_title}
        self.watchListTableModel.appendItems({key: i})
        self.settings["watchList"][key] = i
        self.saveSettings()

    def appendCommunityList(self, com_id, com_name):
        # NOTE: com_idはkeyにもvalueにも入っている。
        u = {"id": com_id, "name": com_name}
        self.communityListTableModel.appendItems({com_id: u})
        self.settings["communityList"][com_id] = u
        self.saveSettings()

    # 各タブの情報を保存する
    def saveTabs(self):
        tabList = []
        for i in xrange(0, self.tabWidget.count()):
            w = self.tabWidget.widget(i)
            # 初期タブ以外を保存
            if not w.initial:
                tabList.append(w.cond())
        self.settings["tabList"] = tabList
        self.saveSettings()

    # NOTE: 小汚い
    def removeWatchList(self, row):
        category, title = map(lambda d: unicode(d.toString()), self.watchListTableModel.raw_row_data(row)[0:2])
        key = u"%s%s" % (category, title)
        self.watchListTableModel.removeRow(row)
        del self.settings["watchList"][key]
        self.saveSettings()

    # NOTE: 小汚い
    def removeCommunityList(self, row):
        com_id = unicode(
            self.communityListTableModel.raw_row_data(row)[self.communityListTableModel.COL_COM_ID_INDEX].toString()
        )
        self.communityListTableModel.removeRow(row)
        del self.settings["communityList"][com_id]
        self.saveSettings()

    def saveSettings(self):
        try:
            tmpfilename = self.SETTINGS_FILE_NAME + ".tmp"
            f = open(tmpfilename, "wb")
            pickle.dump(self.settings, f)
            f.close()
            if os.name == "nt":
                try:
                    os.remove(self.SETTINGS_FILE_NAME)
                except:
                    pass
            os.rename(tmpfilename, self.SETTINGS_FILE_NAME)
        except:
            pass

    def closeEvent(self, event):
        # ×ボタンで、タスクトレイバーのみになる
        self.hide()
        event.ignore()

    def showSettingsDialog(self):
        d = SettingsDialog()
        d.load(self)
        d.show()

    def browserOpen(self, url):
        print self.settings["browserOpenMode"]
        webbrowser.open(url, self.settings["browserOpenMode"])