def __init__(self, app, logger):
# 同時に１起動制限
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())
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):
# 同時に１起動制限
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"])