티스토리 뷰

IT/Flask

[Flask] SQLite3

Dragonz 2020. 9. 9. 15:29
반응형

 


Flask 는 SQLite 를 내장지원한다. 우선 임시 프로그램을 이용하여 데이터베이스와 connection 을 맺고 테이블을 하나 생성한다.

 

파일명 : sqlite_temp.py

import sqlite3

conn = sqlite3.connect('database.db')
print ('Opened database successfully')

conn.execute('CREATE TABLE STUDENTS (NAME TEXT, ADDR TEXT, CITY TEXT, PIN TEXT)')
print ('Table created successfully')
conn.close()

프로그램이 정상 종료 되었다면 'database.db' 라는 파일이 생성되어 있을 것이다.

 

파일명 : 16_sqlite3.py

from flask import Flask, render_template, request
import sqlite3 as sql
app = Flask(__name__)

@app.route('/')
def home() :
    return render_template('16_home.html')

@app.route('/enternew')
def new_student() :
    return render_template('16_student.html')

@app.route('/addrec', methods = ['POST','GET'])
def addrec() :
    if request.method == 'POST' :
        try :
            nm = request.form['nm']
            addr = request.form['add']
            city = request.form['city']
            pin = request.form['pin']

            with sql.connect('database.db') as con :
                cur = con.cursor()

                cur.execute('INSERT INTO STUDENTS (NAME, ADDR, CITY, PIN) VALUES (?,?,?,?)', (nm, addr, city, pin))
                con.commit()
                msg = 'Record successfully added'
        except :
            con.rollback()
            msg = 'Error in insert operation'
        finally :
            return render_template('16_result.html', msg = msg)
            con.close()
    return ''

@app.route('/list')
def list() :
    con = sql.connect('database.db')
    con.row_factory = sql.Row

    cur = con.cursor()
    cur.execute('SELECT * FROM STUDENTS')
    rows = cur.fetchall()
    return render_template('16_list.html', rows = rows)

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

 

파일명 : 16_home.html

<!doctype html>
<html>
    <body>
        <p><h2><a href = "http://localhost:5000/enternew">Add New Record</a></h2></p>
        <p><h2><a href = "http://localhost:5000/list">Show List</a></h2></p>
    </body>
</html>

 

파일명 : 16_student.html

<!doctype html>
<html>
    <body>
        <form action = "{{ url_for('addrec') }}" method = "POST">
            <h3>Student Information</h3>
            <p>Name<br/>
                <input type = "text" name = "nm" />
            </p>
            <p>Address<br/>
                <textarea name = "add"></textarea>
            </p>
            <p>City<br/>
                <input type = "text" name = "city" />
            </p>
            <p>PINCODE<br/>
                <input type = "text" name = "pin" />
            </p>
            <p>
                <input type = "submit" />
            </p>
        </form>
    </body>
</html>

 

파일명 : 16_result.html

<!doctype html>
<html>
    <body>
        <p>result of addition : {{ msg }}</p>
        <p><h2><a href="\">Go back to home page</a></h2></p>
    </body>
</html>

 

파일명 : 16_list.html

<!DOCTYPE html>
<html>
    <body>
        <table border = 1>
            <thead>
                <td>Name</td>
                <td>Address</td>
                <td>City</td>
                <td>Pincode</td>
            </thead>

            {% for row in rows %}
            <tr>
                <td>{{ row["name"] }}</td>
                <td>{{ row["addr"] }}</td>
                <td>{{ row["city"] }}</td>
                <td>{{ row["pin"] }}</td>
            </tr>
            {% endfor %}
        </table>

        <a href = "/">Go back to home page</a>
    </body>
</html>

 

Add Ner Record 를 선택하여 신규 데이터를 입력한다.

 

데이터 입력후 submit 클릭

 

정상 처리되었다는 메세지가 출력된다. 다시 메인화면으로 돌아간다.

 

Show List 를 선택하여 데이터가 정상적으로 INSERT 되었는지 확인한다

 

테스트를 위해 몇 개의 신규 데이터를 더 입력한 후 조회한다

 

END

반응형
저작자표시 비영리 변경금지

'IT > Flask' 카테고리의 다른 글

[Flask] SQLAlchemy  (0) 2020.09.09
[Flask] WTF (Flask-WTF)  (0) 2020.09.09
[Flask] 메일발송 (Mail)  (0) 2020.09.05
[Flask] 파일 업로딩 (File Uploading)  (0) 2020.08.26
[Flask] 메세지 플래싱 (Message Flashing)  (0) 2020.08.14
댓글
공지사항