import psycopg2
import csv
import random
class Grade():
def __init__(self, file_data):
self.file_data = file_data
self.connection = psycopg2.connect("dbname='grade_db' user='postgres' host='localhost' password='fern36612'")
self.cursor = self.connection.cursor()
def create_table(self):
self.cursor.execute("CREATE TABLE IF NOT EXISTS student(number_id TEXT Primary Key, name TEXT, surename TEXT)")
self.connection.commit()
self.cursor.execute("CREATE TABLE IF NOT EXISTS transcript(subject TEXT, credit INTEGER, section INTEGER, grade TEXT, term INTEGER, number_id TEXT, FOREIGN KEY (number_id) REFERENCES student(number_id))")
self.connection.commit()
self.connection.close
def get_data(self):
n = 6900000000001
list_subject = ["010123101 INTRODUCTION TO COMPUTER", "010123102 PROGRAMMING FUNDAMENTALS", "040203111 ENGINEERING MATHEMATICS", "040313005 PHYSICS"]
list_grade = ["A", "B", "C", "D"]
for i in range(1000000):
self.insert_data(n, "Jessada", "Weeradetkumpon")
for j in range(40):
k = random.randint(0,3)
self.insert_grade(list_subject[k],"1", "1", list_grade[k], 1, n)
print(i,": success")
n+=1
self.connection.commit()
self.connection.close
def insert_data(self, number_id, name, surename):
sql = "INSERT INTO student (number_id, name, surename) VALUES (%s, %s, %s)"
self.cursor.execute(sql, (number_id, name, surename))
def insert_grade(self, subject, credit, section, grade, term, number_id):
sql = "INSERT INTO transcript (subject, credit, section, grade, term, number_id) VALUES(%s, %s, %s, %s, %s, %s)"
self.cursor.execute(sql, (subject, credit, section, grade, term, number_id))
def show_grade(self):
self.cursor.execute("SELECT * FROM grade Where number_id = '5801012620097'")
records = self.cursor.fetchall()
self.connection.close
for i in records:
print(i)
a = Grade('grade.csv')
#a.create_table()
#a.get_data()
#a.show_grade()
ไม่มีความคิดเห็น:
แสดงความคิดเห็น