flask postgresql과 연동하기

flask를 데이터 베이스중 이번 프로젝트에서 선택한 postgresql과 연동해 보도록 하겠다.

 

 먼저 pip install psycopg2를 쳐서 설치해준다.

psycopg2는 python에서 postgresql을 사용하기 위해 사용해야하는 라이브러리 이다.

 

설치가 되었다면

import psycopg2

from flask import Flask, jsonify
import psycopg2.extras



app = Flask(__name__)

DB_HOST = "db_url 또는 localhost ip"
DB_NAME = "사용중인 데이터 베이스의 이름"
DB_USER = "user이름 기본값은 postgres"
DB_PASS = "비밀번호"

conn = psycopg2.connect(dbname=DB_NAME,
                      user=DB_USER,
                      password=DB_PASS,
                      host=DB_HOST)

@app.route('/')
def hello_world():
    cur = conn.cursor(cursor_factory=psycopg2.extras.DictCursor)
    s = "SELECT * FROM users"
    cur.execute(s)
    list_users = cur.fetchall()
    return jsonify(list_users)
    
    # 문자열이 아닌 리스트를 화면에 띄우려면 jsonify 라이브러리를 이용해야한다.
    #jsonify는 많은 데이터를 json형태로 보낼수 있다.

이런식으로 코드를 작성하여  postgresql과 연동하고 

 

데이터를 가져와서 화면에 띄워준다.

 

데이터 베이스에 저장된 데이터가 리스트 형태로 화면에 출력되면 정상적으로 작동 중인 것이다.