Add support for data types added in PostgreSQL 10.

Author: elprans

.flake8

@@ -1,3 +1,3 @@
 [flake8]
 ignore = E402,E731
-exclude = .git,__pycache__,build,dist,.eggs,.github
+exclude = .git,__pycache__,build,dist,.eggs,.github,.local

asyncpg/protocol/codecs/misc.pyx

@@ -68,6 +68,7 @@ cdef init_pseudo_codecs():
         ANYOID, TRIGGEROID, EVENT_TRIGGEROID, LANGUAGE_HANDLEROID,
         FDW_HANDLEROID, TSM_HANDLEROID, INTERNALOID, OPAQUEOID,
         ANYELEMENTOID, ANYNONARRAYOID, PG_DDL_COMMANDOID,
+        INDEX_AM_HANDLEROID,
     ]
 
     register_core_codec(ANYENUMOID,
@@ -104,5 +105,16 @@ cdef init_pseudo_codecs():
                         <decode_func>&text_decode,
                         PG_FORMAT_TEXT)
 
+    # pg_dependencies and pg_ndistinct are special types
+    # used in pg_statistic_ext columns.
+    register_core_codec(PG_DEPENDENCIESOID,
+                        <encode_func>&text_encode,
+                        <decode_func>&text_decode,
+                        PG_FORMAT_TEXT)
+
+    register_core_codec(PG_NDISTINCTOID,
+                        <encode_func>&text_encode,
+                        <decode_func>&text_decode,
+                        PG_FORMAT_TEXT)
 
 init_pseudo_codecs()

asyncpg/protocol/codecs/network.pyx

@@ -135,5 +135,10 @@ cdef init_network_codecs():
                         <decode_func>&text_decode,
                         PG_FORMAT_TEXT)
 
+    register_core_codec(MACADDR8OID,
+                        <encode_func>&text_encode,
+                        <decode_func>&text_decode,
+                        PG_FORMAT_TEXT)
+
 
 init_network_codecs()

asyncpg/protocol/pgtypes.pxi

@@ -30,6 +30,7 @@ DEF JSONOID = 114
 DEF XMLOID = 142
 DEF PG_NODE_TREEOID = 194
 DEF SMGROID = 210
+DEF INDEX_AM_HANDLEROID = 325
 DEF POINTOID = 600
 DEF LSEGOID = 601
 DEF PATHOID = 602
@@ -44,6 +45,7 @@ DEF RELTIMEOID = 703
 DEF TINTERVALOID = 704
 DEF UNKNOWNOID = 705
 DEF CIRCLEOID = 718
+DEF MACADDR8OID = 774
 DEF MONEYOID = 790
 DEF MACADDROID = 829
 DEF INETOID = 869
@@ -83,6 +85,8 @@ DEF TXID_SNAPSHOTOID = 2970
 DEF FDW_HANDLEROID = 3115
 DEF PG_LSNOID = 3220
 DEF TSM_HANDLEROID = 3310
+DEF PG_NDISTINCTOID = 3361
+DEF PG_DEPENDENCIESOID = 3402
 DEF ANYENUMOID = 3500
 DEF TSVECTOROID = 3614
 DEF TSQUERYOID = 3615
@@ -122,6 +126,7 @@ TYPEMAP = {
     FLOAT4OID: 'float4',
     FLOAT8OID: 'float8',
     GTSVECTOROID: 'gtsvector',
+    INDEX_AM_HANDLEROID: 'index_am_handler',
     INETOID: 'inet',
     INT2OID: 'int2',
     INT4OID: 'int4',
@@ -133,6 +138,7 @@ TYPEMAP = {
     LANGUAGE_HANDLEROID: 'language_handler',
     LINEOID: 'line',
     LSEGOID: 'lseg',
+    MACADDR8OID: 'macaddr8',
     MACADDROID: 'macaddr',
     MONEYOID: 'money',
     NAMEOID: 'name',
@@ -141,7 +147,9 @@ TYPEMAP = {
     OPAQUEOID: 'opaque',
     PATHOID: 'path',
     PG_DDL_COMMANDOID: 'pg_ddl_command',
+    PG_DEPENDENCIESOID: 'pg_dependencies',
     PG_LSNOID: 'pg_lsn',
+    PG_NDISTINCTOID: 'pg_ndistinct',
     PG_NODE_TREEOID: 'pg_node_tree',
     POINTOID: 'point',
     POLYGONOID: 'polygon',

tools/generate_type_map.py

@@ -29,6 +29,12 @@ async def runner(args):
                                  user=args.pguser)
 
     buf = (
+        '# Copyright (C) 2016-present the asyncpg authors and contributors\n'
+        '# <see AUTHORS file>\n'
+        '#\n'
+        '# This module is part of asyncpg and is released under\n'
+        '# the Apache 2.0 License: http://www.apache.org/licenses/LICENSE-2.0'
+        '\n\n\n'
         '# GENERATED FROM pg_catalog.pg_type\n' +
         '# DO NOT MODIFY, use tools/generate_type_map.py to update\n\n' +
         'DEF INVALIDOID = {}\n'.format(_INVALIDOID) +
@@ -70,7 +76,7 @@ async def runner(args):
 
     buf += '\n'.join(defs)
 
-    buf += '\n\nARRAY_TYPES = ({},)'.format(', '.join(array_types))
+    buf += '\n\ncdef ARRAY_TYPES = ({},)'.format(', '.join(array_types))
 
     f_typemap = ('{}: {!r}'.format(dn, n) for dn, n in sorted(typemap.items()))
     buf += '\n\nTYPEMAP = {{\n    {}}}'.format(',\n    '.join(f_typemap))