python:根据操作系统使用不同的功能

时间:2017-04-21 10:49:48

标签: python cross-platform

我想编写一个可在Linux和Solaris上执行的脚本。大多数逻辑在两个操作系统上都是相同的,因此我只编写一个脚本。但由于某些部署的结构会有所不同(文件位置,文件格式,命令语法),因此两个平台上的一些功能会有所不同。

这可以像

一样处理
if 'linux' in sys.platform:
    result = do_stuff_linux()
if 'sun' in sys.platform:
    result = do_stuff_solaris()
more_stuf(result)
...

然而,在整个代码中使用这些ifs似乎很麻烦且不优雅。我也可以在一些dict中注册函数,然后通过dict调用函数。可能好一点。

关于如何做到这一点的任何更好的想法?

2 个答案:

答案 0 :(得分:1)

你可以这样做:

if 'linux' in sys.platform:
    def do_stuff():
        result = # do linux stuff
        more_stuff(result)
elif 'sun' in sys.platform:
    def do_stuff():
        result = # do solaris stuff
        more_stuff(result)

然后只需致电do_stuff()

答案 1 :(得分:1)

解决方案1:

您为复制所需的每个功能创建单独的文件,并导入正确的文件:

import sys
if 'linux' in sys.platform:
    from .linux import prepare, cook
elif 'sun' in sys.platform:
    from .sun import prepare, cook
else:
    raise RuntimeError("Unsupported operating system: {}".format(sys.platform))

dinner = prepare('pork')
drink_wine()
result = cook(dinner)

解决方案1.5:

如果您需要将所有内容保存在一个文件中,或者只是不喜欢条件导入,您可以随时为这些函数创建别名:

import sys

def prepare_linux(ingredient):
    ...

def prepare_sun(ingredient):
    ...

def cook_linux(meal):
    ...

def cook_sun(meal):
    ...

if 'linux' in sys.platform:
    prepare = prepare_linux
    cook = cook_linux
elif 'sun' in sys.platform:
    prepare = prepare_sun
    cook = cook_sun
else:
    raise RuntimeError("Unsupported operating system: {}".format(sys.platform))

dinner = prepare('chicken')
drink_wine()
result = cook(dinner)