проверка и исправление значений, добавленных к узлу с помощью lxml.objectify в python - PullRequest
Новая
       27

проверка и исправление значений, добавленных к узлу с помощью lxml.objectify в python

0 голосов
/ 22 июля 2011

Я пытаюсь изменить PyKML, который использует lxml.objectify. При добавлении узла дорожки это превращается в строку с поведением по умолчанию str(). Я хотел бы отлавливать добавления списков или кортежей и преобразовывать их в строки, разделенные на правильные места, а не '(xx.xxxx, yy.yyyy)' 

from pykml.factory import GX_ElementMaker as GX

track = GX.Track(id='track_%d' % group_num )
for pt in group:
    when = datetime.datetime.utcfromtimestamp(pt['ts'])
    track.append( KML.when( when ) )  # WHEN?
for pt in group:
    track.append( GX.coord( (pt['x'],pt['y'])) )  # <-- trouble here

Спасибо, -kurt

1 Ответ

0 голосов
/ 03 августа 2011

Создайте чистый класс ElementMaker с информацией о пространстве имен. Затем создайте подкласс с именем узла в качестве метода. В этом методе обработайте все нечетные случаи. Затем создайте строку, которая будет помещена в это место имен узлов, и верните экземпляр чистого класса ElementMaker с этим именем узла.

http://code.google.com/r/schwehr-pykml/source/browse/src/pykml/factory.py?spec=svn05a10cef3fd3c430389e8aca1313a20da932e565&r=05a10cef3fd3c430389e8aca1313a20da932e565

def indexable_levels(args):
    #print 'args:',args
    levels = 0
    while True:
        if isinstance(args,str): break
        try:
            args = args[0]
            levels += 1
        except:
            break
    #print '   levels ->',levels
    return levels

# Create a factory object for the KML Google Extension namespace
_GX_ElementMakerSimple = objectify.ElementMaker(
    annotate=False,
    namespace=nsmap['gx'],
    nsmap={'gx': nsmap['gx']},
)

class _GX_ElementMaker (objectify.ElementMaker):
    'KML ElementMaker with overloads for custom text payloads like coordinates'
    def coord(self, *args):
        #print 'start coord: "%s"' % (str(args)), type(args), len(args)
        levels = indexable_levels(args)
        if levels == 1 and len(args) == 1:
            # This case is really redundant with the next
            assert isinstance(args[0],str)
            return _GX_ElementMakerSimple.coord(args[0])
        if levels == 1:
            return _GX_ElementMakerSimple.coord(' '.join([str(item) for item in args]))
        if levels == 2:
            # ((-121.583851, 37.386052),)
            assert(len(args)==1)
            return _GX_ElementMakerSimple.coord(' '.join([str(item) for item in args[0] ]))
        assert(False)

# Create a factory object for the KML Google Extension namespace
GX_ElementMaker = _GX_ElementMaker(
    annotate=False,
    namespace=nsmap['gx'],
    nsmap={'gx': nsmap['gx']},
)
...