CIText Array about sqlalchemy-citext

[mahmoudimus]

@lukas-gitl this is awesome! I can make it into a proper pull request and see how to integrate it. thank you so much!

[rmihael]

There is a better way to make CITEXT work with arrays. If CITEXT[] is registered as an array type in psycopg2, then everything works just fine without any monkey patching. That can be accomplished with that snippet:

def register_citext_array(engine):
    results = engine.execute(text("SELECT typarray FROM pg_type WHERE typname = 'citext'"))
    oids = tuple(row[0] for row in results)
    array_type = psycopg2.extensions.new_array_type(oids, b'citext[]', psycopg2.STRING)
    psycopg2.extensions.register_type(array_type, None)

It would be great to include something about it to README.

[lukas-gitl]

Would be great to get this into this project somehow.

[mahmoudimus]

@lukas-gitl or..you can take a stab at a PR? :)

[lukas-gitl]

Edited initial post with small fix for stripping quotes

[lukas-gitl]

Updated it again to use csv. This seems to solve all issues here.

[mahmoudimus]

@lukas-gitl thanks for this! I'll get to it this weekend.

[rmihael]

Hi! Any hopes for releasing this fix?

[HansBouwmeester]

@rmihael Thanks!!! Works great except that for me psycopg2 told me it expected b'citext[]' as a regular string (not bytes): 'citext[]'.

[mzpqnxow]

It looks like this works now

On a related note- I stumbled upon this issue while trying to make sqlacodegen produce CIText and CIText[] columns. I ended up using sqlalchemy-citext to do it. It works well and was a very, very small patch. Please see agronholm/sqlacodegen#125 if you use sqlacodegen and want CIText and CIText[] support

[mzpqnxow]

I wonder if this can be closed now as it seems to be working

[mahmoudimus]

I will update the README! Thanks

[mzpqnxow]

@mahmoudimus I sent #19 which is the register function that @rmihael suggested. I tested this on CPython 3.7 and it works well, fixes the problem of getting back characters instead of strings. It needs to be explicitly called, so in theory it shouldn't break anyone who isn't explicitly using it