Question

我开发了一个Python程序，我需要简化的建议。

这是我的代码的一部分：

import wx
import sys
import socket

def error_handler(c):
    if c == 'canceled':
        sys.exit('User canceled configuration.')
    elif c == 'empty':
        sys.exit('Empty value.')


def hostname():
    dlg = wx.TextEntryDialog(None,
                             'What is your default Hostname?',
                             'Hostname',
                             socket.gethostname())

    if dlg.ShowModal() == wx.ID_CANCEL:
        error_handler('canceled')
    else:
        if dlg.GetValue() == "":
            error_handler('empty')
        else:
            HOSTNAME = dlg.GetValue()
            return HOSTNAME


def random_hostname():
    dlg = wx.SingleChoiceDialog(None,
                                'Do you want to randomize your Hostname',
                                'Randomize',
                                ['Yes', 'No', 'Disable'],
                                wx.CHOICEDLG_STYLE)

    if dlg.ShowModal() == wx.ID_CANCEL:
        error_handler('canceled')
    else:
        RANDOM_HOSTNAME = dlg.GetStringSelection()
        return RANDOM_HOSTNAME


def nameserver():
    dlg = wx.TextEntryDialog(None,
                             'Nameserver IP\n',
                             'Nameserver',
                             '127.0.0.1')

    if dlg.ShowModal() == wx.ID_CANCEL:
        error_handler('canceled')
    else:
        if dlg.GetValue() == "":
            error_handler('empty')
        else:
            NAMESERVER = dlg.GetValue()
            return NAMESERVER


def main():
    app = wx.App()
    print 'HOSTNAME =', hostname()
    print 'RANDOM_HOSTNAME =', random_hostname()
    print 'NAMESERVER =', nameserver()
    app.MainLoop()


if __name__ == '__main__':
    main()

在这段代码中，我为Hostname，Random hostname和nameserver创建了函数，但是在所有3个函数中我必须重复几乎相同的代码：

if dlg.ShowModal() == wx.ID_CANCEL:
    error_handler('canceled')
else:
    if dlg.GetValue() == "":
        error_handler('empty')
    else:
        HOSTNAME = dlg.GetValue()
        return HOSTNAME

但我想制作超过20个用于检查某些值的函数。是否（并且我知道有）一些更好的技巧来缩短每个功能？

我想要这样的事情：

import wx
import sys
import socket

def error_handler(c):
    if c == 'canceled':
        sys.exit('User canceled configuration.')
    elif c == 'empty':
        sys.exit('Empty value.')
    else
        return dialog value


def hostname():
    dlg = wx.TextEntryDialog(None,
                             'What is your default Hostname?',
                             'Hostname',
                             socket.gethostname())

    error_handler(dlg)


def random_hostname():
    dlg = wx.SingleChoiceDialog(None,
                                'Do you want to randomize your Hostname',
                                'Randomize',
                                ['Yes', 'No', 'Disable'],
                                wx.CHOICEDLG_STYLE)

    error_handler(dlg)


def nameserver():
    dlg = wx.TextEntryDialog(None,
                             'Nameserver IP\n',
                             'Nameserver',
                             '127.0.0.1')

    error_handler(dlg)


def main():
    app = wx.App()
    print 'HOSTNAME =', hostname()
    print 'RANDOM_HOSTNAME =', random_hostname()
    print 'NAMESERVER =', nameserver()
    app.MainLoop()


if __name__ == '__main__':
    main()

我要感谢所有帮助我的人。

Answer 1

看起来你已经知道了创建另一个函数error_handler来完成重复工作的答案！您可以将错误处理程序包装在try/except块中，该块看起来像这样：

def hostname():
    try:
        dlg = wx.TextEntryDialog(None,
                     'What is your default Hostname?',
                     'Hostname',
                      socket.gethostname())
    except:
        error_handler(dlg)

Answer 2

def funct(dlg, wx, funcCheck):
    if dlg.ShowModal() == wx.ID_CANCEL:
        error_handler('canceled')
    else:
        if dlg.GetValue() == "":
            error_handler('empty')
        else:
            value = funcCheck()
            return value

然后你可以这样称呼它

funct(dlg, wx, dlg.GetValue)

根据您的需要继续为该功能添加条件

Python - 如何缩短代码

2 个答案: