def _create_tweetWidget(self):
if self.action == "comment" and self.tweet.comments_count:
self.tweetWidget = SingleTweetWidget(self.tweet, ["image", "original"], self)
self.replyModel = TweetUnderCommentModel(self.client.api("comments/show"), self.tweet.id, self)
elif self.action == "retweet" and self.tweet.retweets_count:
if self.tweet:
self.tweetWidget = SingleTweetWidget(self.tweet, ["image", "original"], self)
elif self.tweet.original:
self.tweetWidget = SingleTweetWidget(self.tweet.original, ["image", "original"], self)
self.replyModel = TweetRetweetModel(self.client.api("statuses/repost_timeline"), self.tweet.id, self)
else:
return
self.replyModel.load()
self.splitter = QtGui.QSplitter(self)
self.splitter.setOrientation(QtCore.Qt.Vertical)
self.verticalLayout.insertWidget(0, self.splitter)
self.tweetWidget.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
self.splitter.addWidget(self.tweetWidget)
self.commentsWidget = TweetListWidget(self, ["image", "original"])
self.commentsWidget.setModel(self.replyModel)
self.commentsWidget.scrollArea.setMinimumSize(20, 200)
self.commentsWidget.userClicked.connect(self.userClicked)
self.commentsWidget.tagClicked.connect(self.tagClicked)
self.splitter.addWidget(self.commentsWidget)
self.splitter.addWidget(self.textEdit)
def _setupTopicTab(self, topic, switch=True):
index = self._getSameTab("topic", topic)
if index:
if switch:
self.tabWidget.setCurrentIndex(index)
return
view = TweetListWidget()
timeline = TweetTopicModel(self.client.api("search/topics"), topic,
view)
tab = self._setupCommonTab(timeline, view, switch, protect=False)
self._setTabIcon(
tab,
WObjectCache().open(QtGui.QPixmap, ":/IMG/img/topic.jpg"))
def _create_tweetWidget(self):
if self.action == "comment" and self.tweet.comments_count:
self.tweetWidget = SingleTweetWidget(self.tweet,
["image", "original"], self)
self.replyModel = TweetUnderCommentModel(
self.client.api("comments/show"), self.tweet.id, self)
elif self.action == "retweet" and self.tweet.retweets_count:
if self.tweet:
self.tweetWidget = SingleTweetWidget(self.tweet,
["image", "original"],
self)
elif self.tweet.original:
self.tweetWidget = SingleTweetWidget(self.tweet.original,
["image", "original"],
self)
self.replyModel = TweetRetweetModel(
self.client.api("statuses/repost_timeline"), self.tweet.id,
self)
else:
return
self.replyModel.load()
self.splitter = QtGui.QSplitter(self)
self.splitter.setOrientation(QtCore.Qt.Vertical)
self.verticalLayout.insertWidget(0, self.splitter)
self.tweetWidget.setSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Expanding)
self.splitter.addWidget(self.tweetWidget)
self.commentsWidget = TweetListWidget(self, ["image", "original"])
self.commentsWidget.setModel(self.replyModel)
self.commentsWidget.scrollArea.setMinimumSize(20, 200)
self.commentsWidget.userClicked.connect(self.userClicked)
self.commentsWidget.tagClicked.connect(self.tagClicked)
self.splitter.addWidget(self.commentsWidget)
self.splitter.addWidget(self.textEdit)
def _setupUserTab(self, uid, switch=True, myself=False):
index = self._getSameTab("uid", uid)
if index:
if switch:
self.tabWidget.setCurrentIndex(index)
return
view = TweetListWidget()
_timeline = TweetUserModel(self.client.api("statuses/user_timeline"),
uid, view)
timeline = TweetFilterModel(_timeline)
timeline.setModel(_timeline)
tab = self._setupCommonTab(timeline, view, switch, myself)
def setAvatar(f):
self._setTabIcon(tab, WObjectCache().open(QtGui.QPixmap, f))
fetcher = AsyncFetcher("".join(
(path.cache_path, str(self.info["uid"]))))
fetcher.addTask(
self.client.api("users/show").get(uid=uid)["profile_image_url"],
setAvatar)
class NewpostWindow(QtGui.QDialog, Ui_NewPostWindow):
image = None
apiError = QtCore.pyqtSignal(Exception)
commonError = QtCore.pyqtSignal(str, str)
sendSuccessful = QtCore.pyqtSignal()
userClicked = QtCore.pyqtSignal(UserItem, bool)
tagClicked = QtCore.pyqtSignal(str, bool)
tweetRefreshed = QtCore.pyqtSignal()
tweetRejected = QtCore.pyqtSignal()
def __init__(self, action="new", tweet=None, parent=None):
super(NewpostWindow, self).__init__(parent)
self.setAttribute(QtCore.Qt.WA_QuitOnClose, False)
self.client = const.client
self.tweet = tweet
self.action = action
self.setupUi(self)
self.textEdit.callback = self.mentions_suggest
self.textEdit.mention_flag = "@"
self.notify = Notify(timeout=1)
self._sent = False
self.apiError.connect(self.showException)
self.sendSuccessful.connect(self.sent)
self.commonError.connect(self.showErrorMessage)
self.errorWindow = APIErrorWindow(self)
if self.action not in ["new", "reply"]:
self.tweetRefreshed.connect(self._create_tweetWidget)
self.tweetRejected.connect(lambda: self.pushButton_send.setEnabled(False))
self._refresh()
def setupUi(self, widget):
super(NewpostWindow, self).setupUi(widget)
self.sendAction = QtGui.QAction(self)
self.sendAction.triggered.connect(self.send)
self.sendAction.setShortcut(QtGui.QKeySequence("Ctrl+Return"))
self.addAction(self.sendAction)
self.checkChars()
self.setupButtons()
def setupButtons(self):
# Disabled is the default state of buttons
self.pushButton_picture.setEnabled(False)
self.chk_repost.setEnabled(False)
self.chk_comment.setEnabled(False)
self.chk_comment_original.setEnabled(False)
if self.action == "new":
assert (not self.tweet) # Shouldn't have a tweet object.
self.pushButton_picture.setEnabled(True)
elif self.action == "retweet":
self.chk_comment.setEnabled(True)
if self.tweet.type == TweetItem.RETWEET:
self.textEdit.setText(self.tweet.append_existing_replies())
self.chk_comment_original.setEnabled(True)
elif self.action == "comment":
self.chk_repost.setEnabled(True)
if self.tweet.type == TweetItem.RETWEET:
self.chk_comment_original.setEnabled(True)
elif self.action == "reply":
self.chk_repost.setEnabled(True)
if self.tweet.original.type == TweetItem.RETWEET:
self.chk_comment_original.setEnabled(True)
else:
assert False
@async
def _refresh(self):
# The read count is not a real-time value. So refresh it now.
try:
self.tweet.refresh()
except APIError as e:
self.errorWindow.raiseException.emit(e)
self.tweetRejected.emit()
return
self.tweetRefreshed.emit()
def _create_tweetWidget(self):
if self.action == "comment" and self.tweet.comments_count:
self.tweetWidget = SingleTweetWidget(self.tweet, ["image", "original"], self)
self.replyModel = TweetUnderCommentModel(self.client.api("comments/show"), self.tweet.id, self)
elif self.action == "retweet" and self.tweet.retweets_count:
if self.tweet:
self.tweetWidget = SingleTweetWidget(self.tweet, ["image", "original"], self)
elif self.tweet.original:
self.tweetWidget = SingleTweetWidget(self.tweet.original, ["image", "original"], self)
self.replyModel = TweetRetweetModel(self.client.api("statuses/repost_timeline"), self.tweet.id, self)
else:
return
self.replyModel.load()
self.splitter = QtGui.QSplitter(self)
self.splitter.setOrientation(QtCore.Qt.Vertical)
self.verticalLayout.insertWidget(0, self.splitter)
self.tweetWidget.setSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding)
self.splitter.addWidget(self.tweetWidget)
self.commentsWidget = TweetListWidget(self, ["image", "original"])
self.commentsWidget.setModel(self.replyModel)
self.commentsWidget.scrollArea.setMinimumSize(20, 200)
self.commentsWidget.userClicked.connect(self.userClicked)
self.commentsWidget.tagClicked.connect(self.tagClicked)
self.splitter.addWidget(self.commentsWidget)
self.splitter.addWidget(self.textEdit)
def mentions_suggest(self, text):
ret_users = []
try:
word = text.split(' ')[-1]
word = word.split('@')[-1]
except IndexError:
return []
if not word.strip():
return []
users = self.client.api("search/suggestions/at_users").get(q=word, type=0)
for user in users:
ret_users.append("@" + user['nickname'])
return ret_users
def sent(self):
self._sent = True
self.close()
def send(self):
self.pushButton_send.setEnabled(False)
if self.action == "new":
self.new()
elif self.action == "retweet":
self.retweet()
elif self.action == "comment":
self.comment()
elif self.action == "reply":
self.reply()
else:
# If action is in other types, it must be a mistake.
assert False
@async
def retweet(self):
text = self.textEdit.toPlainText()
comment = int(self.chk_comment.isChecked())
comment_ori = int(self.chk_comment_original.isChecked())
try:
self.tweet.retweet(text, comment, comment_ori)
except APIError as e:
self.apiError.emit(e)
return
self.notify.showMessage(self.tr("WeCase"), self.tr("Retweet Success!"))
self.sendSuccessful.emit()
@async
def comment(self):
text = self.textEdit.toPlainText()
retweet = int(self.chk_repost.isChecked())
comment_ori = int(self.chk_comment_original.isChecked())
try:
self.tweet.comment(text, comment_ori, retweet)
except APIError as e:
self.apiError.emit(e)
return
self.notify.showMessage(self.tr("WeCase"), self.tr("Comment Success!"))
self.sendSuccessful.emit()
@async
def reply(self):
text = self.textEdit.toPlainText()
comment_ori = int(self.chk_comment_original.isChecked())
retweet = int(self.chk_repost.isChecked())
try:
self.tweet.reply(text, comment_ori, retweet)
except APIError as e:
self.apiError.emit(e)
return
self.notify.showMessage(self.tr("WeCase"), self.tr("Reply Success!"))
self.sendSuccessful.emit()
@async
def new(self):
text = self.textEdit.toPlainText()
try:
if self.image:
try:
if getsize(self.image) > const.MAX_IMAGE_BYTES:
raise ValueError
with open(self.image, "rb") as image:
self.client.api("statuses/upload").post(status=text, pic=image)
except (OSError, IOError):
self.commonError.emit(self.tr("File not found"),
self.tr("No such file: %s")
% self.image)
self.addImage() # In fact, remove image...
return
except ValueError:
self.commonError.emit(self.tr("Too large size"),
self.tr("This image is too large to upload: %s")
% self.image)
self.addImage() # In fact, remove image...
return
else:
self.client.api("statuses/update").post(status=text)
self.notify.showMessage(self.tr("WeCase"),
self.tr("Tweet Success!"))
self.sendSuccessful.emit()
except APIError as e:
self.apiError.emit(e)
return
self.image = None
def addImage(self):
ACCEPT_TYPE = self.tr("Images") + "(*.png *.jpg *.jpeg *.bmp *.gif)"
if self.image:
self.image = None
self.pushButton_picture.setText(self.tr("&Picture"))
else:
self.image = QtGui.QFileDialog.getOpenFileName(self,
self.tr("Choose a"
" image"),
filter=ACCEPT_TYPE)
# user may cancel the dialog, so check again
if self.image:
self.pushButton_picture.setText(self.tr("Remove the picture"))
self.textEdit.setFocus()
def showException(self, e):
if "Text too long" in str(e):
QtGui.QMessageBox.warning(None, self.tr("Text too long!"),
self.tr("Please remove some text."))
else:
self.errorWindow.raiseException.emit(e)
self.pushButton_send.setEnabled(True)
def showErrorMessage(self, title, text):
QtGui.QMessageBox.warning(self, title, text)
def showSmiley(self):
wecase_smiley = FaceWindow()
if wecase_smiley.exec_():
self.textEdit.textCursor().insertText(wecase_smiley.faceName)
self.textEdit.setFocus()
def checkChars(self):
"""Check textEdit's characters.
If it larger than 140, Send Button will be disabled
and label will show red chars."""
text = self.textEdit.toPlainText()
numLens = 140 - tweetLength(text)
if numLens == 140 and (not self.action == "retweet"):
# you can not send empty tweet, except retweet
self.pushButton_send.setEnabled(False)
elif numLens >= 0:
# length is okay
self.label.setStyleSheet("color:black;")
self.pushButton_send.setEnabled(True)
else:
# text is too long
self.label.setStyleSheet("color:red;")
self.pushButton_send.setEnabled(False)
self.label.setText(str(numLens))
def reject(self):
self.close()
def closeEvent(self, event):
# We have unsend text.
if (not self._sent) and (self.textEdit.toPlainText()):
choice = QtGui.QMessageBox.question(
self, self.tr("Close?"),
self.tr("All unpost text will lost."),
QtGui.QMessageBox.Yes,
QtGui.QMessageBox.No)
if choice == QtGui.QMessageBox.No:
event.ignore()
else:
self.setParent(None)
class WeCaseWindow(QtGui.QMainWindow):
client = None
uid = None
timelineLoaded = QtCore.pyqtSignal(int)
imageLoaded = QtCore.pyqtSignal(str)
tabBadgeChanged = QtCore.pyqtSignal(int, int)
tabAvatarFetched = QtCore.pyqtSignal(str)
def __init__(self, parent=None):
super(WeCaseWindow, self).__init__(parent)
self.errorWindow = APIErrorWindow(self)
self.setAttribute(QtCore.Qt.WA_QuitOnClose, True)
self._iconPixmap = {}
self.setupUi(self)
self._setupSysTray()
self.tweetViews = [self.homeView, self.mentionsView, self.commentsView, self.commentsMentionsTab]
self.info = WeRuntimeInfo()
self.client = const.client
self.loadConfig()
self.init_account()
self.setupModels()
self.IMG_AVATAR = -2
self.IMG_THUMB = -1
self.notify = Notify(timeout=self.notify_timeout)
self.applyConfig()
self.download_lock = []
self._last_reminds_count = 0
self._setupUserTab(self.uid(), False, True)
def _setupTab(self, view):
tab = QtGui.QWidget()
layout = QtGui.QGridLayout(tab)
layout.addWidget(view)
view.setParent(tab)
return tab
def _setupCommonTab(self, timeline, view, switch=True, protect=False):
self._prepareTimeline(timeline)
view.setModel(timeline)
view.userClicked.connect(self.userClicked)
view.tagClicked.connect(self.tagClicked)
tab = self._setupTab(view)
self.tabWidget.addTab(tab, "")
if switch:
self.tabWidget.setCurrentWidget(tab)
if protect:
self.tabWidget.tabBar().setProtectTab(tab, True)
return tab
def _getSameTab(self, attr, value):
for i in range(self.tabWidget.count()):
try:
tab = self.tabWidget.widget(i).layout().itemAt(0).widget()
_value = getattr(tab.model(), attr)()
if _value == value:
return i
except AttributeError:
pass
return False
def _setupUserTab(self, uid, switch=True, myself=False):
index = self._getSameTab("uid", uid)
if index:
if switch:
self.tabWidget.setCurrentIndex(index)
return
view = TweetListWidget()
_timeline = TweetUserModel(self.client.statuses.user_timeline, uid,
view)
timeline = TweetFilterModel(_timeline)
timeline.setModel(_timeline)
tab = self._setupCommonTab(timeline, view, switch, myself)
def setAvatar(f):
self._setTabIcon(tab, WObjectCache().open(QtGui.QPixmap, f))
fetcher = AsyncFetcher("".join((path.cache_path, str(self.info["uid"]))))
fetcher.addTask(self.client.users.show.get(uid=uid)["profile_image_url"], setAvatar)
def _setupTopicTab(self, topic, switch=True):
index = self._getSameTab("topic", topic)
if index:
if switch:
self.tabWidget.setCurrentIndex(index)
return
view = TweetListWidget()
timeline = TweetTopicModel(self.client.search.topics, topic, view)
tab = self._setupCommonTab(timeline, view, switch, protect=False)
self._setTabIcon(tab, WObjectCache().open(
QtGui.QPixmap, ":/IMG/img/topic.jpg"
))
def userClicked(self, userItem, openAtBackend):
try:
self._setupUserTab(userItem.id, switch=(not openAtBackend))
except APIError as e:
self.errorWindow.raiseException.emit(e)
def tagClicked(self, str, openAtBackend):
try:
self._setupTopicTab(str, switch=(not openAtBackend))
except APIError as e:
self.errorWindow.raiseException.emit(e)
def setupUi(self, mainWindow):
mainWindow.setWindowIcon(QtGui.QIcon(":/IMG/img/WeCase.svg"))
mainWindow.setDocumentMode(False)
mainWindow.setDockOptions(QtGui.QMainWindow.AllowTabbedDocks |
QtGui.QMainWindow.AnimatedDocks)
self.centralwidget = QtGui.QWidget(mainWindow)
self.verticalLayout = QtGui.QVBoxLayout(self.centralwidget)
self.tabWidget = QtGui.QTabWidget(self.centralwidget)
self.tabWidget.setTabBar(WTabBar(self.tabWidget))
self.tabWidget.setTabPosition(QtGui.QTabWidget.West)
self.tabWidget.setTabShape(QtGui.QTabWidget.Rounded)
self.tabWidget.setDocumentMode(False)
self.tabWidget.setMovable(True)
self.tabWidget.tabCloseRequested.connect(self.closeTab)
self.homeView = TweetListWidget()
self.homeView.userClicked.connect(self.userClicked)
self.homeView.tagClicked.connect(self.tagClicked)
self.homeTab = self._setupTab(self.homeView)
self.tabWidget.addTab(self.homeTab, "")
self.tabWidget.tabBar().setProtectTab(self.homeTab, True)
self.mentionsView = TweetListWidget()
self.mentionsView.userClicked.connect(self.userClicked)
self.mentionsView.tagClicked.connect(self.tagClicked)
self.mentionsTab = self._setupTab(self.mentionsView)
self.tabWidget.addTab(self.mentionsTab, "")
self.tabWidget.tabBar().setProtectTab(self.mentionsTab, True)
self.commentsView = TweetListWidget()
self.commentsView.userClicked.connect(self.userClicked)
self.commentsView.tagClicked.connect(self.tagClicked)
self.commentsTab = self._setupTab(self.commentsView)
self.tabWidget.addTab(self.commentsTab, "")
self.tabWidget.tabBar().setProtectTab(self.commentsTab, True)
self.commentsMentionsView = TweetListWidget()
self.commentsMentionsView.userClicked.connect(self.userClicked)
self.commentsMentionsView.tagClicked.connect(self.tagClicked)
self.commentsMentionsTab = self._setupTab(self.commentsMentionsView)
self.tabWidget.addTab(self.commentsMentionsTab, "")
self.tabWidget.tabBar().setProtectTab(self.commentsMentionsTab, True)
self.verticalLayout.addWidget(self.tabWidget)
self.widget = QtGui.QWidget(self.centralwidget)
self.verticalLayout.addWidget(self.widget)
mainWindow.setCentralWidget(self.centralwidget)
self.aboutAction = QtGui.QAction(mainWindow)
self.refreshAction = QtGui.QAction(mainWindow)
self.logoutAction = QtGui.QAction(mainWindow)
self.exitAction = QtGui.QAction(mainWindow)
self.settingsAction = QtGui.QAction(mainWindow)
self.aboutAction.setIcon(QtGui.QIcon(QtGui.QPixmap("./IMG/img/where_s_my_weibo.svg")))
self.exitAction.setIcon(QtGui.QIcon(QtGui.QPixmap(":/IMG/img/application-exit.svg")))
self.settingsAction.setIcon(QtGui.QIcon(QtGui.QPixmap(":/IMG/img/preferences-other.png")))
self.refreshAction.setIcon(QtGui.QIcon(QtGui.QPixmap(":/IMG/img/refresh.png")))
self.menubar = QtGui.QMenuBar(mainWindow)
self.menubar.setEnabled(True)
self.menubar.setDefaultUp(False)
mainWindow.setMenuBar(self.menubar)
self.mainMenu = QtGui.QMenu(self.menubar)
self.helpMenu = QtGui.QMenu(self.menubar)
self.optionsMenu = QtGui.QMenu(self.menubar)
self.mainMenu.addAction(self.refreshAction)
self.mainMenu.addSeparator()
self.mainMenu.addAction(self.logoutAction)
self.mainMenu.addAction(self.exitAction)
self.helpMenu.addAction(self.aboutAction)
self.optionsMenu.addAction(self.settingsAction)
self.menubar.addAction(self.mainMenu.menuAction())
self.menubar.addAction(self.optionsMenu.menuAction())
self.menubar.addAction(self.helpMenu.menuAction())
self.exitAction.triggered.connect(mainWindow.close)
self.aboutAction.triggered.connect(mainWindow.showAbout)
self.settingsAction.triggered.connect(mainWindow.showSettings)
self.logoutAction.triggered.connect(mainWindow.logout)
self.refreshAction.triggered.connect(mainWindow.refresh)
self.pushButton_refresh = QtGui.QPushButton(self.widget)
self.pushButton_new = QtGui.QPushButton(self.widget)
self.pushButton_refresh.clicked.connect(mainWindow.refresh)
self.pushButton_new.clicked.connect(mainWindow.postTweet)
self.timelineLoaded.connect(self.moveToTop)
self.tabBadgeChanged.connect(self.drawNotifyBadge)
self.refreshAction.setShortcut(QtGui.QKeySequence("F5"))
self.pushButton_refresh.setIcon(QtGui.QIcon(":/IMG/img/refresh.png"))
self.pushButton_new.setIcon(QtGui.QIcon(":/IMG/img/new.png"))
if self.isGlobalMenu():
self._setupToolBar()
else:
self._setupButtonWidget()
self._setTabIcon(self.homeTab, QtGui.QPixmap(":/IMG/img/sina.png"))
self._setTabIcon(self.mentionsTab, QtGui.QPixmap(":/IMG/img/mentions.png"))
self._setTabIcon(self.commentsTab, QtGui.QPixmap(":/IMG/img/comments2.png"))
self._setTabIcon(self.commentsMentionsTab, QtGui.QPixmap(":/IMG/img/mentions_comments.svg"))
self.retranslateUi(mainWindow)
def isGlobalMenu(self):
if os.environ.get("TOOLBAR") == "1":
return True
elif os.environ.get("TOOLBAR") == "0":
return False
elif os.environ.get('DESKTOP_SESSION') in ["ubuntu", "ubuntu-2d"]:
if not os.environ.get("UBUNTU_MENUPROXY"):
return False
elif os.environ.get("APPMENU_DISPLAY_BOTH"):
return False
else:
return True
elif os.environ.get("DESKTOP_SESSION") == "kde-plasma" and platform.linux_distribution()[0] == "Ubuntu":
return True
elif platform.system() == "Darwin":
return True
return False
def _setupToolBar(self):
self.toolBar = QtGui.QToolBar()
self.toolBar.setToolButtonStyle(QtCore.Qt.ToolButtonTextBesideIcon)
empty = QtGui.QWidget()
empty.setSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Preferred)
self.toolBar.addWidget(empty)
self.toolBar.addAction(self.refreshAction)
newAction = self.toolBar.addAction(QtGui.QIcon(":/IMG/img/new.png"),
"New")
newAction.triggered.connect(self.pushButton_new.clicked)
self.addToolBar(self.toolBar)
def _setupButtonWidget(self):
self.buttonWidget = QtGui.QWidget(self)
self.buttonLayout = QtGui.QHBoxLayout(self.buttonWidget)
self.horizontalSpacer = QtGui.QSpacerItem(40, 20,
QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum)
self.buttonLayout.addSpacerItem(self.horizontalSpacer)
self.buttonLayout.addWidget(self.pushButton_refresh)
self.buttonLayout.addWidget(self.pushButton_new)
def resizeEvent(self, event):
# This is a hack!!!
if self.isGlobalMenu():
return
self.buttonWidget.resize(self.menubar.sizeHint().width(),
self.menubar.sizeHint().height() + 12)
self.buttonWidget.move(self.width() - self.buttonWidget.width(),
self.menubar.geometry().topRight().y() - 5)
def retranslateUi(self, frm_MainWindow):
frm_MainWindow.setWindowTitle(self.tr("WeCase"))
self.mainMenu.setTitle(self.tr("&WeCase"))
self.helpMenu.setTitle(self.tr("&Help"))
self.optionsMenu.setTitle(self.tr("&Options"))
self.aboutAction.setText(self.tr("&About..."))
self.refreshAction.setText(self.tr("Refresh"))
self.logoutAction.setText(self.tr("&Log out"))
self.exitAction.setText(self.tr("&Exit"))
self.settingsAction.setText(self.tr("&Settings"))
def _setupSysTray(self):
self.systray = QtGui.QSystemTrayIcon()
self.systray.activated.connect(self.clickedSystray)
self.systray.setToolTip("WeCase")
self.systray.setIcon(QtGui.QIcon(":/IMG/img/WeCase.svg"))
self.systray.show()
self.visibleAction = QtGui.QAction(self)
self.visibleAction.setText(self.tr("&Hide"))
self.visibleAction.triggered.connect(self._switchVisibility)
self.sysMenu = QtGui.QMenu(self)
self.sysMenu.addAction(self.visibleAction)
self.sysMenu.addAction(self.logoutAction)
self.sysMenu.addAction(self.exitAction)
self.systray.setContextMenu(self.sysMenu)
def clickedSystray(self, reason):
if reason == QtGui.QSystemTrayIcon.Trigger:
self._switchVisibility()
elif reason == QtGui.QSystemTrayIcon.Context:
pass
def _switchVisibility(self):
if self.isVisible():
self.hide()
self.visibleAction.setText(self.tr("&Show"))
else:
self.show()
self.visibleAction.setText(self.tr("&Hide"))
def _setTabIcon(self, tab, icon):
pixmap = icon.transformed(QtGui.QTransform().rotate(90))
icon = QtGui.QIcon(pixmap)
self._iconPixmap[icon.cacheKey()] = pixmap
self.tabWidget.setTabIcon(self.tabWidget.indexOf(tab), icon)
self.tabWidget.setIconSize(QtCore.QSize(24, 24))
def _prepareTimeline(self, timeline):
try:
timeline.setUsersBlacklist(self.usersBlacklist)
timeline.setTweetsKeywordsBlacklist(self.tweetKeywordsBlacklist)
timeline.setWordWarKeywords(self.wordWarKeywords)
timeline.setBlockWordwars(self.blockWordwars)
except AttributeError:
pass
timeline.load()
def closeTab(self, index):
widget = self.tabWidget.widget(index)
self.tabWidget.removeTab(index)
widget.deleteLater()
def init_account(self):
self.uid()
def loadConfig(self):
self.config = WConfigParser(path.myself_path + "WMetaConfig",
path.config_path, "main")
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
self.wordWarKeywords = self.config.wordWarKeywords
self.blockWordwars = self.config.blockWordwars
self.maxRetweets = self.config.maxRetweets
self.maxTweetsPerUser = self.config.maxTweetsPerUser
self.mainWindow_geometry = self.config.mainwindow_geometry
def applyConfig(self):
try:
self.timer.stop()
except AttributeError:
pass
self.timer = WTimer(self.show_notify, self.notify_interval)
self.timer.start()
self.notify.timeout = self.notify_timeout
setGeometry(self, self.mainWindow_geometry)
def setupModels(self):
self._all_timeline = TweetCommonModel(self.client.statuses.home_timeline, self)
self.all_timeline = TweetFilterModel(self._all_timeline)
self.all_timeline.setModel(self._all_timeline)
self._prepareTimeline(self.all_timeline)
# extra rules
self.all_timeline.setMaxRetweets(self.maxRetweets)
self.all_timeline.setMaxTweetsPerUser(self.maxTweetsPerUser)
self.homeView.setModel(self.all_timeline)
self._mentions = TweetCommonModel(self.client.statuses.mentions, self)
self.mentions = TweetFilterModel(self._mentions)
self.mentions.setModel(self._mentions)
self._prepareTimeline(self.mentions)
self.mentionsView.setModel(self.mentions)
self._comment_to_me = TweetCommentModel(self.client.comments.to_me, self)
self.comment_to_me = TweetFilterModel(self._comment_to_me)
self.comment_to_me.setModel(self._comment_to_me)
self._prepareTimeline(self.comment_to_me)
self.commentsView.setModel(self.comment_to_me)
self._comment_mentions = TweetCommentModel(self.client.comments.mentions, self)
self.comment_mentions = TweetFilterModel(self._comment_mentions)
self.comment_mentions.setModel(self._comment_mentions)
self._prepareTimeline(self.comment_mentions)
self.commentsMentionsView.setModel(self.comment_mentions)
@async
def reset_remind(self):
typ = ""
if self.currentTweetView() == self.homeView:
self.tabBadgeChanged.emit(self.tabWidget.currentIndex(), 0)
elif self.currentTweetView() == self.mentionsView:
typ = "mention_status"
self.tabBadgeChanged.emit(self.tabWidget.currentIndex(), 0)
elif self.currentTweetView() == self.commentsView:
typ = "cmt"
self.tabBadgeChanged.emit(self.tabWidget.currentIndex(), 0)
elif self.currentTweetView() == self.commentsMentionsView:
typ = "mention_cmt"
self.tabBadgeChanged.emit(self.tabWidget.currentIndex(), 0)
if typ:
self.client.remind.set_count.post(type=typ)
def get_remind(self, uid):
"""this function is used to get unread_count
from Weibo API. uid is necessary."""
reminds = self.client.remind.unread_count.get(uid=uid)
return reminds
def uid(self):
"""How can I get my uid? here it is"""
if not self.info.get("uid"):
self.info["uid"] = self.client.account.get_uid.get().uid
return self.info["uid"]
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"
reminds_count = 0
if reminds['status'] != 0:
# Note: do NOT send notify here, or users will crazy.
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.homeTab),
reminds['status'])
if reminds['mention_status'] and self.remindMentions:
msg += self.tr("%d unread @ME") % reminds['mention_status'] + "\n"
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.mentionsTab),
reminds['mention_status'])
reminds_count += 1
else:
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.mentionsTab), 0)
if reminds['cmt'] and self.remindComments:
msg += self.tr("%d unread comment(s)") % reminds['cmt'] + "\n"
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.commentsTab),
reminds['cmt'])
reminds_count += 1
else:
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.commentsTab), 0)
if reminds["mention_cmt"] and self.remindMentions:
msg += self.tr("%d unread @ME comment(s)") % reminds["mention_cmt"] + "\n"
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.commentsMentionsTab),
reminds["mention_cmt"])
reminds_count += 1
else:
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.commentsMentionsTab), 0)
if reminds_count and reminds_count != self._last_reminds_count:
self.notify.showMessage(self.tr("WeCase"), msg)
self._last_reminds_count = reminds_count
def drawNotifyBadge(self, index, count):
tabIcon = self.tabWidget.tabIcon(index)
_tabPixmap = self._iconPixmap[tabIcon.cacheKey()]
tabPixmap = _tabPixmap.transformed(QtGui.QTransform().rotate(-90))
icon = NotifyBadgeDrawer().draw(tabPixmap, str(count))
icon = icon.transformed(QtGui.QTransform().rotate(90))
icon = QtGui.QIcon(icon)
self._iconPixmap[icon.cacheKey()] = _tabPixmap
self.tabWidget.setTabIcon(index, icon)
def moveToTop(self):
self.currentTweetView().moveToTop()
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):
self.wecase_new = NewpostWindow()
self.wecase_new.userClicked.connect(self.userClicked)
self.wecase_new.tagClicked.connect(self.tagClicked)
self.wecase_new.show()
def refresh(self):
tweetView = self.currentTweetView()
tweetView.model().timelineLoaded.connect(self.moveToTop)
tweetView.refresh()
self.reset_remind()
def currentTweetView(self):
# The most tricky part of MainWindow.
return self.tabWidget.currentWidget().layout().itemAt(0).widget()
def saveConfig(self):
self.config.mainwindow_geometry = getGeometry(self)
self.config.save()
def closeEvent(self, event):
self.systray.hide()
self.hide()
self.saveConfig()
self.timer.stop(True)
# Reset uid when the thread exited.
self.info["uid"] = None
logging.info("Die")
def setupUi(self, mainWindow):
mainWindow.setWindowIcon(QtGui.QIcon(":/IMG/img/WeCase.svg"))
mainWindow.setDocumentMode(False)
mainWindow.setDockOptions(QtGui.QMainWindow.AllowTabbedDocks |
QtGui.QMainWindow.AnimatedDocks)
self.centralwidget = QtGui.QWidget(mainWindow)
self.verticalLayout = QtGui.QVBoxLayout(self.centralwidget)
self.tabWidget = QtGui.QTabWidget(self.centralwidget)
self.tabWidget.setTabBar(WTabBar(self.tabWidget))
self.tabWidget.setTabPosition(QtGui.QTabWidget.West)
self.tabWidget.setTabShape(QtGui.QTabWidget.Rounded)
self.tabWidget.setDocumentMode(False)
self.tabWidget.setMovable(True)
self.tabWidget.tabCloseRequested.connect(self.closeTab)
self.homeView = TweetListWidget()
self.homeView.userClicked.connect(self.userClicked)
self.homeView.tagClicked.connect(self.tagClicked)
self.homeTab = self._setupTab(self.homeView)
self.tabWidget.addTab(self.homeTab, "")
self.tabWidget.tabBar().setProtectTab(self.homeTab, True)
self.mentionsView = TweetListWidget()
self.mentionsView.userClicked.connect(self.userClicked)
self.mentionsView.tagClicked.connect(self.tagClicked)
self.mentionsTab = self._setupTab(self.mentionsView)
self.tabWidget.addTab(self.mentionsTab, "")
self.tabWidget.tabBar().setProtectTab(self.mentionsTab, True)
self.commentsView = TweetListWidget()
self.commentsView.userClicked.connect(self.userClicked)
self.commentsView.tagClicked.connect(self.tagClicked)
self.commentsTab = self._setupTab(self.commentsView)
self.tabWidget.addTab(self.commentsTab, "")
self.tabWidget.tabBar().setProtectTab(self.commentsTab, True)
self.commentsMentionsView = TweetListWidget()
self.commentsMentionsView.userClicked.connect(self.userClicked)
self.commentsMentionsView.tagClicked.connect(self.tagClicked)
self.commentsMentionsTab = self._setupTab(self.commentsMentionsView)
self.tabWidget.addTab(self.commentsMentionsTab, "")
self.tabWidget.tabBar().setProtectTab(self.commentsMentionsTab, True)
self.verticalLayout.addWidget(self.tabWidget)
self.widget = QtGui.QWidget(self.centralwidget)
self.verticalLayout.addWidget(self.widget)
mainWindow.setCentralWidget(self.centralwidget)
self.aboutAction = QtGui.QAction(mainWindow)
self.refreshAction = QtGui.QAction(mainWindow)
self.logoutAction = QtGui.QAction(mainWindow)
self.exitAction = QtGui.QAction(mainWindow)
self.settingsAction = QtGui.QAction(mainWindow)
self.aboutAction.setIcon(QtGui.QIcon(QtGui.QPixmap("./IMG/img/where_s_my_weibo.svg")))
self.exitAction.setIcon(QtGui.QIcon(QtGui.QPixmap(":/IMG/img/application-exit.svg")))
self.settingsAction.setIcon(QtGui.QIcon(QtGui.QPixmap(":/IMG/img/preferences-other.png")))
self.refreshAction.setIcon(QtGui.QIcon(QtGui.QPixmap(":/IMG/img/refresh.png")))
self.menubar = QtGui.QMenuBar(mainWindow)
self.menubar.setEnabled(True)
self.menubar.setDefaultUp(False)
mainWindow.setMenuBar(self.menubar)
self.mainMenu = QtGui.QMenu(self.menubar)
self.helpMenu = QtGui.QMenu(self.menubar)
self.optionsMenu = QtGui.QMenu(self.menubar)
self.mainMenu.addAction(self.refreshAction)
self.mainMenu.addSeparator()
self.mainMenu.addAction(self.logoutAction)
self.mainMenu.addAction(self.exitAction)
self.helpMenu.addAction(self.aboutAction)
self.optionsMenu.addAction(self.settingsAction)
self.menubar.addAction(self.mainMenu.menuAction())
self.menubar.addAction(self.optionsMenu.menuAction())
self.menubar.addAction(self.helpMenu.menuAction())
self.exitAction.triggered.connect(mainWindow.close)
self.aboutAction.triggered.connect(mainWindow.showAbout)
self.settingsAction.triggered.connect(mainWindow.showSettings)
self.logoutAction.triggered.connect(mainWindow.logout)
self.refreshAction.triggered.connect(mainWindow.refresh)
self.pushButton_refresh = QtGui.QPushButton(self.widget)
self.pushButton_new = QtGui.QPushButton(self.widget)
self.pushButton_refresh.clicked.connect(mainWindow.refresh)
self.pushButton_new.clicked.connect(mainWindow.postTweet)
self.timelineLoaded.connect(self.moveToTop)
self.tabBadgeChanged.connect(self.drawNotifyBadge)
self.refreshAction.setShortcut(QtGui.QKeySequence("F5"))
self.pushButton_refresh.setIcon(QtGui.QIcon(":/IMG/img/refresh.png"))
self.pushButton_new.setIcon(QtGui.QIcon(":/IMG/img/new.png"))
if self.isGlobalMenu():
self._setupToolBar()
else:
self._setupButtonWidget()
self._setTabIcon(self.homeTab, QtGui.QPixmap(":/IMG/img/sina.png"))
self._setTabIcon(self.mentionsTab, QtGui.QPixmap(":/IMG/img/mentions.png"))
self._setTabIcon(self.commentsTab, QtGui.QPixmap(":/IMG/img/comments2.png"))
self._setTabIcon(self.commentsMentionsTab, QtGui.QPixmap(":/IMG/img/mentions_comments.svg"))
self.retranslateUi(mainWindow)
class WeCaseWindow(QtGui.QMainWindow):
client = None
uid = None
timelineLoaded = QtCore.pyqtSignal(int)
imageLoaded = QtCore.pyqtSignal(str)
tabBadgeChanged = QtCore.pyqtSignal(int, int)
tabAvatarFetched = QtCore.pyqtSignal(str)
def __init__(self, username, parent=None):
super(WeCaseWindow, self).__init__(parent)
self.username = username
LoginInfo().add_account(self.username)
self.errorWindow = APIErrorWindow(self)
self.setAttribute(QtCore.Qt.WA_QuitOnClose, True)
self._iconPixmap = {}
self.setupUi(self)
self._setupSysTray()
self.tweetViews = [
self.homeView, self.mentionsView, self.commentsView,
self.commentsMentionsTab
]
self.info = WeRuntimeInfo()
self.client = const.client
self.loadConfig()
self.init_account()
self.setupModels()
self.IMG_AVATAR = -2
self.IMG_THUMB = -1
self.notify = Notify(timeout=self.notify_timeout)
self.applyConfig()
self.download_lock = []
self._last_reminds_count = 0
self._setupUserTab(self.uid(), False, True)
def _setupTab(self, view):
tab = QtGui.QWidget()
layout = QtGui.QGridLayout(tab)
layout.addWidget(view)
view.setParent(tab)
return tab
def _setupCommonTab(self, timeline, view, switch=True, protect=False):
self._prepareTimeline(timeline)
view.setModel(timeline)
view.userClicked.connect(self.userClicked)
view.tagClicked.connect(self.tagClicked)
tab = self._setupTab(view)
self.tabWidget.addTab(tab, "")
if switch:
self.tabWidget.setCurrentWidget(tab)
if protect:
self.tabWidget.tabBar().setProtectTab(tab, True)
return tab
def _getSameTab(self, attr, value):
for i in range(self.tabWidget.count()):
try:
tab = self.tabWidget.widget(i).layout().itemAt(0).widget()
_value = getattr(tab.model(), attr)()
if _value == value:
return i
except AttributeError:
pass
return False
def _setupUserTab(self, uid, switch=True, myself=False):
index = self._getSameTab("uid", uid)
if index:
if switch:
self.tabWidget.setCurrentIndex(index)
return
view = TweetListWidget()
_timeline = TweetUserModel(self.client.api("statuses/user_timeline"),
uid, view)
timeline = TweetFilterModel(_timeline)
timeline.setModel(_timeline)
tab = self._setupCommonTab(timeline, view, switch, myself)
def setAvatar(f):
self._setTabIcon(tab, WObjectCache().open(QtGui.QPixmap, f))
fetcher = AsyncFetcher("".join(
(path.cache_path, str(self.info["uid"]))))
fetcher.addTask(
self.client.api("users/show").get(uid=uid)["profile_image_url"],
setAvatar)
def _setupTopicTab(self, topic, switch=True):
index = self._getSameTab("topic", topic)
if index:
if switch:
self.tabWidget.setCurrentIndex(index)
return
view = TweetListWidget()
timeline = TweetTopicModel(self.client.api("search/topics"), topic,
view)
tab = self._setupCommonTab(timeline, view, switch, protect=False)
self._setTabIcon(
tab,
WObjectCache().open(QtGui.QPixmap, ":/IMG/img/topic.jpg"))
def userClicked(self, userItem, openAtBackend):
try:
self._setupUserTab(userItem.id, switch=(not openAtBackend))
except APIError as e:
self.errorWindow.raiseException.emit(e)
def tagClicked(self, str, openAtBackend):
try:
self._setupTopicTab(str, switch=(not openAtBackend))
except APIError as e:
self.errorWindow.raiseException.emit(e)
def setupUi(self, mainWindow):
mainWindow.setWindowIcon(QtGui.QIcon(":/IMG/img/WeCase.svg"))
mainWindow.setDocumentMode(False)
mainWindow.setDockOptions(QtGui.QMainWindow.AllowTabbedDocks
| QtGui.QMainWindow.AnimatedDocks)
self.centralwidget = QtGui.QWidget(mainWindow)
self.verticalLayout = QtGui.QVBoxLayout(self.centralwidget)
self.tabWidget = QtGui.QTabWidget(self.centralwidget)
self.tabWidget.setTabBar(WTabBar(self.tabWidget))
self.tabWidget.setTabPosition(QtGui.QTabWidget.West)
self.tabWidget.setTabShape(QtGui.QTabWidget.Rounded)
self.tabWidget.setDocumentMode(False)
self.tabWidget.setMovable(True)
self.tabWidget.tabCloseRequested.connect(self.closeTab)
self.homeView = TweetListWidget()
self.homeView.userClicked.connect(self.userClicked)
self.homeView.tagClicked.connect(self.tagClicked)
self.homeTab = self._setupTab(self.homeView)
self.tabWidget.addTab(self.homeTab, "")
self.tabWidget.tabBar().setProtectTab(self.homeTab, True)
self.mentionsView = TweetListWidget()
self.mentionsView.userClicked.connect(self.userClicked)
self.mentionsView.tagClicked.connect(self.tagClicked)
self.mentionsTab = self._setupTab(self.mentionsView)
self.tabWidget.addTab(self.mentionsTab, "")
self.tabWidget.tabBar().setProtectTab(self.mentionsTab, True)
self.commentsView = TweetListWidget()
self.commentsView.userClicked.connect(self.userClicked)
self.commentsView.tagClicked.connect(self.tagClicked)
self.commentsTab = self._setupTab(self.commentsView)
self.tabWidget.addTab(self.commentsTab, "")
self.tabWidget.tabBar().setProtectTab(self.commentsTab, True)
self.commentsMentionsView = TweetListWidget()
self.commentsMentionsView.userClicked.connect(self.userClicked)
self.commentsMentionsView.tagClicked.connect(self.tagClicked)
self.commentsMentionsTab = self._setupTab(self.commentsMentionsView)
self.tabWidget.addTab(self.commentsMentionsTab, "")
self.tabWidget.tabBar().setProtectTab(self.commentsMentionsTab, True)
self.verticalLayout.addWidget(self.tabWidget)
self.widget = QtGui.QWidget(self.centralwidget)
self.verticalLayout.addWidget(self.widget)
mainWindow.setCentralWidget(self.centralwidget)
self.aboutAction = QtGui.QAction(mainWindow)
self.refreshAction = QtGui.QAction(mainWindow)
self.logoutAction = QtGui.QAction(mainWindow)
self.exitAction = QtGui.QAction(mainWindow)
self.settingsAction = QtGui.QAction(mainWindow)
self.aboutAction.setIcon(
QtGui.QIcon(QtGui.QPixmap("./IMG/img/where_s_my_weibo.svg")))
self.exitAction.setIcon(
QtGui.QIcon(QtGui.QPixmap(":/IMG/img/application-exit.svg")))
self.settingsAction.setIcon(
QtGui.QIcon(QtGui.QPixmap(":/IMG/img/preferences-other.png")))
self.refreshAction.setIcon(
QtGui.QIcon(QtGui.QPixmap(":/IMG/img/refresh.png")))
self.menubar = QtGui.QMenuBar(mainWindow)
self.menubar.setEnabled(True)
self.menubar.setDefaultUp(False)
mainWindow.setMenuBar(self.menubar)
self.mainMenu = QtGui.QMenu(self.menubar)
self.helpMenu = QtGui.QMenu(self.menubar)
self.optionsMenu = QtGui.QMenu(self.menubar)
self.mainMenu.addAction(self.refreshAction)
self.mainMenu.addSeparator()
self.mainMenu.addAction(self.logoutAction)
self.mainMenu.addAction(self.exitAction)
self.helpMenu.addAction(self.aboutAction)
self.optionsMenu.addAction(self.settingsAction)
self.menubar.addAction(self.mainMenu.menuAction())
self.menubar.addAction(self.optionsMenu.menuAction())
self.menubar.addAction(self.helpMenu.menuAction())
self.exitAction.triggered.connect(mainWindow.close)
self.aboutAction.triggered.connect(mainWindow.showAbout)
self.settingsAction.triggered.connect(mainWindow.showSettings)
self.logoutAction.triggered.connect(mainWindow.logout)
self.refreshAction.triggered.connect(mainWindow.refresh)
self.pushButton_refresh = QtGui.QPushButton(self.widget)
self.pushButton_new = QtGui.QPushButton(self.widget)
self.pushButton_refresh.clicked.connect(mainWindow.refresh)
self.pushButton_new.clicked.connect(mainWindow.postTweet)
self.timelineLoaded.connect(self.moveToTop)
self.tabBadgeChanged.connect(self.drawNotifyBadge)
self.refreshAction.setShortcut(QtGui.QKeySequence("F5"))
self.pushButton_refresh.setIcon(QtGui.QIcon(":/IMG/img/refresh.png"))
self.pushButton_new.setIcon(QtGui.QIcon(":/IMG/img/new.png"))
if self.isGlobalMenu():
self._setupToolBar()
else:
self._setupButtonWidget()
self._setTabIcon(self.homeTab, QtGui.QPixmap(":/IMG/img/sina.png"))
self._setTabIcon(self.mentionsTab,
QtGui.QPixmap(":/IMG/img/mentions.png"))
self._setTabIcon(self.commentsTab,
QtGui.QPixmap(":/IMG/img/comments2.png"))
self._setTabIcon(self.commentsMentionsTab,
QtGui.QPixmap(":/IMG/img/mentions_comments.svg"))
self.retranslateUi(mainWindow)
def isGlobalMenu(self):
if os.environ.get("TOOLBAR") == "1":
return True
elif os.environ.get("TOOLBAR") == "0":
return False
elif os.environ.get('DESKTOP_SESSION') in ["ubuntu", "ubuntu-2d"]:
if not os.environ.get("UBUNTU_MENUPROXY"):
return False
elif os.environ.get("APPMENU_DISPLAY_BOTH"):
return False
else:
return True
elif os.environ.get(
"DESKTOP_SESSION"
) == "kde-plasma" and platform.linux_distribution()[0] == "Ubuntu":
return True
elif platform.system() == "Darwin":
return True
return False
def _setupToolBar(self):
self.toolBar = QtGui.QToolBar()
self.toolBar.setToolButtonStyle(QtCore.Qt.ToolButtonTextBesideIcon)
empty = QtGui.QWidget()
empty.setSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Preferred)
self.toolBar.addWidget(empty)
self.toolBar.addAction(self.refreshAction)
newAction = self.toolBar.addAction(QtGui.QIcon(":/IMG/img/new.png"),
"New")
newAction.triggered.connect(self.pushButton_new.clicked)
self.addToolBar(self.toolBar)
def _setupButtonWidget(self):
self.buttonWidget = QtGui.QWidget(self)
self.buttonLayout = QtGui.QHBoxLayout(self.buttonWidget)
self.horizontalSpacer = QtGui.QSpacerItem(40, 20,
QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Minimum)
self.buttonLayout.addSpacerItem(self.horizontalSpacer)
self.buttonLayout.addWidget(self.pushButton_refresh)
self.buttonLayout.addWidget(self.pushButton_new)
def resizeEvent(self, event):
# This is a hack!!!
if self.isGlobalMenu():
return
self.buttonWidget.resize(self.menubar.sizeHint().width(),
self.menubar.sizeHint().height() + 12)
self.buttonWidget.move(self.width() - self.buttonWidget.width(),
self.menubar.geometry().topRight().y() - 5)
def retranslateUi(self, frm_MainWindow):
frm_MainWindow.setWindowTitle(self.tr("WeCase"))
self.mainMenu.setTitle(self.tr("&WeCase"))
self.helpMenu.setTitle(self.tr("&Help"))
self.optionsMenu.setTitle(self.tr("&Options"))
self.aboutAction.setText(self.tr("&About..."))
self.refreshAction.setText(self.tr("Refresh"))
self.logoutAction.setText(self.tr("&Log out"))
self.exitAction.setText(self.tr("&Exit"))
self.settingsAction.setText(self.tr("&Settings"))
def _setupSysTray(self):
self.systray = QtGui.QSystemTrayIcon()
self.systray.activated.connect(self.clickedSystray)
self.systray.setToolTip("WeCase")
self.systray.setIcon(QtGui.QIcon(":/IMG/img/WeCase.svg"))
self.systray.show()
self.visibleAction = QtGui.QAction(self)
self.visibleAction.setText(self.tr("&Hide"))
self.visibleAction.triggered.connect(self._switchVisibility)
self.sysMenu = QtGui.QMenu(self)
self.sysMenu.addAction(self.visibleAction)
self.sysMenu.addAction(self.logoutAction)
self.sysMenu.addAction(self.exitAction)
self.systray.setContextMenu(self.sysMenu)
def clickedSystray(self, reason):
if reason == QtGui.QSystemTrayIcon.Trigger:
self._switchVisibility()
elif reason == QtGui.QSystemTrayIcon.Context:
pass
def _switchVisibility(self):
if self.isVisible():
self.hide()
self.visibleAction.setText(self.tr("&Show"))
else:
self.show()
self.visibleAction.setText(self.tr("&Hide"))
def _setTabIcon(self, tab, icon):
pixmap = icon.transformed(QtGui.QTransform().rotate(90))
icon = QtGui.QIcon(pixmap)
self._iconPixmap[icon.cacheKey()] = pixmap
self.tabWidget.setTabIcon(self.tabWidget.indexOf(tab), icon)
self.tabWidget.setIconSize(QtCore.QSize(24, 24))
def _prepareTimeline(self, timeline):
try:
timeline.setUsersBlacklist(self.usersBlacklist)
timeline.setTweetsKeywordsBlacklist(self.tweetKeywordsBlacklist)
timeline.setWordWarKeywords(self.wordWarKeywords)
timeline.setBlockWordwars(self.blockWordwars)
timeline.setKeywordsAsRegexs(self.keywordsAsRegex)
except AttributeError:
pass
timeline.load()
def closeTab(self, index):
widget = self.tabWidget.widget(index)
self.tabWidget.removeTab(index)
widget.deleteLater()
def init_account(self):
self.uid()
def loadConfig(self):
self.config = WConfigParser(path.myself_path + "WMetaConfig",
path.config_path, "main")
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
self.wordWarKeywords = self.config.wordWarKeywords
self.blockWordwars = self.config.blockWordwars
self.maxRetweets = self.config.maxRetweets
self.maxTweetsPerUser = self.config.maxTweetsPerUser
self.mainWindow_geometry = self.config.mainwindow_geometry
self.keywordsAsRegex = self.config.keywordsAsRegex
def applyConfig(self):
try:
self.timer.stop()
except AttributeError:
pass
self.timer = WTimer(self.show_notify, self.notify_interval)
self.timer.start()
self.notify.timeout = self.notify_timeout
setGeometry(self, self.mainWindow_geometry)
def setupModels(self):
self._all_timeline = TweetCommonModel(
self.client.api("statuses/home_timeline"), self)
self.all_timeline = TweetFilterModel(self._all_timeline)
self.all_timeline.setModel(self._all_timeline)
self._prepareTimeline(self.all_timeline)
# extra rules
self.all_timeline.setMaxRetweets(self.maxRetweets)
self.all_timeline.setMaxTweetsPerUser(self.maxTweetsPerUser)
self.homeView.setModel(self.all_timeline)
self._mentions = TweetCommonModel(self.client.api("statuses/mentions"),
self)
self.mentions = TweetFilterModel(self._mentions)
self.mentions.setModel(self._mentions)
self._prepareTimeline(self.mentions)
self.mentionsView.setModel(self.mentions)
self._comment_to_me = TweetCommentModel(
self.client.api("comments/to_me"), self)
self.comment_to_me = TweetFilterModel(self._comment_to_me)
self.comment_to_me.setModel(self._comment_to_me)
self._prepareTimeline(self.comment_to_me)
self.commentsView.setModel(self.comment_to_me)
self._comment_mentions = TweetCommentModel(
self.client.api("comments/mentions"), self)
self.comment_mentions = TweetFilterModel(self._comment_mentions)
self.comment_mentions.setModel(self._comment_mentions)
self._prepareTimeline(self.comment_mentions)
self.commentsMentionsView.setModel(self.comment_mentions)
@async
def reset_remind(self):
typ = ""
if self.currentTweetView() == self.homeView:
self.tabBadgeChanged.emit(self.tabWidget.currentIndex(), 0)
elif self.currentTweetView() == self.mentionsView:
typ = "mention_status"
self.tabBadgeChanged.emit(self.tabWidget.currentIndex(), 0)
elif self.currentTweetView() == self.commentsView:
typ = "cmt"
self.tabBadgeChanged.emit(self.tabWidget.currentIndex(), 0)
elif self.currentTweetView() == self.commentsMentionsView:
typ = "mention_cmt"
self.tabBadgeChanged.emit(self.tabWidget.currentIndex(), 0)
if typ:
self.client.api("remind/set_count").post(type=typ)
def get_remind(self, uid):
"""this function is used to get unread_count
from Weibo API. uid is necessary."""
reminds = self.client.api("remind/unread_count").get(uid=uid)
return reminds
def uid(self):
"""How can I get my uid? here it is"""
if not self.info.get("uid"):
self.info["uid"] = self.client.api("account/get_uid").get().uid
return self.info["uid"]
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"
reminds_count = 0
if reminds['status'] != 0:
# Note: do NOT send notify here, or users will crazy.
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.homeTab),
reminds['status'])
if reminds['mention_status'] and self.remindMentions:
msg += self.tr("%d unread @ME") % reminds['mention_status'] + "\n"
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.mentionsTab),
reminds['mention_status'])
reminds_count += 1
else:
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.mentionsTab),
0)
if reminds['cmt'] and self.remindComments:
msg += self.tr("%d unread comment(s)") % reminds['cmt'] + "\n"
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.commentsTab),
reminds['cmt'])
reminds_count += 1
else:
self.tabBadgeChanged.emit(self.tabWidget.indexOf(self.commentsTab),
0)
if reminds["mention_cmt"] and self.remindMentions:
msg += self.tr(
"%d unread @ME comment(s)") % reminds["mention_cmt"] + "\n"
self.tabBadgeChanged.emit(
self.tabWidget.indexOf(self.commentsMentionsTab),
reminds["mention_cmt"])
reminds_count += 1
else:
self.tabBadgeChanged.emit(
self.tabWidget.indexOf(self.commentsMentionsTab), 0)
if reminds_count and reminds_count != self._last_reminds_count:
self.notify.showMessage(self.tr("WeCase"), msg)
self._last_reminds_count = reminds_count
def drawNotifyBadge(self, index, count):
tabIcon = self.tabWidget.tabIcon(index)
_tabPixmap = self._iconPixmap[tabIcon.cacheKey()]
tabPixmap = _tabPixmap.transformed(QtGui.QTransform().rotate(-90))
icon = NotifyBadgeDrawer().draw(tabPixmap, str(count))
icon = icon.transformed(QtGui.QTransform().rotate(90))
icon = QtGui.QIcon(icon)
self._iconPixmap[icon.cacheKey()] = _tabPixmap
self.tabWidget.setTabIcon(index, icon)
def moveToTop(self):
self.currentTweetView().moveToTop()
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):
self.wecase_new = NewpostWindow()
self.wecase_new.userClicked.connect(self.userClicked)
self.wecase_new.tagClicked.connect(self.tagClicked)
self.wecase_new.show()
def refresh(self):
tweetView = self.currentTweetView()
tweetView.model().timelineLoaded.connect(self.moveToTop)
tweetView.refresh()
self.reset_remind()
def currentTweetView(self):
# The most tricky part of MainWindow.
return self.tabWidget.currentWidget().layout().itemAt(0).widget()
def saveConfig(self):
self.config.mainwindow_geometry = getGeometry(self)
self.config.save()
def closeEvent(self, event):
self.systray.hide()
self.hide()
self.saveConfig()
self.timer.stop(True)
# Reset uid when the thread exited.
self.info["uid"] = None
LoginInfo().remove_account(self.username)
logging.info("Die")
def setupUi(self, mainWindow):
mainWindow.setWindowIcon(QtGui.QIcon(":/IMG/img/WeCase.svg"))
mainWindow.setDocumentMode(False)
mainWindow.setDockOptions(QtGui.QMainWindow.AllowTabbedDocks
| QtGui.QMainWindow.AnimatedDocks)
self.centralwidget = QtGui.QWidget(mainWindow)
self.verticalLayout = QtGui.QVBoxLayout(self.centralwidget)
self.tabWidget = QtGui.QTabWidget(self.centralwidget)
self.tabWidget.setTabBar(WTabBar(self.tabWidget))
self.tabWidget.setTabPosition(QtGui.QTabWidget.West)
self.tabWidget.setTabShape(QtGui.QTabWidget.Rounded)
self.tabWidget.setDocumentMode(False)
self.tabWidget.setMovable(True)
self.tabWidget.tabCloseRequested.connect(self.closeTab)
self.homeView = TweetListWidget()
self.homeView.userClicked.connect(self.userClicked)
self.homeView.tagClicked.connect(self.tagClicked)
self.homeTab = self._setupTab(self.homeView)
self.tabWidget.addTab(self.homeTab, "")
self.tabWidget.tabBar().setProtectTab(self.homeTab, True)
self.mentionsView = TweetListWidget()
self.mentionsView.userClicked.connect(self.userClicked)
self.mentionsView.tagClicked.connect(self.tagClicked)
self.mentionsTab = self._setupTab(self.mentionsView)
self.tabWidget.addTab(self.mentionsTab, "")
self.tabWidget.tabBar().setProtectTab(self.mentionsTab, True)
self.commentsView = TweetListWidget()
self.commentsView.userClicked.connect(self.userClicked)
self.commentsView.tagClicked.connect(self.tagClicked)
self.commentsTab = self._setupTab(self.commentsView)
self.tabWidget.addTab(self.commentsTab, "")
self.tabWidget.tabBar().setProtectTab(self.commentsTab, True)
self.commentsMentionsView = TweetListWidget()
self.commentsMentionsView.userClicked.connect(self.userClicked)
self.commentsMentionsView.tagClicked.connect(self.tagClicked)
self.commentsMentionsTab = self._setupTab(self.commentsMentionsView)
self.tabWidget.addTab(self.commentsMentionsTab, "")
self.tabWidget.tabBar().setProtectTab(self.commentsMentionsTab, True)
self.verticalLayout.addWidget(self.tabWidget)
self.widget = QtGui.QWidget(self.centralwidget)
self.verticalLayout.addWidget(self.widget)
mainWindow.setCentralWidget(self.centralwidget)
self.aboutAction = QtGui.QAction(mainWindow)
self.refreshAction = QtGui.QAction(mainWindow)
self.logoutAction = QtGui.QAction(mainWindow)
self.exitAction = QtGui.QAction(mainWindow)
self.settingsAction = QtGui.QAction(mainWindow)
self.aboutAction.setIcon(
QtGui.QIcon(QtGui.QPixmap("./IMG/img/where_s_my_weibo.svg")))
self.exitAction.setIcon(
QtGui.QIcon(QtGui.QPixmap(":/IMG/img/application-exit.svg")))
self.settingsAction.setIcon(
QtGui.QIcon(QtGui.QPixmap(":/IMG/img/preferences-other.png")))
self.refreshAction.setIcon(
QtGui.QIcon(QtGui.QPixmap(":/IMG/img/refresh.png")))
self.menubar = QtGui.QMenuBar(mainWindow)
self.menubar.setEnabled(True)
self.menubar.setDefaultUp(False)
mainWindow.setMenuBar(self.menubar)
self.mainMenu = QtGui.QMenu(self.menubar)
self.helpMenu = QtGui.QMenu(self.menubar)
self.optionsMenu = QtGui.QMenu(self.menubar)
self.mainMenu.addAction(self.refreshAction)
self.mainMenu.addSeparator()
self.mainMenu.addAction(self.logoutAction)
self.mainMenu.addAction(self.exitAction)
self.helpMenu.addAction(self.aboutAction)
self.optionsMenu.addAction(self.settingsAction)
self.menubar.addAction(self.mainMenu.menuAction())
self.menubar.addAction(self.optionsMenu.menuAction())
self.menubar.addAction(self.helpMenu.menuAction())
self.exitAction.triggered.connect(mainWindow.close)
self.aboutAction.triggered.connect(mainWindow.showAbout)
self.settingsAction.triggered.connect(mainWindow.showSettings)
self.logoutAction.triggered.connect(mainWindow.logout)
self.refreshAction.triggered.connect(mainWindow.refresh)
self.pushButton_refresh = QtGui.QPushButton(self.widget)
self.pushButton_new = QtGui.QPushButton(self.widget)
self.pushButton_refresh.clicked.connect(mainWindow.refresh)
self.pushButton_new.clicked.connect(mainWindow.postTweet)
self.timelineLoaded.connect(self.moveToTop)
self.tabBadgeChanged.connect(self.drawNotifyBadge)
self.refreshAction.setShortcut(QtGui.QKeySequence("F5"))
self.pushButton_refresh.setIcon(QtGui.QIcon(":/IMG/img/refresh.png"))
self.pushButton_new.setIcon(QtGui.QIcon(":/IMG/img/new.png"))
if self.isGlobalMenu():
self._setupToolBar()
else:
self._setupButtonWidget()
self._setTabIcon(self.homeTab, QtGui.QPixmap(":/IMG/img/sina.png"))
self._setTabIcon(self.mentionsTab,
QtGui.QPixmap(":/IMG/img/mentions.png"))
self._setTabIcon(self.commentsTab,
QtGui.QPixmap(":/IMG/img/comments2.png"))
self._setTabIcon(self.commentsMentionsTab,
QtGui.QPixmap(":/IMG/img/mentions_comments.svg"))
self.retranslateUi(mainWindow)
def setupUi(self, frm_MainWindow):
frm_MainWindow.setObjectName(_fromUtf8("frm_MainWindow"))
frm_MainWindow.resize(315, 637)
icon = QtGui.QIcon()
icon.addPixmap(QtGui.QPixmap(_fromUtf8(":/IMG/img/WeCase.svg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
frm_MainWindow.setWindowIcon(icon)
frm_MainWindow.setDocumentMode(False)
frm_MainWindow.setDockOptions(QtGui.QMainWindow.AllowTabbedDocks|QtGui.QMainWindow.AnimatedDocks)
self.centralwidget = QtGui.QWidget(frm_MainWindow)
self.centralwidget.setObjectName(_fromUtf8("centralwidget"))
self.verticalLayout = QtGui.QVBoxLayout(self.centralwidget)
self.verticalLayout.setObjectName(_fromUtf8("verticalLayout"))
self.tabWidget = QtGui.QTabWidget(self.centralwidget)
self.tabWidget.setTabPosition(QtGui.QTabWidget.West)
self.tabWidget.setTabShape(QtGui.QTabWidget.Rounded)
self.tabWidget.setDocumentMode(False)
self.tabWidget.setTabsClosable(True)
self.tabWidget.setMovable(True)
self.tabWidget.setObjectName(_fromUtf8("tabWidget"))
self.homeTab = QtGui.QWidget()
self.homeTab.setObjectName(_fromUtf8("homeTab"))
self.gridLayout_2 = QtGui.QGridLayout(self.homeTab)
self.gridLayout_2.setObjectName(_fromUtf8("gridLayout_2"))
self.homeView = TweetListWidget(self.homeTab)
self.homeView.setObjectName(_fromUtf8("homeView"))
self.gridLayout_2.addWidget(self.homeView, 0, 0, 1, 1)
self.tabWidget.addTab(self.homeTab, _fromUtf8(""))
self.mentionsTab = QtGui.QWidget()
self.mentionsTab.setObjectName(_fromUtf8("mentionsTab"))
self.verticalLayout_3 = QtGui.QVBoxLayout(self.mentionsTab)
self.verticalLayout_3.setObjectName(_fromUtf8("verticalLayout_3"))
self.mentionsView = TweetListWidget(self.mentionsTab)
self.mentionsView.setObjectName(_fromUtf8("mentionsView"))
self.verticalLayout_3.addWidget(self.mentionsView)
self.tabWidget.addTab(self.mentionsTab, _fromUtf8(""))
self.commentsTab = QtGui.QWidget()
self.commentsTab.setObjectName(_fromUtf8("commentsTab"))
self.verticalLayout_4 = QtGui.QVBoxLayout(self.commentsTab)
self.verticalLayout_4.setObjectName(_fromUtf8("verticalLayout_4"))
self.commentsView = TweetListWidget(self.commentsTab)
self.commentsView.setObjectName(_fromUtf8("commentsView"))
self.verticalLayout_4.addWidget(self.commentsView)
self.tabWidget.addTab(self.commentsTab, _fromUtf8(""))
self.myTab = QtGui.QWidget()
self.myTab.setObjectName(_fromUtf8("myTab"))
self.verticalLayout_5 = QtGui.QVBoxLayout(self.myTab)
self.verticalLayout_5.setObjectName(_fromUtf8("verticalLayout_5"))
self.myView = TweetListWidget(self.myTab)
self.myView.setObjectName(_fromUtf8("myView"))
self.verticalLayout_5.addWidget(self.myView)
self.tabWidget.addTab(self.myTab, _fromUtf8(""))
self.verticalLayout.addWidget(self.tabWidget)
self.widget = QtGui.QWidget(self.centralwidget)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Preferred, QtGui.QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(self.widget.sizePolicy().hasHeightForWidth())
self.widget.setSizePolicy(sizePolicy)
self.widget.setMinimumSize(QtCore.QSize(0, 30))
self.widget.setObjectName(_fromUtf8("widget"))
self.gridLayout = QtGui.QGridLayout(self.widget)
self.gridLayout.setMargin(0)
self.gridLayout.setObjectName(_fromUtf8("gridLayout"))
spacerItem = QtGui.QSpacerItem(40, 20, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum)
self.gridLayout.addItem(spacerItem, 0, 0, 1, 1)
spacerItem1 = QtGui.QSpacerItem(40, 20, QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Minimum)
self.gridLayout.addItem(spacerItem1, 0, 3, 1, 1)
self.pushButton_refresh = QtGui.QPushButton(self.widget)
self.pushButton_refresh.setObjectName(_fromUtf8("pushButton_refresh"))
self.gridLayout.addWidget(self.pushButton_refresh, 0, 1, 1, 1)
self.pushButton_new = QtGui.QPushButton(self.widget)
self.pushButton_new.setObjectName(_fromUtf8("pushButton_new"))
self.gridLayout.addWidget(self.pushButton_new, 0, 2, 1, 1)
self.verticalLayout.addWidget(self.widget)
frm_MainWindow.setCentralWidget(self.centralwidget)
self.menubar = QtGui.QMenuBar(frm_MainWindow)
self.menubar.setEnabled(True)
self.menubar.setGeometry(QtCore.QRect(0, 0, 315, 22))
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Preferred)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(self.menubar.sizePolicy().hasHeightForWidth())
self.menubar.setSizePolicy(sizePolicy)
self.menubar.setMinimumSize(QtCore.QSize(0, 0))
self.menubar.setDefaultUp(False)
self.menubar.setObjectName(_fromUtf8("menubar"))
self.menu_WeCase = QtGui.QMenu(self.menubar)
self.menu_WeCase.setObjectName(_fromUtf8("menu_WeCase"))
self.menuHelp = QtGui.QMenu(self.menubar)
self.menuHelp.setObjectName(_fromUtf8("menuHelp"))
self.menuO_ptions = QtGui.QMenu(self.menubar)
self.menuO_ptions.setObjectName(_fromUtf8("menuO_ptions"))
frm_MainWindow.setMenuBar(self.menubar)
self.action_About = QtGui.QAction(frm_MainWindow)
icon1 = QtGui.QIcon()
icon1.addPixmap(QtGui.QPixmap(_fromUtf8(":/IMG/img/where_s_my_weibo.svg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.action_About.setIcon(icon1)
self.action_About.setObjectName(_fromUtf8("action_About"))
self.action_Refresh = QtGui.QAction(frm_MainWindow)
self.action_Refresh.setObjectName(_fromUtf8("action_Refresh"))
self.action_Log_out = QtGui.QAction(frm_MainWindow)
self.action_Log_out.setObjectName(_fromUtf8("action_Log_out"))
self.action_Exit = QtGui.QAction(frm_MainWindow)
icon2 = QtGui.QIcon()
icon2.addPixmap(QtGui.QPixmap(_fromUtf8(":/IMG/img/application-exit.svg")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.action_Exit.setIcon(icon2)
self.action_Exit.setObjectName(_fromUtf8("action_Exit"))
self.action_Settings = QtGui.QAction(frm_MainWindow)
icon3 = QtGui.QIcon()
icon3.addPixmap(QtGui.QPixmap(_fromUtf8(":/IMG/img/preferences-other.png")), QtGui.QIcon.Normal, QtGui.QIcon.Off)
self.action_Settings.setIcon(icon3)
self.action_Settings.setObjectName(_fromUtf8("action_Settings"))
self.menu_WeCase.addAction(self.action_Refresh)
self.menu_WeCase.addSeparator()
self.menu_WeCase.addAction(self.action_Log_out)
self.menu_WeCase.addAction(self.action_Exit)
self.menuHelp.addAction(self.action_About)
self.menuO_ptions.addAction(self.action_Settings)
self.menubar.addAction(self.menu_WeCase.menuAction())
self.menubar.addAction(self.menuO_ptions.menuAction())
self.menubar.addAction(self.menuHelp.menuAction())
self.retranslateUi(frm_MainWindow)
self.tabWidget.setCurrentIndex(0)
QtCore.QObject.connect(self.action_Exit, QtCore.SIGNAL(_fromUtf8("triggered()")), frm_MainWindow.close)
QtCore.QObject.connect(self.action_About, QtCore.SIGNAL(_fromUtf8("triggered()")), frm_MainWindow.showAbout)
QtCore.QObject.connect(self.action_Settings, QtCore.SIGNAL(_fromUtf8("triggered()")), frm_MainWindow.showSettings)
QtCore.QObject.connect(self.action_Log_out, QtCore.SIGNAL(_fromUtf8("triggered()")), frm_MainWindow.logout)
QtCore.QObject.connect(self.action_Refresh, QtCore.SIGNAL(_fromUtf8("triggered()")), frm_MainWindow.refresh)
QtCore.QObject.connect(self.pushButton_refresh, QtCore.SIGNAL(_fromUtf8("clicked()")), frm_MainWindow.refresh)
QtCore.QObject.connect(self.pushButton_new, QtCore.SIGNAL(_fromUtf8("clicked()")), frm_MainWindow.postTweet)
QtCore.QObject.connect(frm_MainWindow, QtCore.SIGNAL(_fromUtf8("timelineLoaded(int)")), frm_MainWindow.moveToTop)
QtCore.QObject.connect(frm_MainWindow, QtCore.SIGNAL(_fromUtf8("imageLoaded(QString)")), frm_MainWindow.setLoaded)
QtCore.QObject.connect(frm_MainWindow, QtCore.SIGNAL(_fromUtf8("tabTextChanged(int,QString)")), frm_MainWindow.setTabText)
QtCore.QMetaObject.connectSlotsByName(frm_MainWindow)
class NewpostWindow(QtGui.QDialog, Ui_NewPostWindow):
image = None
apiError = QtCore.pyqtSignal(Exception)
commonError = QtCore.pyqtSignal(str, str)
sendSuccessful = QtCore.pyqtSignal()
userClicked = QtCore.pyqtSignal(UserItem, bool)
tagClicked = QtCore.pyqtSignal(str, bool)
tweetRefreshed = QtCore.pyqtSignal()
tweetRejected = QtCore.pyqtSignal()
def __init__(self, action="new", tweet=None, parent=None):
super(NewpostWindow, self).__init__(parent)
self.setAttribute(QtCore.Qt.WA_QuitOnClose, False)
self.client = const.client
self.tweet = tweet
self.action = action
self.setupUi(self)
self.textEdit.callback = self.mentions_suggest
self.textEdit.mention_flag = "@"
self.notify = Notify(timeout=1)
self._sent = False
self.apiError.connect(self.showException)
self.sendSuccessful.connect(self.sent)
self.commonError.connect(self.showErrorMessage)
self.errorWindow = APIErrorWindow(self)
if self.action not in ["new", "reply"]:
self.tweetRefreshed.connect(self._create_tweetWidget)
self.tweetRejected.connect(
lambda: self.pushButton_send.setEnabled(False))
self._refresh()
def setupUi(self, widget):
super(NewpostWindow, self).setupUi(widget)
self.sendAction = QtGui.QAction(self)
self.sendAction.triggered.connect(self.send)
self.sendAction.setShortcut(QtGui.QKeySequence("Ctrl+Return"))
self.addAction(self.sendAction)
self.checkChars()
self.setupButtons()
def setupButtons(self):
# Disabled is the default state of buttons
self.pushButton_picture.setEnabled(False)
self.chk_repost.setEnabled(False)
self.chk_comment.setEnabled(False)
self.chk_comment_original.setEnabled(False)
if self.action == "new":
assert (not self.tweet) # Shouldn't have a tweet object.
self.pushButton_picture.setEnabled(True)
elif self.action == "retweet":
self.chk_comment.setEnabled(True)
if self.tweet.type == TweetItem.RETWEET:
self.textEdit.setText(self.tweet.append_existing_replies())
self.chk_comment_original.setEnabled(True)
elif self.action == "comment":
self.chk_repost.setEnabled(True)
if self.tweet.type == TweetItem.RETWEET:
self.chk_comment_original.setEnabled(True)
elif self.action == "reply":
self.chk_repost.setEnabled(True)
if self.tweet.original.type == TweetItem.RETWEET:
self.chk_comment_original.setEnabled(True)
else:
assert False
@async
def _refresh(self):
# The read count is not a real-time value. So refresh it now.
try:
self.tweet.refresh()
except APIError as e:
self.errorWindow.raiseException.emit(e)
self.tweetRejected.emit()
return
self.tweetRefreshed.emit()
def _create_tweetWidget(self):
if self.action == "comment" and self.tweet.comments_count:
self.tweetWidget = SingleTweetWidget(self.tweet,
["image", "original"], self)
self.replyModel = TweetUnderCommentModel(
self.client.api("comments/show"), self.tweet.id, self)
elif self.action == "retweet" and self.tweet.retweets_count:
if self.tweet:
self.tweetWidget = SingleTweetWidget(self.tweet,
["image", "original"],
self)
elif self.tweet.original:
self.tweetWidget = SingleTweetWidget(self.tweet.original,
["image", "original"],
self)
self.replyModel = TweetRetweetModel(
self.client.api("statuses/repost_timeline"), self.tweet.id,
self)
else:
return
self.replyModel.load()
self.splitter = QtGui.QSplitter(self)
self.splitter.setOrientation(QtCore.Qt.Vertical)
self.verticalLayout.insertWidget(0, self.splitter)
self.tweetWidget.setSizePolicy(QtGui.QSizePolicy.Expanding,
QtGui.QSizePolicy.Expanding)
self.splitter.addWidget(self.tweetWidget)
self.commentsWidget = TweetListWidget(self, ["image", "original"])
self.commentsWidget.setModel(self.replyModel)
self.commentsWidget.scrollArea.setMinimumSize(20, 200)
self.commentsWidget.userClicked.connect(self.userClicked)
self.commentsWidget.tagClicked.connect(self.tagClicked)
self.splitter.addWidget(self.commentsWidget)
self.splitter.addWidget(self.textEdit)
def mentions_suggest(self, text):
ret_users = []
try:
word = text.split(' ')[-1]
word = word.split('@')[-1]
except IndexError:
return []
if not word.strip():
return []
users = self.client.api("search/suggestions/at_users").get(q=word,
type=0)
for user in users:
ret_users.append("@" + user['nickname'])
return ret_users
def sent(self):
self._sent = True
self.close()
def send(self):
self.pushButton_send.setEnabled(False)
if self.action == "new":
self.new()
elif self.action == "retweet":
self.retweet()
elif self.action == "comment":
self.comment()
elif self.action == "reply":
self.reply()
else:
# If action is in other types, it must be a mistake.
assert False
@async
def retweet(self):
text = self.textEdit.toPlainText()
comment = int(self.chk_comment.isChecked())
comment_ori = int(self.chk_comment_original.isChecked())
try:
self.tweet.retweet(text, comment, comment_ori)
except APIError as e:
self.apiError.emit(e)
return
self.notify.showMessage(self.tr("WeCase"), self.tr("Retweet Success!"))
self.sendSuccessful.emit()
@async
def comment(self):
text = self.textEdit.toPlainText()
retweet = int(self.chk_repost.isChecked())
comment_ori = int(self.chk_comment_original.isChecked())
try:
self.tweet.comment(text, comment_ori, retweet)
except APIError as e:
self.apiError.emit(e)
return
self.notify.showMessage(self.tr("WeCase"), self.tr("Comment Success!"))
self.sendSuccessful.emit()
@async
def reply(self):
text = self.textEdit.toPlainText()
comment_ori = int(self.chk_comment_original.isChecked())
retweet = int(self.chk_repost.isChecked())
try:
self.tweet.reply(text, comment_ori, retweet)
except APIError as e:
self.apiError.emit(e)
return
self.notify.showMessage(self.tr("WeCase"), self.tr("Reply Success!"))
self.sendSuccessful.emit()
@async
def new(self):
text = self.textEdit.toPlainText()
try:
if self.image:
try:
if getsize(self.image) > const.MAX_IMAGE_BYTES:
raise ValueError
with open(self.image, "rb") as image:
self.client.api("statuses/upload").post(status=text,
pic=image)
except (OSError, IOError):
self.commonError.emit(
self.tr("File not found"),
self.tr("No such file: %s") % self.image)
self.addImage() # In fact, remove image...
return
except ValueError:
self.commonError.emit(
self.tr("Too large size"),
self.tr("This image is too large to upload: %s") %
self.image)
self.addImage() # In fact, remove image...
return
else:
self.client.api("statuses/update").post(status=text)
self.notify.showMessage(self.tr("WeCase"),
self.tr("Tweet Success!"))
self.sendSuccessful.emit()
except APIError as e:
self.apiError.emit(e)
return
self.image = None
def addImage(self):
ACCEPT_TYPE = self.tr("Images") + "(*.png *.jpg *.jpeg *.bmp *.gif)"
if self.image:
self.image = None
self.pushButton_picture.setText(self.tr("&Picture"))
else:
self.image = QtGui.QFileDialog.getOpenFileName(self,
self.tr("Choose a"
" image"),
filter=ACCEPT_TYPE)
# user may cancel the dialog, so check again
if self.image:
self.pushButton_picture.setText(self.tr("Remove the picture"))
self.textEdit.setFocus()
def showException(self, e):
if "Text too long" in str(e):
QtGui.QMessageBox.warning(None, self.tr("Text too long!"),
self.tr("Please remove some text."))
else:
self.errorWindow.raiseException.emit(e)
self.pushButton_send.setEnabled(True)
def showErrorMessage(self, title, text):
QtGui.QMessageBox.warning(self, title, text)
def showSmiley(self):
wecase_smiley = FaceWindow()
if wecase_smiley.exec_():
self.textEdit.textCursor().insertText(wecase_smiley.faceName)
self.textEdit.setFocus()
def checkChars(self):
"""Check textEdit's characters.
If it larger than 140, Send Button will be disabled
and label will show red chars."""
text = self.textEdit.toPlainText()
numLens = 140 - tweetLength(text)
if numLens == 140 and (not self.action == "retweet"):
# you can not send empty tweet, except retweet
self.pushButton_send.setEnabled(False)
elif numLens >= 0:
# length is okay
self.label.setStyleSheet("color:black;")
self.pushButton_send.setEnabled(True)
else:
# text is too long
self.label.setStyleSheet("color:red;")
self.pushButton_send.setEnabled(False)
self.label.setText(str(numLens))
def reject(self):
self.close()
def closeEvent(self, event):
# We have unsend text.
if (not self._sent) and (self.textEdit.toPlainText()):
choice = QtGui.QMessageBox.question(
self, self.tr("Close?"), self.tr("All unpost text will lost."),
QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
if choice == QtGui.QMessageBox.No:
event.ignore()
else:
self.setParent(None)
