본문 바로가기
코딩과 알고리즘

구글코랩에서 sqlite 파일 생성! ( 파이썬 )

구글 코랩을 이것 저것 다루어보던중 sqlite db 파일을 다루는 기술이 있어 공유 차원에서 글을 적습니다 :)


뭐 나중에 크레이가 참조할 목적이기도 하지요 ㅎㅎ
소형 Database 인 sqlite 를 간이형식으로 아주 간단하게 다룰 수 있어 유용하네요.
설명은 주석으로 대체합니다.
아무 것도 설치하지 않아도 잘 동작하더라구요.

import sqlite3

try:
    # SQLITE 데이터베이스를 연결 또는 없으면 자동 생성합니다.
    sqliteConnection = sqlite3.connect('상품.db')

    # SQLITE 를 제어할 핸들을 가져옵니다.
    cursor = sqliteConnection.cursor()
    
    # 데이터베이스 내에 테이블을 생성합니다.
    cursor.execute ("""CREATE TABLE IF NOT EXISTS 상품 (
      상품번호 INT PRIMARY KEY,
      상품명 VARCHAR(30) NOT NULL,
      가격 INT NOT NULL )""")
    sqliteConnection.commit()

    # 데이터베이스에 입력할 데이터를 정의합니다.
    data=[
      [1, '고나초콜랫🧆', 2000],
      [2, '웨아스🧇', 2500],
      [3, '사망고🥮', 3000]
    ]

    # 데이터를 한꺼번에 입력합니다.
    cursor.executemany("""INSERT INTO 상품(상품번호, 상품명, 가격)
					VALUES(?, ?, ?)""", data)
    sqliteConnection.commit()

except sqlite3.Error as error:
     print("Failed to insert data into sqlite table", error)

# SQLITE 를 제어할 핸들을 가져옵니다.
cursor = sqliteConnection.cursor()
cursor.execute("SELECT * FROM 상품 ORDER BY 상품번호 DESC")
rows = cursor.fetchall()
for row in rows:
    print(row)

cursor.close()

실행 결과로 SQLITE 데이터베이스 파일이 생성되고,

데이터를 입력한 후 다시 데이터를 꺼내 역순으로 출력하는 간단한 소스입니다.

이걸 응용하면 구글 시트 <=> SQLITE 간 데이터를 왕래할 수도 있지요.
언젠가 유용하게 써먹기를 기대하며 글을 남깁니다 :)