Python 包和方法未导入

Question

我用几种方法构建了一个简单的类，让我在使用 Python 将数据加载到 Postgres 时更轻松一些。我还尝试打包它，以便我可以 pip 安装它(只是为了实验，以前从未这样做过(。

import psycopg2
from sqlalchemy import create_engine
import io

class py_psql:
engine = None
def engine(self, username, password, hostname, port, database):
connection = 'postgresql+psycopg2://{}:{}@{}:{}/{}'.format(ntid.lower(), pw, hostname, port, database)
self.engine = create_engine(connection)

def query(self, query):
pg_eng = self.engine
return pd.read_sql_query(query, pg_eng)

def write(self, write_name, df, if_exists='replace', index=False):
mem_size = df.memory_usage().sum()/1024**2
pg_eng = self.engine
def write_data():
df.head(0).to_sql(write_name, pg_eng, if_exists=if_exists,index=index)
conn = pg_eng.raw_connection()
cur = conn.cursor()
output = io.StringIO()
df.to_csv(output, sep='t', header=False, index=False)
output.seek(0)
contents = output.getvalue()
cur.copy_from(output, write_name, null="")
conn.commit()
if mem_size > 100:
validate_size = input('DataFrame is {}mb, proceed anyway? (y/n): '.format(mem_size))
if validate_size == 'y':        
write_data()
else:
print("Canceling write to database")
else:
write_data()

我的包目录如下所示：

py_psql
py_psql.py
__init__.py
setup.py

我的init.py 是空的，因为我在其他地方读到我能够做到这一点。我不是这里的专家...

我能够 pip 安装该包并导入它，如果我要将此类粘贴到 python shell 中，我将能够执行类似

test = py_psql()
test.engine(ntid, pw, hostname, port, database)

并让它创建SQLaLchemy引擎。但是，当我在 pip 安装后导入它时，我什至无法初始化py_psql对象：

>>> test = py_psql()
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: 'module' object is not callable
>>> py_psql.engine(ntid, pw, hostname, port, database)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AttributeError: module 'py_psql' has no attribute 'engine'

我确定我在这里搞砸了一些明显的东西，但我发现在研究这个时打包的过程相当混乱。我做错了什么？

Answer 1

你确定在pip安装后正确导入了你的软件包吗？

例如：

from py_psql.py_psql import py_psql
test = py_psql()
test.engine(ntid, pw, hostname, port, database)