pythonTools/createTables.py

   1 #!/usr/bin/env python
   2 # -*- coding:latin-1 -*-
   3
   4 import sys
   5 import pg
   6
   7 def create_fullcredits_table(db):
   8    cmd = """CREATE TABLE fullcredits (
   9 imdb_id           integer,
  10 Directed_by       integer[],
  11 Writing_credits   integer[],
  12 Fullcast              integer[],
  13 User_rating       float8,
  14 MPAA_rating       text,
  15 Produced_by       integer[],
  16 Original_Music_by integer[],
  17 Cinematography_by integer[],
  18 Film_Editing_by   integer[],
  19 Casting_by        integer[],
  20 Production_Design_by integer[],
  21 Art_Direction_by     integer[],
  22 Genre                text[],
  23 Country              text[],
  24 Keywords             text[],
  25 Plot_Summary         text[] );"""
  26
  27    print cmd
  28    db.query(cmd)
  29
  30
  31 def create_people_table(db):
  32    cmd = """CREATE TABLE people (
  33 id integer,
  34 name text,
  35 first_name text,
  36 profession text[]);"""
  37
  38    db.query(cmd)
  39
  40
  41 if __name__ == '__main__':
  42    db = pg.connect('imdb_project','localhost')
  43    #create_people_table(db)
  44    create_fullcredits_table(db)
  45    db.close()