mirror of
https://github.com/evennia/evennia.git
synced 2026-03-19 14:26:30 +01:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""
|
|
some utilities
|
|
"""
|
|
import os
|
|
import sys
|
|
import __main__
|
|
|
|
from twisted.python.filepath import FilePath
|
|
from twisted.python.reflect import namedAny
|
|
# from twisted.python.modules import theSystemPath
|
|
|
|
def findPackagePath(modulePath):
|
|
"""
|
|
Try to find the sys.path entry from a modulePath object, simultaneously
|
|
computing the module name of the targetted file.
|
|
"""
|
|
p = modulePath
|
|
l = [p.basename().split(".")[0]]
|
|
while p.parent() != p:
|
|
for extension in ['py', 'pyc', 'pyo', 'pyd', 'dll']:
|
|
sib = p.sibling("__init__."+extension)
|
|
if sib.exists():
|
|
p = p.parent()
|
|
l.insert(0, p.basename())
|
|
break
|
|
else:
|
|
return p.parent(), '.'.join(l)
|
|
|
|
|
|
def mainpoint(function):
|
|
"""
|
|
Decorator which declares a function to be an object's mainpoint.
|
|
"""
|
|
if function.__module__ == '__main__':
|
|
# OK time to run a function
|
|
p = FilePath(__main__.__file__)
|
|
p, mn = findPackagePath(p)
|
|
pname = p.path
|
|
if pname not in map(os.path.abspath, sys.path):
|
|
sys.path.insert(0, pname)
|
|
# Maybe remove the module's path?
|
|
exitcode = namedAny(mn+'.'+function.__name__)(sys.argv)
|
|
if exitcode is None:
|
|
exitcode = 0
|
|
sys.exit(exitcode)
|
|
return function
|