Locator

# Locator at pivot
# cwisbg.com
from pymel.core import *
currentSceneScale = currentUnit(q=1, l=1)
tempSceneScale = currentUnit(l = "cm")
def getPiv(s, type):
    if type == "transform":
        papCheck = listRelatives(s,p=1)
        if papCheck:
            try:
                parent(s,w=1)
                rot = s.getRotation()
                parent(s, papCheck)
            except:
                pass
        else:
            rot = s.getRotation()
    else:
        rot = [0,0,0]
    select(s)
    setToolTo('Move')
    piv = manipMoveContext('Move', q=True, p=True,m=2)
    return piv, rot
sl = selected()
type = nodeType(sl[0])
print type
if type == "mesh":
    t , r = getPiv(sl, type)
    l = spaceLocator(p = (0,0,0), n = "{0}_saveLocation".format(sl[0]))
    move(l, t, ws=1)
    rotate(l, r)
else:
    for s in sl:
        type = nodeType(s)
        t , r = getPiv(s, type)
        l = spaceLocator(p = (0,0,0), n = "{0}_saveLocation".format(s))
        move(l, t, ws=1)
        rotate(l, r)
currentUnit(l = currentSceneScale)
print "done"