API서버 - Python MySQL Connector를 이용해 select하기(GET 메소드)

mysql_connection.py 파일

# MySQL에 접속하는 함수
import mysql.connector

from config import Config 

def get_connection():

    connection = mysql.connector.connect(
        host= Config.HOST,
        database= Config.DATABASE,
        user= Config.DB_USER,
        password= Config.DB_PASSWORD
    )
    return connection

 

 

API 리소스를 관리하는 recipe.py 파일

class RecipeListResource(Resource) :

    # GET 메소드를 처리하는 함수
    def get(self):

        # 1. 클라이언트로부터 데이터를 받아온다.
        #    get은 받아오는 데이터 없다!!!

        # 2. DB에 저장된 데이터를 가져온다.
        try :
            connection = get_connection()

            query = '''select *
                    from recipe;'''

            ### 중요!!! 
            # select 문은 커서 가져올때, dictionary=True로 해준다.
            cursor = connection.cursor(dictionary=True)
            cursor.execute(query)
            result_list = cursor.fetchall()
            # print(result_list)

            ### 중요!!!
            # DB에서 가져온 timestamp는 파이썬에서 datetime으로 자동 변환된다.
            # 문제는, 우리는 json으로 클라이언트한테 데이터를 보내줘야 하는데
            # datetime은 json으로 보낼 수 없다.
            # 따라서, 시간을 문자열로 변환해서 보내준다.
            i = 0
            for row in result_list:
                result_list[i]['created_at'] = row['created_at'].isoformat()
                result_list[i]['updated_at'] = row['updated_at'].isoformat()
                i += 1

            cursor.close()
            connection.close()

        except mysql.connector.Error as e :
            print(e)
            cursor.close()
            connection.close()

            return {"result" : "fail", "error" : str(e)}, 500
        
        return {"result" : "success",
                "items" : result_list,
                "count" : len(result_list)}, 200

 

 

메인파일 app.py

from flask import Flask
from flask_restful import Api
from config import Config

from resources.recipe import RecipeListResource, RecipeResource, RecipePublishResource
from resources.user import UserRegisterResource



app = Flask(__name__)
# 환경변수 셋팅
app.config.from_object(Config)

api = Api(app)

# 경로와 리소스를 연결한다.
api.add_resource(RecipeListResource, '/recipes')
api.add_resource(RecipeResource, '/recipes/<int:recipe_id>')
api.add_resource(RecipePublishResource, '/recipes/<int:recipe_id>/publish')

api.add_resource(UserRegisterResource, '/user/register')


if __name__ == '__main__':
    app.run()

 

 

포스트맨에서 테스트

정상적으로 데이터를 가져온다.