вставить enum тип данных через pandas.to_sql

Question

У меня есть pandas dataframe следующим образом:

In [93]: df = pd.DataFrame([[1, 'a'], [2, 'b'], [3, 'a']], columns=['val', 'types'])

In [94]: df
Out[94]:
   val types
0    1    a
1    2    b
2    3    a

In [95]: df.dtypes
Out[95]:
val       int64
types    object
dtype: object

# convert the 'types' column to category
In [96]: df.types = pd.Categorical(df.types)

Если у меня уже есть таблица postgres со столбцом types как ENUM , как мне вставить данные в мою БД черезsqlachemy типы с pandas to_sql method?

Я попробовал это, но не смог заставить его работать:

In [101]: class myEnum(enum.Enum):
     ...:     a = 1
     ...:     b = 2
     ...:

In [102]: from sqlalchemy.types import *

In [103]: sql_dtypes = {'val' : SmallInteger(), 'types': Enum(myEnum)}

In [104]: df.to_sql('tablename', conn, if_exists='replace', index=False,
                    dtype=sql_dtypes)

работает, что дает мне следующую ошибку:

sqlalchemy.exc.ProgrammingError: (psycopg2.ProgrammingError) type "myEnum" already 
exists

[SQL: "CREATE TYPE myEnum AS ENUM ('a', 'b')"] (Background on this 
error at: http://sqlalche.me/e/f405)

Указание с помощью опции schema из pd.io.json.build_table_schema тоже не помогло.

Answer 1

Обходной путь, который я нашел на данный момент из здесь , состоит в том, чтобы сначала записать тип данных как varchar, а затем преобразовать его в enum.

In [104]: df.to_sql('tablename', conn, if_exists='replace', index=False,
                    dtype=sql_dtypes)

In [105]: conn.execute("""ALTER TABLE tablename 
                          ALTER COLUMN types TYPE myEnum using types::myEnum;""")

Но я надеюсь, что естьлучшее решение.