def addUserContent(self,
user_id,
content_id,
completion_status=None,
owned=None,
user_rating=None):
columns = "user_id,content_id,"
values = "'{}','{}',".format(user_id, content_id)
columns += "completion_status,"
if completion_status == "on":
values += "'true',"
else:
values += "'false',"
columns += "owned,"
if owned == "on":
values += "'true',"
else:
values += "'false',"
if user_rating is not None:
columns += "user_rating,"
values += "'{}',".format(user_rating)
if len(columns) > 0 and columns[-1] == ',':
columns = columns[:-1]
if len(values) > 0 and values[-1] == ',':
values = values[:-1]
myDB.insertTable("user_content", columns, values, "user_id")
def addUser(self, username, password_hash):
columns = "username, password_hash"
username = username.replace("'", "''")
values = "'{}','{}'".format(username, password_hash)
user_id = myDB.insertTable("user_table", columns, values, "user_id")
return user_id
def addMovie(self,
director=None,
release_year=None,
language=None,
length=None,
imdb_id=None):
columns = ""
values = ""
columns += "director,"
if director is not None and director != '' and director != "None":
director = director.replace("'", "''")
values += "'{}',".format(director)
else:
values += "NULL,"
columns += "release_year,"
if release_year is not None and release_year != '':
values += "'{}',".format(release_year)
else:
values += "NULL,"
columns += "language,"
if language is not None and language != '':
language = language.replace("'", "''")
values += "'{}',".format(language)
else:
values += "NULL,"
columns += "length,"
if length is not None and length != '':
values += "'{}',".format(length)
else:
values += "NULL,"
columns += "imdb_id"
if imdb_id is not None and imdb_id != '':
imdb_id = imdb_id.replace("'", "''")
values += "'{}'".format(imdb_id)
else:
values += "NULL"
movie_id = myDB.insertTable("movie", columns, values, "movie_id")[0]
return movie_id
def addBook(self,
author=None,
release_year=None,
language=None,
no_pages=None,
isbn=None):
columns = ""
values = ""
columns += "author,"
if author is not None and author != '' and author != "None":
author = author.replace("'", "''")
values += "'{}',".format(author)
else:
values += "NULL,"
columns += "release_year,"
if release_year is not None and release_year != '':
values += "'{}',".format(release_year)
else:
values += "NULL,"
columns += "language,"
if language is not None and language != '':
language = language.replace("'", "''")
values += "'{}',".format(language)
else:
values += "NULL,"
columns += "no_pages,"
if no_pages is not None and no_pages != '':
values += "'{}',".format(no_pages)
else:
values += "NULL,"
columns += "isbn"
if isbn is not None and isbn != '':
isbn = isbn.replace("'", "''")
values += "'{}'".format(isbn)
else:
values += "NULL"
book_id = myDB.insertTable("book", columns, values, "book_id")[0]
return book_id
def addContent(self, title, content_type=None, type_specific_id=None):
columns = "title,"
title = title.replace("'", "''")
values = "'{}',".format(title)
if content_type is not None and content_type != '':
content_type = content_type.replace("'", "''")
columns += "content_type,"
values += "'{}',".format(content_type)
if type_specific_id is not None and type_specific_id != '':
columns += "type_specific_id,"
values += "'{}',".format(type_specific_id)
if len(columns) > 0 and columns[-1] == ',':
columns = columns[:-1]
if len(values) > 0 and values[-1] == ',':
values = values[:-1]
content_id = myDB.insertTable("content", columns, values,
"content_id")[0]
return content_id
def addSeries(self,
release_year=None,
language=None,
no_seasons=None,
imdb_id=None):
columns = ""
values = ""
columns += "release_year,"
if release_year is not None and release_year != '':
values += "'{}',".format(release_year)
else:
values += "NULL,"
columns += "language,"
if language is not None and language != '':
language = language.replace("'", "''")
values += "'{}',".format(language)
else:
values += "NULL,"
columns += "no_seasons,"
if no_seasons is not None and no_seasons != '':
values += "'{}',".format(no_seasons)
else:
values += "NULL,"
columns += "imdb_id"
if imdb_id is not None and imdb_id != '':
imdb_id = imdb_id.replace("'", "''")
values += "'{}'".format(imdb_id)
else:
values += "NULL"
series_id = myDB.insertTable("series", columns, values, "series_id")[0]
return series_id
def addContentGenre(self, content_id, genre):
columns = "content_id, genre"
genre = genre.replace("'", "''")
values = "'{}','{}'".format(content_id, genre)
myDB.insertTable("content_genre", columns, values)