spacepaste

  1.  
  2. from imdb import IMDb, IMDbError
  3. imdb = IMDb()
  4. def roleWriterDirectorActor(nmID):
  5. """Find movie with same Writer, Director, & Actor"""
  6. writer,director,actor,triRoles = [],[],[],[]
  7. for i in range(1):
  8. try:
  9. person = imdb.get_person(nmID)
  10. for i in person['writer']:
  11. writer.append(i)
  12. for i in person['director']:
  13. director.append(i)
  14. for i in person['actor']:
  15. actor.append(i)
  16. except KeyError:
  17. continue
  18. for x in writer:
  19. for y in director:
  20. for z in actor:
  21. if x == y == z: # Check and compare three roles, if true then add to triRoles
  22. triRoles.append(x)
  23. if len(triRoles) > 0:
  24. print person['name']
  25. for i in triRoles:
  26. print " ", i, " - ", imdb.get_imdbURL(i)
  27. for i in range(1,235): # Automatically check all nm0000001 to nm0000235
  28. roleWriterDirectorActor(str(i))
  29. #roleWriterDirectorActor('233') # Manually check once with nm####### (note: can't have leading zeros)
  30.