Wrap pm.commandPort with defaults set for the MayaSublime plugin

commandPortWrapper.py

from __future__ import print_function

import pymel.core as pm


def openPort(port=":7002", **kwargs):
    print("Opening command port", port)
    commandPortWrapper(port, False, **kwargs)

def closePort(port=":7002", **kwargs):
    print("Closing command port", port)
    commandPortWrapper(port, True, **kwargs)

def commandPortWrapper(port=":7002", close=False, **kwargs):
    # the default port is set for the justinfx/MayaSublime plugin
    
    if not port.startswith(':'):
        port = ''.join([':', port])

    if "sourceType" not in kwargs:
        # assume python, instead of the maya default, which is to assume mel
        kwargs["sourceType"] = "python"

    if not close:
        if "noreturn" not in kwargs:
            kwargs["noreturn"] = True

        if "echoOutput" not in kwargs:
            kwargs["echoOutput"] = True
        kwargs["close"] = False
    else:
        kwargs["close"] = True

    try:
        pm.commandPort(name=port, **kwargs)
    except RuntimeError as e:
        print(e)