ValueError: недопустимый литерал для int () с основанием 10: 'SOH'

Question

Я получил следующую ошибку при попытке вставить значения в базу данных sqlite

Traceback (most recent call last):
  File "C:\SS\Problem13\src\database.py", line 23, in <module>
    sql = """INSERT INTO students (name, student_id, email, school_id, gender, gpa, address) VALUES ('%s','%i','%s','%i','%s','%f','%s')"""%(data[0], int(data[1]),data[2],int(data[3]),data[4],float(data[5]),data[6])
ValueError: invalid literal for int() with base 10: 'SOH'

Ниже приведен мой код:

import sqlite3
import csv
# Connect to database, if it does not exist, it will create it.
conn = sqlite3.connect("nopoly.db")
# Creating table.
conn.execute("PRAGMA foreign_keys = 1")
conn.execute("CREATE TABLE schools (id INTEGER PRIMARY KEY, name TEXT NOT NULL, description TEXT NOT NULL, director TEXT NOT NULL, email TEXT NOT NULL)")
conn.execute("CREATE TABLE students (id INTEGER PRIMARY KEY, name TEXT NOT NULL, student_id INTEGER NOT NULL, email TEXT NOT NULL, school_id INTEGER NOT NULL REFERENCES schools, gender TEXT NOT NULL, gpa FLOAT NOT NULL, address TEXT NOT NULL)")
# Opening the csv files
schoolreader = csv.reader(open("files/schools.csv"))
studentreader = csv.reader(open("files/studentsp13.csv"))
# Skip the headers
schoolreader.next()
studentreader.next()
# Extract each row and print
for data in schoolreader:
    print data
    sql = """INSERT INTO schools (name, description, director, email) VALUES ('%s','%s','%s','%s')"""%(data[0],data[1],data[2],data[3])
    print "DEBUG", sql
    conn.execute(sql)
for data in studentreader:
    print data
    sql = """INSERT INTO students (name, student_id, email, school_id, gender, gpa, address) VALUES ('%s','%i','%s','%i','%s','%f','%s')"""%(data[0], int(data[1]),data[2],int(data[3]),data[4],float(data[5]),data[6])
    print "DEBUG", sql
    conn.execute(sql)
#
conn.commit()

Я понимаю, что в студентов p13.csv школыпомечены как строки.Я хочу знать, как отобразить его в целые числа так, чтобы оно соответствовало таблице "школы".

Answer 1

Я получил следующую ошибку при попытке вставить значения в базу данных sqlite

Вместо того, чтобы использовать форматирование строки для вставки ваших параметров в ваши запросы, вы должны передать их как параметры методу conn.execute(). Таким образом вы защищаете себя от атак с использованием SQL-инъекций, и данные обрабатываются базой данных как правильный тип. Документы SQLite Python содержат подробности и примеры того, как это сделать.

Я понимаю, что в studentp13.csv школы помечены как строки. Я хотел бы знать, как отобразить его в целые числа так, чтобы оно соответствовало таблице "школы".

Когда вы вставляете записи в таблицу школ, вы можете записать идентификаторы, сгенерированные SQLite, в словарь, а затем снова просмотреть их, когда будете вставлять записи учеников. Это будет выглядеть примерно так:

schools = {}
for data in schoolreader:
    print data
    sql = """INSERT INTO schools (name, description, director, email) VALUES ('%s','%s','%s','%s')"""%(data[0],data[1],data[2],data[3])
    conn.execute(sql)
    schools[data[0]] = conn.lastrowid
for data in studentreader:
    school_id = schools[data[3]]
    sql = """INSERT INTO students (name, student_id, email, school_id, gender, gpa, address) VALUES ('%s','%i','%s','%i','%s','%f','%s')"""%(data[0], int(data[1]),data[2],school_id,data[4],float(data[5]),data[6])
    conn.execute(sql)