postgresのDBのcursor処理

Question

pythonでのpostgresのDB処理で、最初にcursorを作成して、
必要な時に別クラス(別ファイル)からcursorを使用したいと考えています。
現状は以下のようにDbaccessクラスのglobal変数に設定しています。
grobal変数はあまり使用したくないのですが、
使用しないでできる方法教えていただけませんでしょうか

main.py
class test_g():
test = Dbaccess():
test.db_init()

test2 = Select1()
test2.db_select()

test3 = Select2()
test3.db_select()

---

Dbaccess.py

import psycopg2
class Dbaccess():
cur = ""

def db_init():
global cur
conn = psycopg2.connect("dbname=test user=postgres")
cur = conn.cursor()

def get_cur():
global cur
return cur

---

select1.py

import psycopg2
import Dbaccess

class Select1():
def db_select():
cur_sel = Dbaccess.get_cur()
sql = "select * from table_a"
cur_sel.execute(sql)

---

Select2.py

import psycopg2
import Dbaccess

class Select2():
def db_select():
cur_sel = Dbaccess.get_cur()
sql = "select * from table_b"
cur_sel.execute(sql)

Answer 1

関数がなにか値を返せることは理解していますね。return文を使っていますから。
関数は引数を受け取れることも理解していますよね。
ある関数が返してくれたcursorを、他の関数の引数に渡してみてはいかがでしょうか。

def get_cur():
  ...
  return cur

def db_select(cur):
  sql = "select * from table_b"
  cur_sel.execute(sql)
  ...

xx = get_cur()
db_select(xx)

Comments

[yamadataro3]

遅くなりました。ありがとうございました。