Пример #1
0
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        self.server_model = ServerModel()
        self.platform_model = PlatformModel()
        self.solution_model = SolutionModel()

        self.upload_thread = UploadThread()
        self.upload_thread.finished.connect(self.on_upload_thread_finished)
        self.upload_thread.information.connect(
            self.on_upload_thread_information)

        super(MainWindow, self).__init__(parent)
        self.setupUi(self)
        self.initUi()
        self.initData()
        self.createContextMenu()
Пример #2
0
 def __init__(self, parent=None):
     """
     Constructor
     
     @param parent reference to the parent widget (QWidget)
     """
     self.server_model = ServerModel()
     self.platform_model = PlatformModel()
     self.solution_model = SolutionModel()
     
     self.upload_thread = UploadThread()
     self.upload_thread.finished.connect(self.on_upload_thread_finished)
     self.upload_thread.information.connect(self.on_upload_thread_information)
     
     super(MainWindow,self).__init__(parent)
     self.setupUi(self)      
     self.initUi()
     self.initData()
     self.createContextMenu()
Пример #3
0
def echo():
    global LAST_UPDATE_ID

    # Request updates from last updated_id
    for update in bot.getUpdates(offset=LAST_UPDATE_ID):
        if LAST_UPDATE_ID < update.update_id:
            # chat_id is required to reply any message
            chat_id = update.message.chat_id
            message = update.message.text

            if (message):
                if '/start' in message or '/help' in message or '/list' in message or '/commands' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    bot.sendMessage(chat_id=chat_id,
                                    text=help.encode('utf8'),
                                    disable_web_page_preview=True)

                ##################----- PublicPlugins -----##################

                #Require API keys
                '''Youtube search'''
                if '/yt' in message or '/youtube' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    replacer = {'/youtube': '', '/yt': ''}
                    search_term = replace_all(message, replacer)
                    if len(search_term) < 1:
                        bot.sendMessage(
                            chat_id=chat_id,
                            text=
                            'Youtube API calls are costly. Use it like /yt keywords; Ex, /yt Iron Maiden'
                        )
                    else:
                        bot.sendMessage(
                            chat_id=chat_id,
                            text=youtube(search_term).encode('utf8'))
                '''Twitter latest tweets of user'''
                if '/twitter' in message or '/tw' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    replacer = {'/twitter': '', '/tw': ''}
                    username = replace_all(message, replacer)
                    if len(username) < 1:
                        bot.sendMessage(
                            chat_id=chat_id,
                            text='Use it like: /tw username; Ex, /tw pytacular'
                        )
                    else:
                        bot.sendMessage(chat_id=chat_id,
                                        text=twitter(username).encode('utf8'))
                '''Gets twitter trends by country /tt countryname, ex /tt India '''
                if '/tt' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    replacer = {'/tt ': ''}
                    place = replace_all(message, replacer)
                    bot.sendMessage(chat_id=chat_id,
                                    text=twittertrends(place).encode('utf8'))
                '''Search twitter for top 4 related tweets. /ts #Privacy'''
                if '/ts' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    replacer = {'/ts ': ''}
                    search_term = replace_all(message, replacer)
                    bot.sendMessage(
                        chat_id=chat_id,
                        text=twittersearch(search_term).encode('utf8'))
                '''Instagram latest posts of user'''
                if '/insta' in message or '/instagram' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    bot.sendMessage(
                        chat_id=chat_id,
                        text=
                        'Instagram has restricted API access. This will not work anymore. Sorry :('
                    )
                '''Game "Hot or Not" '''
                if '/hon' in message or '/hotornot' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    custom_keyboard = [[
                        telegram.Emoji.THUMBS_UP_SIGN,
                        telegram.Emoji.THUMBS_DOWN_SIGN
                    ]]
                    reply_markup = telegram.ReplyKeyboardMarkup(
                        custom_keyboard,
                        resize_keyboard=True,
                        one_time_keyboard=True)
                    image = imgur_hon()
                    bot.sendMessage(
                        chat_id=chat_id,
                        text='Fetched from Imgur Subreddit r/models : ' +
                        image,
                        reply_markup=reply_markup)
                '''Bing Image Search'''
                if '/image' in message or '/img' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    replacer = {'/image': '', '/img': ''}
                    search_term = replace_all(message, replacer)
                    pic_link = bingsearch(search_term, 'Image')
                    bot.sendMessage(chat_id=chat_id, text=pic_link)
                '''Microsoft translator'''
                if '/translate' in message:
                    message = message.replace('/translate', '').encode('utf8')
                    message_broken = shlex.split(message)
                    error = 'Not enough parameters. Use, /translate en hi "Hello world" or /translate help to know more'
                    if not len(message_broken) < 1:
                        if message_broken[0] == 'help':
                            help_string = """ Example, /translate en hi "Hello world"
                                    ar-Arabic | bs-Latn-Bosnian (Latin) | bg-Bulgarian | ca-Catalan | zh-CHS-Chinese Simplified |
                                    zh-CHT-Chinese Traditional|hr-Croatian | cs-Czech | da-Danish | nl-Dutch |en-English | cy-Welsh |
                                    et-Estonian | fi-Finnish | fr-French | de-German | el-Greek | ht-Haitian Creole | he-Hebrew |
                                    hi-Hindi | mww-Hmong Daw | hu-Hungarian | id-Indonesian | it-Italian | ja-Japanese | tlh-Klingon |
                                    tlh - Qaak-Klingon (pIqaD) | ko-Korean | lv-Latvian | lt-Lithuanian | ms-Malay | mt-Maltese |
                                    no-Norwegian | fa-Persian | pl-Polish | pt-Portuguese | otq-Querétaro Otomi | ro-Romanian |
                                    ru-Russian | sr-Cyrl-Serbian (Cyrillic) | sr-Latn-Serbian (Latin) | sk-Slovak | sl-Slovenian |
                                    es-Spanish | sv-Swedish | th-Thai | tr-Turkish | uk-Ukrainian | ur-Urdu | vi-Vietnamese |
                                    """
                            bot.sendMessage(chat_id=chat_id, text=help_string)
                        else:
                            if len(message_broken) < 3:
                                bot.sendMessage(chat_id=chat_id, text=error)
                            else:
                                lang_from = message_broken[0]
                                lang_to = message_broken[1]
                                lang_text = message_broken[2]
                                print(lang_from + lang_to + lang_text)
                                bot.sendMessage(chat_id=chat_id,
                                                text=btranslate(
                                                    lang_text, lang_from,
                                                    lang_to))
                    else:
                        bot.sendMessage(chat_id=chat_id, text=error)
                '''Random cat pic'''
                if '/cats' in message:
                    bot.sendMessage(
                        chat_id=chat_id,
                        text='Hold on, digging out a random cat pic!')
                    url = "http://thecatapi.com/api/images/get?api_key=" + cat_API_key + "&format=xml"
                    xml_src = urllib.request.urlopen(url)
                    data = xml_src.read()
                    xml_src.close()
                    data = xmltodict.parse(data)
                    piclink = data['response']['data']['images']['image'][
                        'url']
                    source_url = data['response']['data']['images']['image'][
                        'source_url']
                    threadobjcats = UploadThread(bot,
                                                 chat_id,
                                                 piclink,
                                                 caption=source_url)
                    threadobjcats.setName('catsthread')
                    threadobjcats.start()

                # Don't need an API key
                '''Google search'''
                if '/google' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    search_term = message.replace('/google', '')
                    if len(search_term) < 1:
                        bot.sendMessage(
                            chat_id=chat_id,
                            text='Use it like: /google what is a bot')
                    else:
                        bot.sendMessage(chat_id=chat_id,
                                        text=google(search_term))
                '''Wikipedia search'''
                if '/wiki' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    search_term = message.replace('/wiki ', '')
                    if len(search_term) < 1:
                        bot.sendMessage(chat_id=chat_id,
                                        text='Use it like: /wiki Anaconda')
                    else:
                        reply = wiki(search_term)
                        bot.sendMessage(chat_id=chat_id, text=reply)
                        if ("Cannot acces link!" in reply):
                            reply = "No wikipedia article on that but got some google results for you \n" + google(
                                message)
                            bot.sendMessage(chat_id=chat_id, text=reply)
                '''Weather by city,state'''
                if '/weather' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    reply = weather(message)
                    bot.sendMessage(chat_id=chat_id, text=reply)
                '''Github public feed of any user'''
                if '/github' in message or '/gh' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    replacer = {'/github': '', '/gh': ''}
                    username = replace_all(message, replacer)
                    if len(username) < 1:
                        bot.sendMessage(
                            chat_id=chat_id,
                            text=
                            'Use it like: /github username Ex, /github bhavyanshu or /gh bhavyanshu'
                        )
                    else:
                        bot.sendMessage(chat_id=chat_id,
                                        text=gitfeed(username))
                '''Giphy to search for gif by keyword'''
                if '/giphy' in message or '/gif' in message:
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    replacer = {'/giphy': '', '/gif': ''}
                    search_term = replace_all(message, replacer)
                    if len(search_term) < 1:
                        bot.sendMessage(
                            chat_id=chat_id,
                            text=
                            'Use it like: /giphy keyword ; Ex, /giphy cats or /gif cats'
                        )
                    else:
                        img = translate(search_term)
                        print(img.fixed_height.downsampled.url)
                        bot.sendMessage(
                            chat_id=chat_id,
                            text=
                            'Hang in there. Fetching gif..-Powered by GIPHY!')
                        bot.sendChatAction(
                            chat_id=chat_id,
                            action=telegram.ChatAction.UPLOAD_PHOTO)
                        threadobjgiphy = UploadThread(
                            bot, chat_id,
                            img.fixed_height.downsampled.url.encode('utf-8'))
                        threadobjgiphy.setName('giphythread')
                        threadobjgiphy.start()
                '''Basic calculator'''
                if '/calc' in message:
                    head, sep, tail = message.partition('/')
                    input_nums = tail.replace('calc', '')
                    input_nums = input_nums.replace('\'', '')
                    finalexp = shlex.split(input_nums)
                    exp = finalexp[0]
                    bot.sendChatAction(chat_id=chat_id,
                                       action=telegram.ChatAction.TYPING)
                    error = 'You think I can compute apple+mongo? Don\'t add alphabet in between please. Use like, /calc 2+2-5(4+8)'
                    if not exp:
                        bot.sendMessage(
                            chat_id=chat_id,
                            text='Y u no type math expression? >.<')
                    elif re.search('[a-zA-Z]', exp):
                        bot.sendMessage(chat_id=chat_id, text=error)
                    else:
                        bot.sendMessage(chat_id=chat_id, text=calculate(exp))

                # Updates global offset to get the new updates
                LAST_UPDATE_ID = update.update_id
Пример #4
0
def echo():
    global LAST_UPDATE_ID

    # Request updates from last updated_id
    for update in bot.getUpdates(offset=LAST_UPDATE_ID):
        if LAST_UPDATE_ID < update.update_id:
            # chat_id is required to reply any message
            chat_id = update.message.chat_id
            message = update.message.text

            if (message):
                if '/start' in message or '/help' in message or '/list' in message or '/commands' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    bot.sendMessage(chat_id=chat_id,text=help.encode('utf8'),disable_web_page_preview=True)

                ##################----- PublicPlugins -----##################

                #Require API keys

                '''Youtube search'''
                if '/yt' in message or '/youtube' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    replacer = {'/youtube':'','/yt':''}
                    search_term = replace_all(message,replacer)
                    if len(search_term)<1:
                       bot.sendMessage(chat_id=chat_id,text='Youtube API calls are costly. Use it like /yt keywords; Ex, /yt Iron Maiden')
                    else:
                       bot.sendMessage(chat_id=chat_id,text=youtube(search_term).encode('utf8'))

                '''Twitter latest tweets of user'''
                if '/twitter' in message or '/tw' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    replacer = {'/twitter':'','/tw':''}
                    username = replace_all(message,replacer)
                    if len(username)<1:
                        bot.sendMessage(chat_id=chat_id,text='Use it like: /tw username; Ex, /tw pytacular')
                    else:
                        bot.sendMessage(chat_id=chat_id,text=twitter(username).encode('utf8'))

                '''Gets twitter trends by country /tt countryname, ex /tt India '''
                if '/tt' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    replacer = {'/tt ':''}
                    place = replace_all(message,replacer)
                    bot.sendMessage(chat_id=chat_id,text=twittertrends(place).encode('utf8'))

                '''Search twitter for top 4 related tweets. /ts #Privacy'''
                if '/ts' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    replacer = {'/ts ':''}
                    search_term = replace_all(message,replacer)
                    bot.sendMessage(chat_id=chat_id,text=twittersearch(search_term).encode('utf8'))

                '''Instagram latest posts of user'''
                if '/insta' in message or '/instagram' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    bot.sendMessage(chat_id=chat_id,text='Instagram has restricted API access. This will not work anymore. Sorry :(')

                '''Game "Hot or Not" '''
                if '/hon' in message or '/hotornot' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    custom_keyboard = [[ telegram.Emoji.THUMBS_UP_SIGN, telegram.Emoji.THUMBS_DOWN_SIGN ]]
                    reply_markup = telegram.ReplyKeyboardMarkup(custom_keyboard,resize_keyboard=True,one_time_keyboard=True)
                    image = imgur_hon()
                    bot.sendMessage(chat_id=chat_id,text='Fetched from Imgur Subreddit r/models : '+image, reply_markup=reply_markup)

                '''Bing Image Search'''
                if '/image' in message or '/img' in message:
                    bot.sendChatAction(chat_id=chat_id,action=telegram.ChatAction.TYPING)
                    replacer = {'/image':'','/img':''}
                    search_term = replace_all(message,replacer)
                    pic_link = bingsearch(search_term,'Image')
                    bot.sendMessage(chat_id=chat_id,text=pic_link)

                '''Microsoft translator'''
                if '/translate' in message:
                    message = message.replace('/translate','').encode('utf8')
                    message_broken = shlex.split(message)
                    error = 'Not enough parameters. Use, /translate en hi "Hello world" or /translate help to know more'
                    if not len(message_broken)<1:
                        if message_broken[0] == 'help':
                            help_string = """ Example, /translate en hi "Hello world"
                                    ar-Arabic | bs-Latn-Bosnian (Latin) | bg-Bulgarian | ca-Catalan | zh-CHS-Chinese Simplified |
                                    zh-CHT-Chinese Traditional|hr-Croatian | cs-Czech | da-Danish | nl-Dutch |en-English | cy-Welsh |
                                    et-Estonian | fi-Finnish | fr-French | de-German | el-Greek | ht-Haitian Creole | he-Hebrew |
                                    hi-Hindi | mww-Hmong Daw | hu-Hungarian | id-Indonesian | it-Italian | ja-Japanese | tlh-Klingon |
                                    tlh - Qaak-Klingon (pIqaD) | ko-Korean | lv-Latvian | lt-Lithuanian | ms-Malay | mt-Maltese |
                                    no-Norwegian | fa-Persian | pl-Polish | pt-Portuguese | otq-Querétaro Otomi | ro-Romanian |
                                    ru-Russian | sr-Cyrl-Serbian (Cyrillic) | sr-Latn-Serbian (Latin) | sk-Slovak | sl-Slovenian |
                                    es-Spanish | sv-Swedish | th-Thai | tr-Turkish | uk-Ukrainian | ur-Urdu | vi-Vietnamese |
                                    """
                            bot.sendMessage(chat_id=chat_id,text=help_string)
                        else:
                            if len(message_broken)<3:
                                bot.sendMessage(chat_id=chat_id,text=error)
                            else:
                                lang_from = message_broken[0]
                                lang_to = message_broken[1]
                                lang_text = message_broken[2]
                                print lang_from+lang_to+lang_text
                                bot.sendMessage(chat_id=chat_id,text=btranslate(lang_text,lang_from,lang_to))
                    else:
                        bot.sendMessage(chat_id=chat_id,text=error)

                '''Random cat pic'''
                if '/cats' in message:
                    bot.sendMessage(chat_id=chat_id,text='Hold on, digging out a random cat pic!')
                    url = "http://thecatapi.com/api/images/get?api_key="+cat_API_key+"&format=xml"
                    xml_src = urllib2.urlopen(url)
                    data = xml_src.read()
                    xml_src.close()
                    data = xmltodict.parse(data)
                    piclink =  data['response']['data']['images']['image']['url']
                    source_url = data['response']['data']['images']['image']['source_url']
                    threadobjcats = UploadThread(bot,chat_id,piclink,caption=source_url)
                    threadobjcats.setName('catsthread')
                    threadobjcats.start()

                # Don't need an API key

                '''Google search'''
                if '/google' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    search_term = message.replace('/google','')
                    if len(search_term)<1:
                        bot.sendMessage(chat_id=chat_id,text='Use it like: /google what is a bot')
                    else:
                        bot.sendMessage(chat_id=chat_id,text=google(search_term))
                '''Wikipedia search'''
                if '/wiki' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    search_term = message.replace('/wiki ','')
                    if len(search_term)<1:
                        bot.sendMessage(chat_id=chat_id,text='Use it like: /wiki Anaconda')
                    else:
                        reply=wiki(search_term)
                        bot.sendMessage(chat_id=chat_id,text=reply)
                        if ("Cannot acces link!" in reply):
                            reply="No wikipedia article on that but got some google results for you \n"+google(message)
                            bot.sendMessage(chat_id=chat_id,text=reply)

                '''Weather by city,state'''
                if '/weather' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    reply=weather(message)
                    bot.sendMessage(chat_id=chat_id,text=reply)

                '''Github public feed of any user'''
                if '/github' in message or '/gh' in message:
                    bot.sendChatAction(chat_id=chat_id,action=telegram.ChatAction.TYPING)
                    replacer = {'/github':'','/gh':''}
                    username = replace_all(message,replacer)
                    if len(username)<1:
                        bot.sendMessage(chat_id=chat_id,text='Use it like: /github username Ex, /github bhavyanshu or /gh bhavyanshu')
                    else:
                        bot.sendMessage(chat_id=chat_id,text=gitfeed(username))

                '''Giphy to search for gif by keyword'''
                if '/giphy' in message or '/gif' in message:
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    replacer = {'/giphy':'','/gif':''}
                    search_term = replace_all(message,replacer)
                    if len(search_term)<1:
                        bot.sendMessage(chat_id=chat_id,text='Use it like: /giphy keyword ; Ex, /giphy cats or /gif cats')
                    else:
                        img = translate(search_term)
                        print img.fixed_height.downsampled.url
                        bot.sendMessage(chat_id=chat_id,text='Hang in there. Fetching gif..-Powered by GIPHY!')
                        bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.UPLOAD_PHOTO)
                        threadobjgiphy = UploadThread(bot,chat_id,img.fixed_height.downsampled.url.encode('utf-8'))
                        threadobjgiphy.setName('giphythread')
                        threadobjgiphy.start()

                '''Basic calculator'''
                if '/calc' in message:
                    head, sep, tail = message.partition('/')
                    input_nums = tail.replace('calc','')
                    input_nums = input_nums.replace('\'','')
                    finalexp = shlex.split(input_nums)
                    exp = finalexp[0]
                    bot.sendChatAction(chat_id=chat_id, action=telegram.ChatAction.TYPING)
                    error = 'You think I can compute apple+mongo? Don\'t add alphabet in between please. Use like, /calc 2+2-5(4+8)'
                    if not exp:
                        bot.sendMessage(chat_id=chat_id,text='Y u no type math expression? >.<')
                    elif re.search('[a-zA-Z]', exp):
                        bot.sendMessage(chat_id=chat_id,text=error)
                    else:
                        bot.sendMessage(chat_id=chat_id,text=calculate(exp))

                # Updates global offset to get the new updates
                LAST_UPDATE_ID = update.update_id
Пример #5
0
class MainWindow(QMainWindow, Ui_MainWindow):
    """
    Class documentation goes here.
    """
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        self.server_model = ServerModel()
        self.platform_model = PlatformModel()
        self.solution_model = SolutionModel()

        self.upload_thread = UploadThread()
        self.upload_thread.finished.connect(self.on_upload_thread_finished)
        self.upload_thread.information.connect(
            self.on_upload_thread_information)

        super(MainWindow, self).__init__(parent)
        self.setupUi(self)
        self.initUi()
        self.initData()
        self.createContextMenu()

    def initUi(self):
        self.file_list.setColumnWidth(0, 400)
        self.file_list.setColumnWidth(2, 150)

        self.selected_servers.setColumnWidth(2, 80)
        self.server_list.setColumnWidth(2, 80)

        self.server_list.hideColumn(0)
        self.selected_servers.hideColumn(0)

        self.server_list.keyPressEvent = self.on_server_list_keyPressEvent
        self.selected_servers.keyPressEvent = self.on_selected_servers_keyPressEvent

        self.statusBar.showMessage(u"状态")

    def initData(self):
        self.platform.addItems(self.platform_model.list())
        self.upload_solution.addItems(self.solution_model.list())

    def createContextMenu(self):
        self.server_list_contextmenu = QtGui.QMenu(self)
        self.server_list_contextmenu.addAction(self.action_move2left)
        self.server_list_contextmenu.addSeparator()
        self.server_list_contextmenu.addAction(self.action_add_server)
        self.server_list_contextmenu.addAction(self.action_edit_server)
        self.server_list_contextmenu.addAction(self.action_copy_server)
        self.server_list_contextmenu.addSeparator()
        self.server_list_contextmenu.addAction(self.action_delete_server)

    @pyqtSlot(QPoint)
    def on_server_list_customContextMenuRequested(self, point):
        """
        Slot documentation goes here.
        """
        point.setY(point.y() + 25)
        if self.server_list.rowCount():
            point.setX(point.x() + 15)
        self.server_list_contextmenu.exec_(self.server_list.mapToGlobal(point))

    @pyqtSlot()
    def on_start_upload_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.file_list.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加要上传的文件!")
            return

        if self.selected_servers.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加服务器到上传列表中!")
            return

        answer = QtGui.QMessageBox.warning(self, u"提示",
                                           u"上传将覆盖服务器设置路径下的同名文件,确定上传吗?",
                                           QtGui.QMessageBox.Yes,
                                           QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.No:
            return

        if self.upload_thread.isRunning() == False:
            files = [
                self.file_list.item(i, 0).text()
                for i in range(self.file_list.rowCount())
            ]
            server_ids = [
                str(self.selected_servers.item(i, 0).text())
                for i in range(self.selected_servers.rowCount())
            ]
            servers = self.server_model.findByIds(",".join(server_ids))
            self.upload_thread.setData(files, servers)
            self.upload_thread.start()

            self.stop_upload.setEnabled(True)
            self.start_upload.setEnabled(False)

    @pyqtSlot()
    def on_stop_upload_clicked(self):
        """
        Slot documentation goes here.
        """

        if self.upload_thread.isRunning() == True:
            answer = QtGui.QMessageBox.warning(self, u"提示", u"确定终止当前上传任务?",
                                               QtGui.QMessageBox.Yes,
                                               QtGui.QMessageBox.No)
            if answer == QtGui.QMessageBox.No:
                return

            self.start_upload.setEnabled(True)
            self.stop_upload.setEnabled(False)
            self.upload_thread.terminate()

    def on_upload_thread_finished(self):
        self.start_upload.setEnabled(True)
        self.stop_upload.setEnabled(False)

    @pyqtSlot(str)
    def on_upload_thread_information(self, info):
        time = QTime(QTime.currentTime())
        self.textBrowser.append(u"[%s] %s" % (time.toString(), info))

    @pyqtSlot(str)
    def on_upload_solution_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        if self.upload_solution.currentIndex() > 0:
            self.setSelectedServersData(self.solution_model.load(p0))
            self.remove_solution.setEnabled(True)
            self.refresh_server_list.clicked.emit(True)
        else:
            self.setSelectedServersData([])
            self.remove_solution.setEnabled(False)

    @pyqtSlot(str)
    def on_platform_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        self.setServerListData(self.server_model.findByPlatform(p0))

    @pyqtSlot()
    def on_move2left_clicked(self):
        """
        Slot documentation goes here.
        """
        rows = list(
            set([index.row() for index in self.server_list.selectedIndexes()]))
        for i in range(len(rows)):
            row = rows[i] - i
            index = self.selected_servers.rowCount()
            self.selected_servers.insertRow(index)
            self.selected_servers.setItem(
                index, 0,
                self.server_list.item(row, 0).clone())
            self.selected_servers.setItem(
                index, 1,
                self.server_list.item(row, 1).clone())
            self.selected_servers.setItem(
                index, 2,
                self.server_list.item(row, 2).clone())
            self.selected_servers.setItem(
                index, 3,
                self.server_list.item(row, 3).clone())
            self.server_list.removeRow(row)

    @pyqtSlot()
    def on_move2right_clicked(self):
        """
        Slot documentation goes here.
        """
        rows = list(
            set([
                index.row()
                for index in self.selected_servers.selectedIndexes()
            ]))
        for i in range(len(rows)):
            row = rows[i] - i
            self.selected_servers.removeRow(row)
        self.refresh_server_list.clicked.emit(True)

    @pyqtSlot()
    def on_menu_about_triggered(self):
        """
        Slot documentation goes here.
        """
        QtGui.QMessageBox.about(self, u"FTP Client", u"Powered by ZLH!")

    @pyqtSlot()
    def on_menu_clear_log_triggered(self):
        self.textBrowser.clear()

    @pyqtSlot()
    def on_add_solution_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.selected_servers.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加服务器到上传列表中!")
            return

        text, ok = QtGui.QInputDialog.getText(self, u"添加", u"上传方案名称:")
        if ok == True:
            if self.solution_model.check(text) > 0:
                QtGui.QMessageBox.critical(self, u"提示", u"上传方案名称已存在!")
                return
            servers = [
                str(self.selected_servers.item(i, 0).text())
                for i in range(self.selected_servers.rowCount())
            ]
            if self.solution_model.add(text, ",".join(servers)):
                self.upload_solution.addItem(text)
                self.upload_solution.setCurrentIndex(
                    self.upload_solution.count() - 1)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"操作失败!")

    @pyqtSlot()
    def on_remove_solution_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.upload_solution.currentIndex() == 0:
            QtGui.QMessageBox.information(self, u"FTP Client", u"请选择要删除的上传方案!")
            return

        answer = QtGui.QMessageBox.question(self, u"提示", u"确定删除当前选中上传方案?",
                                            QtGui.QMessageBox.Yes,
                                            QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:
            name = self.upload_solution.currentText()
            if self.solution_model.delete(name):
                self.upload_solution.removeItem(
                    self.upload_solution.currentIndex())
                self.selected_servers.setRowCount(0)
                self.refresh_server_list.clicked.emit(True)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")

    @pyqtSlot()
    def on_refresh_server_list_clicked(self):
        """
        Slot documentation goes here.
        """
        platform = self.platform.currentText()
        self.setServerListData(self.server_model.findByPlatform(platform))

    @pyqtSlot()
    def on_server_list_itemSelectionChanged(self):
        rows = self.server_list.selectedIndexes()
        if (rows):
            self.move2left.setEnabled(True)
            self.action_move2left.setEnabled(True)
            self.action_edit_server.setEnabled(True)
            self.action_copy_server.setEnabled(True)
            self.action_delete_server.setEnabled(True)
        else:
            self.move2left.setEnabled(False)
            self.action_move2left.setEnabled(False)
            self.action_edit_server.setEnabled(False)
            self.action_copy_server.setEnabled(False)
            self.action_delete_server.setEnabled(False)

    @pyqtSlot()
    def on_selected_servers_itemSelectionChanged(self):
        rows = self.selected_servers.selectedIndexes()
        if rows:
            self.move2right.setEnabled(True)
        else:
            self.move2right.setEnabled(False)

    def on_server_list_keyPressEvent(self, event):
        if event.key() == Qt.Key_Escape:
            self.server_list.clearSelection()
        elif event.key() == Qt.Key_Left:
            self.move2left.clicked.emit(True)

    def on_selected_servers_keyPressEvent(self, event):
        if event.key() == Qt.Key_Escape:
            self.selected_servers.clearSelection()
        elif event.key() == Qt.Key_Delete or event.key() == Qt.Key_Right:
            self.move2right.clicked.emit(True)

    @pyqtSlot()
    def on_action_add_server_triggered(self):
        """
        Slot documentation goes here.
        """
        self.server = Server()
        self.server.saved.connect(self.on_server_saved)
        self.server.show()

    @pyqtSlot()
    def on_action_edit_server_triggered(self):
        """
        Slot documentation goes here.
        """
        server_id = self.server_list.item(self.server_list.currentRow(),
                                          0).text()

        self.server = Server()
        self.server.saved.connect(self.on_server_saved)
        self.server.setData(self.server_model.item(server_id))
        self.server.show()

    @pyqtSlot()
    def on_action_copy_server_triggered(self):
        """
        Slot documentation goes here.
        """
        server_id = self.server_list.item(self.server_list.currentRow(),
                                          0).text()
        server_data = self.server_model.item(server_id)
        server_data["id"] = None

        self.server = Server()
        self.server.saved.connect(self.on_server_saved)
        self.server.setData(server_data)
        self.server.show()

    @pyqtSlot()
    def on_action_delete_server_triggered(self):
        """
        Slot documentation goes here.
        """
        index = self.server_list.currentRow()
        platform = self.server_list.item(index, 1).text()
        server = self.server_list.item(index, 2).text()
        answer = QtGui.QMessageBox.question(
            self, u"提示", u"确定删除 %s %s?" % (platform, server),
            QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:
            server_id = self.server_list.item(index, 0).text()
            if self.server_model.delete(server_id):
                self.server_list.removeRow(index)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"操作失败!")

    def on_server_saved(self):
        self.refresh_server_list.clicked.emit(True)

    @pyqtSlot()
    def on_action_move2left_triggered(self):
        self.move2left.clicked.emit(True)

    def setServerListData(self, data):
        self.server_list.setRowCount(0)
        for row in data:
            if self.inSelectedServers(row["id"]) == False:
                index = self.server_list.rowCount()
                self.server_list.insertRow(index)
                self.server_list.setItem(index, 0, QTableWidgetItem(row["id"]))
                self.server_list.setItem(index, 1,
                                         QTableWidgetItem(row["platform"]))
                self.server_list.setItem(index, 2,
                                         QTableWidgetItem(row["server_name"]))
                self.server_list.setItem(index, 3,
                                         QTableWidgetItem(row["path"]))

    def setSelectedServersData(self, data):
        self.selected_servers.setRowCount(0)
        for row in data:
            index = self.selected_servers.rowCount()
            self.selected_servers.insertRow(index)
            self.selected_servers.setItem(index, 0,
                                          QTableWidgetItem(row["id"]))
            self.selected_servers.setItem(index, 1,
                                          QTableWidgetItem(row["platform"]))
            self.selected_servers.setItem(index, 2,
                                          QTableWidgetItem(row["server_name"]))
            self.selected_servers.setItem(index, 3,
                                          QTableWidgetItem(row["path"]))

        self.refresh_server_list.clicked.emit(True)

    def inSelectedServers(self, server_id):
        for i in range(self.selected_servers.rowCount()):
            if self.selected_servers.item(i, 0).text() == server_id:
                return True
        return False
Пример #6
0
class MainWindow(QMainWindow, Ui_MainWindow):
    
    """
    Class documentation goes here.
    """
    def __init__(self, parent=None):
        """
        Constructor
        
        @param parent reference to the parent widget (QWidget)
        """
        self.server_model = ServerModel()
        self.platform_model = PlatformModel()
        self.solution_model = SolutionModel()
        
        self.upload_thread = UploadThread()
        self.upload_thread.finished.connect(self.on_upload_thread_finished)
        self.upload_thread.information.connect(self.on_upload_thread_information)
        
        super(MainWindow,self).__init__(parent)
        self.setupUi(self)      
        self.initUi()
        self.initData()
        self.createContextMenu()
        
        
    def initUi(self):        
        self.file_list.setColumnWidth(0, 400)
        self.file_list.setColumnWidth(2, 150)
        
        self.selected_servers.setColumnWidth(2, 80)
        self.server_list.setColumnWidth(2, 80)
        
        self.server_list.hideColumn(0)        
        self.selected_servers.hideColumn(0)
        
        self.server_list.keyPressEvent = self.on_server_list_keyPressEvent
        self.selected_servers.keyPressEvent = self.on_selected_servers_keyPressEvent
        
        self.statusBar.showMessage(u"状态")
        
    def initData(self):
        self.platform.addItems(self.platform_model.list())        
        self.upload_solution.addItems(self.solution_model.list())   
                
    def createContextMenu(self):        
        self.server_list_contextmenu = QtGui.QMenu(self)  
        self.server_list_contextmenu.addAction(self.action_move2left) 
        self.server_list_contextmenu.addSeparator()
        self.server_list_contextmenu.addAction(self.action_add_server)  
        self.server_list_contextmenu.addAction(self.action_edit_server)  
        self.server_list_contextmenu.addAction(self.action_copy_server)  
        self.server_list_contextmenu.addSeparator()
        self.server_list_contextmenu.addAction(self.action_delete_server)  
       
    @pyqtSlot(QPoint)
    def on_server_list_customContextMenuRequested(self, point):
        """
        Slot documentation goes here.
        """
        point.setY(point.y() + 25)
        if self.server_list.rowCount():
            point.setX(point.x() + 15)
        self.server_list_contextmenu.exec_(self.server_list.mapToGlobal(point))    
        
    @pyqtSlot()
    def on_start_upload_clicked(self):
        """
        Slot documentation goes here.
        """               
        if self.file_list.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加要上传的文件!")
            return
            
        if self.selected_servers.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加服务器到上传列表中!")
            return
            
        answer = QtGui.QMessageBox.warning(self, u"提示", u"上传将覆盖服务器设置路径下的同名文件,确定上传吗?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.No:
            return
            
        if self.upload_thread.isRunning() == False:
            files = [self.file_list.item(i, 0).text() for i in range(self.file_list.rowCount())]
            server_ids = [str(self.selected_servers.item(i, 0).text()) for i in range(self.selected_servers.rowCount())]
            servers = self.server_model.findByIds(",".join(server_ids))
            self.upload_thread.setData(files, servers)            
            self.upload_thread.start();
            
            self.stop_upload.setEnabled(True)
            self.start_upload.setEnabled(False)
              
    @pyqtSlot()
    def on_stop_upload_clicked(self):
        """
        Slot documentation goes here.
        """
                    
        if self.upload_thread.isRunning() == True:
            answer = QtGui.QMessageBox.warning(self, u"提示", u"确定终止当前上传任务?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
            if answer == QtGui.QMessageBox.No:
                return
        
            self.start_upload.setEnabled(True)
            self.stop_upload.setEnabled(False)
            self.upload_thread.terminate();
    
    def on_upload_thread_finished(self):
        self.start_upload.setEnabled(True)
        self.stop_upload.setEnabled(False)
    
    @pyqtSlot(str)
    def on_upload_thread_information(self, info):
        time = QTime(QTime.currentTime())
        self.textBrowser.append(u"[%s] %s" % (time.toString(), info))
    
    @pyqtSlot(str)
    def on_upload_solution_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        if self.upload_solution.currentIndex() >  0:
            self.setSelectedServersData(self.solution_model.load(p0))
            self.remove_solution.setEnabled(True)
            self.refresh_server_list.clicked.emit(True)
        else:
            self.setSelectedServersData([])
            self.remove_solution.setEnabled(False)
        
    @pyqtSlot(str)
    def on_platform_currentIndexChanged(self, p0):
        """
        Slot documentation goes here.
        """
        self.setServerListData(self.server_model.findByPlatform(p0))
    
    @pyqtSlot()
    def on_move2left_clicked(self):
        """
        Slot documentation goes here.
        """        
        rows = list(set([index.row() for index in self.server_list.selectedIndexes()]))
        for i in range(len(rows)):
            row = rows[i]-i
            index = self.selected_servers.rowCount()
            self.selected_servers.insertRow(index)
            self.selected_servers.setItem(index, 0, self.server_list.item(row, 0).clone())
            self.selected_servers.setItem(index, 1, self.server_list.item(row, 1).clone())
            self.selected_servers.setItem(index, 2, self.server_list.item(row, 2).clone())
            self.selected_servers.setItem(index, 3, self.server_list.item(row, 3).clone())
            self.server_list.removeRow(row)
    
    @pyqtSlot()
    def on_move2right_clicked(self):
        """
        Slot documentation goes here.
        """
        rows = list(set([index.row() for index in self.selected_servers.selectedIndexes()]))
        for i in range(len(rows)):
            row = rows[i]-i
            self.selected_servers.removeRow(row)
        self.refresh_server_list.clicked.emit(True)
             
    @pyqtSlot()
    def on_menu_about_triggered(self):
        """
        Slot documentation goes here.
        """
        QtGui.QMessageBox.about(self, u"FTP Client", u"Powered by ZLH!")
    
    @pyqtSlot()
    def on_menu_clear_log_triggered(self):
        self.textBrowser.clear()
        
    @pyqtSlot()
    def on_add_solution_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.selected_servers.rowCount() < 1:
            QtGui.QMessageBox.information(self, u"提示", u"请先添加服务器到上传列表中!")
            return
        
        text, ok = QtGui.QInputDialog.getText(self, u"添加", u"上传方案名称:")
        if ok == True:
            if self.solution_model.check(text) > 0:
                QtGui.QMessageBox.critical(self, u"提示", u"上传方案名称已存在!")
                return
            servers = [str(self.selected_servers.item(i, 0).text()) for i in range(self.selected_servers.rowCount())]
            if self.solution_model.add(text, ",".join(servers)):
                self.upload_solution.addItem(text)
                self.upload_solution.setCurrentIndex(self.upload_solution.count()-1)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"操作失败!")
    
    @pyqtSlot()
    def on_remove_solution_clicked(self):
        """
        Slot documentation goes here.
        """
        if self.upload_solution.currentIndex() ==  0:
            QtGui.QMessageBox.information(self, u"FTP Client", u"请选择要删除的上传方案!")
            return
            
        answer = QtGui.QMessageBox.question(self, u"提示", u"确定删除当前选中上传方案?", QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:
            name = self.upload_solution.currentText()
            if self.solution_model.delete(name):
                self.upload_solution.removeItem(self.upload_solution.currentIndex())
                self.selected_servers.setRowCount(0)
                self.refresh_server_list.clicked.emit(True)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")
        
    @pyqtSlot()
    def on_refresh_server_list_clicked(self):
        """
        Slot documentation goes here.
        """
        platform = self.platform.currentText()
        self.setServerListData(self.server_model.findByPlatform(platform))
         
    @pyqtSlot()
    def on_server_list_itemSelectionChanged(self):
        rows = self.server_list.selectedIndexes()
        if(rows):
            self.move2left.setEnabled(True)
            self.action_move2left.setEnabled(True)
            self.action_edit_server.setEnabled(True)
            self.action_copy_server.setEnabled(True)
            self.action_delete_server.setEnabled(True)
        else:
            self.move2left.setEnabled(False)
            self.action_move2left.setEnabled(False)
            self.action_edit_server.setEnabled(False)
            self.action_copy_server.setEnabled(False)
            self.action_delete_server.setEnabled(False)
   
    @pyqtSlot()
    def on_selected_servers_itemSelectionChanged(self):
        rows = self.selected_servers.selectedIndexes()
        if rows:
            self.move2right.setEnabled(True)
        else:
            self.move2right.setEnabled(False)
        
    def on_server_list_keyPressEvent(self, event):
        if event.key() == Qt.Key_Escape:
            self.server_list.clearSelection()
        elif event.key() == Qt.Key_Left:
            self.move2left.clicked.emit(True)
            
    def on_selected_servers_keyPressEvent(self, event):
        if event.key() == Qt.Key_Escape:
            self.selected_servers.clearSelection()
        elif event.key() == Qt.Key_Delete or event.key() == Qt.Key_Right:
            self.move2right.clicked.emit(True)        
            
    @pyqtSlot()
    def on_action_add_server_triggered(self):
        """
        Slot documentation goes here.
        """
        self.server = Server()
        self.server.saved.connect(self.on_server_saved)
        self.server.show()
    
    @pyqtSlot()
    def on_action_edit_server_triggered(self):
        """
        Slot documentation goes here.
        """
        server_id = self.server_list.item(self.server_list.currentRow(), 0).text()     
        
        self.server = Server()   
        self.server.saved.connect(self.on_server_saved)
        self.server.setData(self.server_model.item(server_id))     
        self.server.show() 
    
    @pyqtSlot()
    def on_action_copy_server_triggered(self):
        """
        Slot documentation goes here.
        """
        server_id = self.server_list.item(self.server_list.currentRow(), 0).text()  
        server_data = self.server_model.item(server_id)      
        server_data["id"] = None
        
        self.server = Server()   
        self.server.saved.connect(self.on_server_saved)
        self.server.setData(server_data)     
        self.server.show() 
    
    @pyqtSlot()
    def on_action_delete_server_triggered(self):
        """
        Slot documentation goes here.
        """
        index = self.server_list.currentRow()
        platform = self.server_list.item(index, 1).text()
        server = self.server_list.item(index, 2).text()
        answer = QtGui.QMessageBox.question(self, u"提示", u"确定删除 %s %s?" % (platform, server), QtGui.QMessageBox.Yes, QtGui.QMessageBox.No)
        if answer == QtGui.QMessageBox.Yes:            
            server_id = self.server_list.item(index, 0).text()      
            if self.server_model.delete(server_id):
                self.server_list.removeRow(index)
                QtGui.QMessageBox.information(self, u"提示", u"操作成功!")                
            else:
                QtGui.QMessageBox.critical(self, u"提示", u"操作失败!")
      
    def on_server_saved(self):
        self.refresh_server_list.clicked.emit(True)
        
    @pyqtSlot()
    def on_action_move2left_triggered(self):
        self.move2left.clicked.emit(True)
        
    def setServerListData(self, data):
        self.server_list.setRowCount(0)
        for row in data:
            if self.inSelectedServers(row["id"]) == False:
                index = self.server_list.rowCount()
                self.server_list.insertRow(index)
                self.server_list.setItem(index, 0, QTableWidgetItem(row["id"]))
                self.server_list.setItem(index, 1, QTableWidgetItem(row["platform"]))
                self.server_list.setItem(index, 2, QTableWidgetItem(row["server_name"]))
                self.server_list.setItem(index, 3, QTableWidgetItem(row["path"]))
            
    def setSelectedServersData(self, data):    
        self.selected_servers.setRowCount(0)
        for row in data:
            index = self.selected_servers.rowCount()
            self.selected_servers.insertRow(index)
            self.selected_servers.setItem(index, 0, QTableWidgetItem(row["id"]))
            self.selected_servers.setItem(index, 1, QTableWidgetItem(row["platform"]))
            self.selected_servers.setItem(index, 2, QTableWidgetItem(row["server_name"]))
            self.selected_servers.setItem(index, 3, QTableWidgetItem(row["path"]))
             
        self.refresh_server_list.clicked.emit(True)
                
    def inSelectedServers(self, server_id):
        for i in range(self.selected_servers.rowCount()):
            if self.selected_servers.item(i, 0).text() == server_id:
                return True
        return False