class WeCaseWindow(QtGui.QMainWindow, Ui_frm_MainWindow):
client = None
uid = None
imageLoaded = QtCore.pyqtSignal(str)
tabTextChanged = QtCore.pyqtSignal(int, str)
def __init__(self, parent=None):
super(WeCaseWindow, self).__init__(parent)
self.setupUi(self)
self.tweetViews = [self.homeView, self.mentionsView,
self.commentsView, self.myView]
self.client = const.client
self.loadConfig()
self.setupModels()
self.init_account()
self.IMG_AVATAR = -2
self.IMG_THUMB = -1
self.notify = Notify(timeout=self.notify_timeout)
self.applyConfig()
self.download_lock = []
def init_account(self):
self.get_uid()
def loadConfig(self):
self.config = WeCaseConfig(const.config_path)
self.notify_interval = self.config.notify_interval
self.notify_timeout = self.config.notify_timeout
self.usersBlacklist = self.config.usersBlacklist
self.tweetKeywordsBlacklist = self.config.tweetsKeywordsBlacklist
self.remindMentions = self.config.remind_mentions
self.remindComments = self.config.remind_comments
def applyConfig(self):
try:
self.timer.stop_event.set()
except AttributeError:
pass
self.timer = WTimer(self.notify_interval, self.show_notify)
self.timer.start()
self.notify.timeout = self.notify_timeout
def setupModels(self):
self.all_timeline = TweetCommonModel(self.client.statuses.home_timeline, self)
self.all_timeline.setUsersBlacklist(self.usersBlacklist)
self.all_timeline.setTweetsKeywordsBlacklist(self.tweetKeywordsBlacklist)
self.all_timeline.load()
self.homeView.setModel(self.all_timeline)
self.mentions = TweetCommonModel(self.client.statuses.mentions, self)
self.mentions.setUsersBlacklist(self.usersBlacklist)
self.mentions.setTweetsKeywordsBlacklist(self.tweetKeywordsBlacklist)
self.mentions.load()
self.mentionsView.setModel(self.mentions)
self.comment_to_me = TweetCommentModel(self.client.comments.to_me, self)
self.comment_to_me.setUsersBlacklist(self.usersBlacklist)
self.comment_to_me.setTweetsKeywordsBlacklist(self.tweetKeywordsBlacklist)
self.comment_to_me.load()
self.commentsView.setModel(self.comment_to_me)
self.my_timeline = TweetCommonModel(self.client.statuses.user_timeline, self)
self.my_timeline.setUsersBlacklist(self.usersBlacklist)
self.my_timeline.setTweetsKeywordsBlacklist(self.tweetKeywordsBlacklist)
self.my_timeline.load()
self.myView.setModel(self.my_timeline)
def reset_remind(self):
if self.tabWidget.currentIndex() == 0:
self.tabWidget.setTabText(0, self.tr("Weibo"))
elif self.tabWidget.currentIndex() == 1:
self.client.remind.set_count.post(type="mention_status")
self.tabWidget.setTabText(1, self.tr("@Me"))
elif self.tabWidget.currentIndex() == 2:
self.client.remind.set_count.post(type="cmt")
self.tabWidget.setTabText(2, self.tr("Comments"))
def get_remind(self, uid):
"""this function is used to get unread_count
from Weibo API. uid is necessary."""
while 1:
try:
reminds = self.client.remind.unread_count.get(uid=uid)
break
except http.client.BadStatusLine:
sleep(0.2)
continue
return reminds
def get_uid(self):
"""How can I get my uid? here it is"""
try:
self.uid = self.client.account.get_uid.get().uid
except AttributeError:
return None
def show_notify(self):
# This function is run in another thread by WTimer.
# Do not modify UI directly. Send signal and react it in a slot only.
# We use SIGNAL self.tabTextChanged and SLOT self.setTabText()
# to display unread count
reminds = self.get_remind(self.uid)
msg = self.tr("You have:") + "\n"
num_msg = 0
if reminds['status'] != 0:
# Note: do NOT send notify here, or users will crazy.
self.tabTextChanged.emit(0, self.tr("Weibo(%d)")
% reminds['status'])
if reminds['mention_status'] and self.remindMentions:
msg += self.tr("%d unread @ME") % reminds['mention_status'] + "\n"
self.tabTextChanged.emit(1, self.tr("@Me(%d)")
% reminds['mention_status'])
num_msg += 1
if reminds['cmt'] and self.remindComments:
msg += self.tr("%d unread comment(s)") % reminds['cmt'] + "\n"
self.tabTextChanged.emit(2, self.tr("Comments(%d)")
% reminds['cmt'])
num_msg += 1
if num_msg:
self.notify.showMessage(self.tr("WeCase"), msg)
def setTabText(self, index, string):
self.tabWidget.setTabText(index, string)
def moveToTop(self):
self.get_current_tweetView().moveToTop()
def setLoaded(self):
pass
def showSettings(self):
wecase_settings = WeSettingsWindow()
if wecase_settings.exec_():
self.loadConfig()
self.applyConfig()
def showAbout(self):
wecase_about = AboutWindow()
wecase_about.exec_()
def logout(self):
self.close()
# This is a model dialog, if we exec it before we close MainWindow
# MainWindow won't close
from LoginWindow import LoginWindow
wecase_login = LoginWindow(allow_auto_login=False)
wecase_login.exec_()
def postTweet(self):
wecase_new = NewpostWindow()
wecase_new.exec_()
def refresh(self):
tweetView = self.get_current_tweetView()
tweetView.model().timelineLoaded.connect(self.moveToTop)
tweetView.refresh()
self.reset_remind()
def get_current_tweetView(self):
tweetViews = {0: self.homeView, 1: self.mentionsView,
2: self.commentsView, 3: self.myView}
return tweetViews[self.tabWidget.currentIndex()]
def closeEvent(self, event):
self.timer.stop_event.set()
