python: use different function depending on os

Question

I want to write a script that will execute on Linux and Solaris. Most of the logic will be identical on both OS, therefore I write just one script. But because some deployed str

Answer 1

You can do it like this:

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)

And then simply call do_stuff().

Answer 2

Solution 1:

You create separate files for each of the functions you need to duplicate and import the right one:

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)

Solution 1.5:

If you need to keep everything in a single file, or just don't like the conditional import, you can always just create aliases for the functions like so:

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)