[python] 别人的模块有两套代码针对 2 和 3 的，我怎样合并？

比如

有两个文件夹

alib/python2/
alib/python2/__init__py
alib/python2/mainfile.py

alib/python3/
alib/python3/__init__py
alib/python3/mainfile.py

我怎样在我的代码中引用这个库呢？
我想到的一个办法是，在 alib 中新建一个 __init__.py

import sys
PY2 = sys.version_info[0] == 2
if not PY2:
from python2 import *
else:
from python3 import *


然后在外部的 myapp.py

from alib import *

这样子是正确的吗？