Extending Construct

Adapters

Adapters are the standard way to extend and customize the library. Adapters operate at the object level (unlike constructs, which operate at the stream level), and are thus easy to write and more flexible. For more info see the adapter tutorial.

In order to write custom adapters, implement _decode and _encode:

class MyAdapter(Adapter):
    def _decode(self, obj, context, path):
        # called at parsing time to return a modified version of obj
        pass

    def _encode(self, obj, context, path):
        # called at building time to return a modified version of obj
        pass

Constructs

Note

This is a last line of defense. If you are not able to achieve whatever that is you are trying to do with Construct existing classes, you can always write your own classes. In particular, you can copy-paste existing classes into your source code and modify them slighly. This is the best way to go.

Generally speaking, you should not write constructs by yourself:

  • It’s a craft that requires skills and understanding of the internals of the library (which change over time).

  • Adapters should really be all you need and are much simpler to implement.

  • To make things faster, try using compilation feature, or pypy. The python-level classes are as fast as it gets, assuming generality.

The only reason you might want to write a custom class is to achieve something that’s not currently possible. This might be a construct that computes/corrects the checksum of data, altough that already exists. Or a compression, or hashing. These also exist. But surely there is something that was not invented yet. If you need a semantics modification to an existing class, you can post a feature request, or copy the code of existing class into your project and modify it.

There are at least two kinds of constructs: raw construct and subconstructs.

Raw constructs

Deriving directly from class Construct, raw constructs can do as they wish by implementing _parse, _build, and _sizeof:

class MyConstruct(Construct):
    def _parse(self, stream, context, path):
        # read from the stream
        # return object
        pass

    def _build(self, obj, stream, context, path):
        # write obj to the stream
        # return same value (obj) or a modified value
        # that will replace the context dictionary entry
        pass

    def _sizeof(self, context, path):
        # return computed size (when fixed size or depends on context)
        # or raise SizeofError (when variable size or unknown)
        pass

Variable size fields typically raise SizeofError, for example VarInt and CString.

Subconstructs

Deriving from class Subconstruct, these wrap an inner construct, inheriting it’s properties (name and flags). In their _parse and _build and _sizeof methods, they will call self.subcon._parse and self.subcon._build and self.subcon._sizeof respectively.

class MySubconstruct(Subconstruct):
    def __init__(self, subcon):
        self.name = subcon.name
        self.subcon = subcon
        self.flagbuildnone = subcon.flagbuildnone

    def _parse(self, stream, context, path):
        obj = self.subcon._parse(stream, context, path)
        # do something with obj
        return obj

    def _build(self, obj, stream, context, path):
        # do something with obj
        return self.subcon._build(obj, stream, context, path)
        # return same value (obj) or a modified value
        # that will replace the context dictionary entry

    def _sizeof(self, context, path):
        # if not overriden, defers to subcon size
        return self.subcon._sizeof(context, path)