mirror of
https://github.com/qgis/QGIS.git
synced 2025-02-27 00:33:48 -05:00
45c9ea2355
* drop libpasteurize.fixes.fix_division * replace lib2to3.fixes.fix_import with libfuturize.fixes.fix_absolute_import (with PyQt exclude plus AlgorithmsTestBase) * drop lib2to3.fixes.fix_print (replaced with libfuturize.fixes.fix_print_with_import)
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
"""Fixer for import of generated UIs
|
|
"""
|
|
|
|
# Local imports
|
|
from lib2to3 import fixer_base
|
|
from lib2to3.fixes.fix_import import FixImport
|
|
from lib2to3.fixer_util import FromImport, Node, Leaf, syms
|
|
|
|
|
|
class FixUiimport(fixer_base.BaseFix):
|
|
BM_compatible = True
|
|
|
|
PATTERN = """
|
|
import_from< 'from' imp=any 'import' ['('] any [')'] >
|
|
|
|
|
import_name< 'import' imp=any >
|
|
"""
|
|
|
|
def transform(self, node, results):
|
|
imp = results.get('imp')
|
|
|
|
if node.type == syms.import_from:
|
|
# Some imps are top-level (eg: 'import ham')
|
|
# some are first level (eg: 'import ham.eggs')
|
|
# some are third level (eg: 'import ham.eggs as spam')
|
|
# Hence, the loop
|
|
while not hasattr(imp, 'value'):
|
|
imp = imp.children[0]
|
|
if imp.value.startswith("ui_"):
|
|
imp.value = u"." + imp.value
|
|
imp.changed()
|
|
else:
|
|
first = imp
|
|
if isinstance(imp, Node):
|
|
first = next(imp.leaves())
|
|
|
|
if not isinstance(first, Leaf):
|
|
return
|
|
|
|
if not first.value.startswith("ui_"):
|
|
return
|
|
|
|
new = FromImport(".", [imp])
|
|
new.prefix = node.prefix
|
|
return new
|