X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=src%2Ffilm.rb;h=b56ad072cd78573f0fdb122fe065df198d608f1b;hb=f2d0f55dc2e3dc77561c26703e382b1917498242;hp=91dcf37f18c3056a17aef89aaceefb3279a2a903;hpb=2a499b47db2419a221fed7cc0e9b899b2505d26f;p=pompage.git diff --git a/src/film.rb b/src/film.rb index 91dcf37..b56ad07 100644 --- a/src/film.rb +++ b/src/film.rb @@ -1,219 +1,38 @@ -require 'rexml/document' -require 'net/http' -require 'thread' -require 'thwait' -require 'cgi' +# Représente un Film. +# Permet de charger des données depuis allocine.fr +# Permet d'effectuer des recherches depuis allocine.fr +# TODO : rendre la classe indépendant de la source de donnée pour pouvoir utiliser d'autres sites comme par exemple imdb.com require 'pays.rb' require 'genre.rb' require 'personne.rb' - require 'constantes.rb' +require 'modules/allocine.rb' + +# ajout de deux méthodes à la classe String class String def virerBalisesHTML - return self.gsub(/<(.*?)>/, '') + return gsub(/<(.*?)>/, '') + end + def virerBalisesHTML! + gsub!(/<(.*?)>/, '') end end class Film - attr_accessor :id, :titre, :fichiers, :annee, :realisateurs, :acteurs, :pays, :duree, :critiquePresse, :critiqueSpectateur, :genres, :synopsis, :budget - - # Les films indexés par leur titre - @@films = {} - - # Les films indexés par leur nom de fichier, deux fichiers différents peuvent pointer sur le même film - @@filmsFichier = {} - - # Les films qui ont plusieurs réponses lors de la recherche, traités à la fin - @@filmsPlusieursReponses = [] - - @@mutex = Mutex::new - @@threadsWait = ThreadsWait::new - @@nbConn = 0 - - # le prochain id disponible - @@idDisponible = 1 - - # retourne un nouvel id, utilisé lors de la création d'un nouveau film - def Film::getNewId - id = @@idDisponible - @@idDisponible += 1 - return id - end - - # Lit un repertoire de manière recursive - def Film::litRepertoire(r) - Film::litRepertoireR(r) - # on attends que les threads se terminent - @@threadsWait.all_waits - - # traite les films qui avaient plusieurs réponses lors de la recherche - # l'utilisateur doit faire un choix - @@filmsPlusieursReponses.each{|f| - if f.reglerConflitPlusieursReponses - Film::ajouterFilm(f) - end - } - end - # Charge les films contenus dans un fichier XML. - def Film::loadFilmsXml(xmlFile) - # si le fichier n'existe pas il n'y a rien à charger - if !File.exists?(xmlFile) - return - end - - racine = REXML::Document::new(File::new(xmlFile)).root - racine.each_element{|e| - id = e.attribute('id').to_s.to_i - - if id > @@idDisponible - @@idDisponible = id + 1 - end - - titre = e.get_elements('titre')[0].get_text - - fichiers = e.get_elements('fichiers')[0].get_elements('fichier') - annee = e.get_elements('annee')[0].get_text - duree = e.get_elements('duree')[0].get_text - critiquePresse = e.get_elements('critiquePresse')[0].get_text - critiqueSpectateur = e.get_elements('critiqueSpectateur')[0].get_text - synopsis = e.get_elements('synopsis')[0].get_text - budget = e.get_elements('budget')[0].get_text - realisateurs = e.get_elements('realisateurs')[0].get_elements('realisateur') - acteurs = e.get_elements('acteurs')[0].get_elements('acteur') - pays = e.get_elements('lespays')[0].get_elements('pays') - genres = e.get_elements('genres')[0].get_elements('genre') - - film = Film::new(fichiers[0].get_text.value) - - film.titre = titre.value unless titre.nil? - film.id = id - fichiers.each{|e| - film.addFichier(e.get_text.value) - @@filmsFichier[e.get_text.value] = film - } - film.annee = annee.value unless annee.nil? - acteurs.each{|e| - film.acteurs << Personne::ajouter(e.get_text.value) - } - pays.each{|e| - film.pays << Pays::ajouter(e.get_text.value) - } - film.duree = duree.value unless duree.nil? - film.critiquePresse = critiquePresse.value unless critiquePresse.nil? - film.critiqueSpectateur = critiqueSpectateur.value unless critiqueSpectateur.nil? - genres.each{|e| - film.genres << Genre::ajouter(e.get_text.value) if e.get_text != nil - } - film.synopsis = synopsis.value unless synopsis.nil? - film.budget = budget.value unless budget.nil? - @@films[film.titre] = film - } - end - - # Renvoie tous les films sous la forme d'un document XML. - def Film::getFilmsXml - # le document - docXml = REXML::Document::new - docXml.xml_decl().encoding = "UTF-8" # normalement UTF-8 - docXml.xml_decl().dowrite - - # la racine du document - racine = REXML::Element::new('filmographie') - docXml.add(racine) - pi = REXML::Instruction.new("xml-stylesheet", "type=\"text/xsl\" href=\"../xsl/yopyop.xsl\"") - racine.previous_sibling = pi - - # on ajoute chaque film à la racine - @@films.each{|nom, f| - racine.add(f.getXml) - } - - # revoie le document - docXml + def Film::setModule(m) + @@module = m end -private + # toutes les données membres sont accessibles par défaut (écriture/lecture) + attr_accessor :id, :titre, :url, :fichiers, :annee, :realisateurs, :acteurs, :pays, :duree, :critiquePresse, :critiqueSpectateur, :genres, :synopsis, :budget - def Film::filmsFactory(fichier) - Film::new(fichier).loadData - end - - def Film::litRepertoireR(r) - Dir::foreach(r){|f| - next if f == '.' or f == '..' - fichier = r + "/" + f - if File::directory?(fichier) - litRepertoireR(fichier) - else - - # vérification de l'extension - /^.*?\.(.{3,4})$/ =~ fichier - if !FILMS_EXTENSIONS.include?($1) - next - end - - fichier = CGI::escapeHTML(fichier.unpack("C*").pack("U*")) - - # on skip si le film est déjàa dans la BD - if film = @@filmsFichier[fichier] - puts "[i] Already exists in DB : #{film.titre} (#{fichier})" - next - end - - #p fichier - - film = nil - - @@nbConn += 1 - @@threadsWait.join_nowait( - Thread::new{ - begin - @@mutex.lock if @@nbConn >= NB_CONN_MAX - film = Film::filmsFactory(fichier) - unless film.nil? # le film a été correctement construit - Film::ajouterFilm(film) - end - @@nbConn -= 1 - @@mutex.unlock - rescue Exception => e - puts e.message - puts e.backtrace - end - } - ) - end - } - end - - def Film::ajouterFilm(film) - if film.plusieursReponses? - @@filmsPlusieursReponses << film - return - end - - # le film existe déjà - if @@films.has_key?(film.titre) - # le fichier n'est pas connu -> nième partie d'un film - if !@@filmsFichier.has_key?(film.fichiers[0]) - puts "[i] movie #{film.titre} has a another file part : #{film.fichiers[0]}" - @@films[film.titre].addFichier(film.fichiers[0]) - @@filmsFichier[film.fichiers[0]] = @@films[film.titre] - else - puts "[!] Duplicate movie : #{film.titre} (#{film.fichiers[0]})" - end - else - puts "[i] movie added : #{film.titre} (#{film.fichiers[0]})" - @@films[film.titre] = film - @@filmsFichier[film.fichiers[0]] = film - end - end - - + # Constructeur. N'entreprend aucune action (chargement), crée juste un film vide. + # p1 [String] : le fichier correspondant au film def initialize(fichier) - @fichiers = [fichier] + @fichiers = [fichier] # le chemin des fichiers est relatif au repertoire de base @id = 0 @titre = '' @@ -229,28 +48,31 @@ private @budget = nil @budgetUnite = 'euro' @url + + @nbReponses = 0 - @aPlusieursReponses = false - # mémorise les tuples {nom => id} dans le cas ou il y a plusieurs réponses - @idsAllocine = {} + # mémorise les tuples {nom => id} dans le cas ou il y a plusieurs choix de films après une recherche + @choix = {} end -public - + # Est-ce qu'il y a eu plusieurs réponses pour ce film lors de la cherche sur le net ? def plusieursReponses? - return @aPlusieursReponses + return @nbReponses > 1 + end + + # Est-ce qu'il y a eu plusieurs réponses pour ce film lors de la cherche sur le net ? + def nbReponses + return @nbReponses end - # demande à l'utilisateur de faire un choix + # Demande à l'utilisateur de faire un choix. # ret : true si le conflit à été résolu sinon false def reglerConflitPlusieursReponses + @nbReponses = 1 - @aPlusieursReponses = false # pour faire les choses bien - - puts - puts "Plop, ya un conflit : #{@fichiers[0]}" + puts " -> " + @fichiers[0] puts "Fais ton choix jeune padawan (un caractère et pas plus)" - tabNoms = @idsAllocine.keys + tabNoms = @choix.keys choix = 1 loop do i = 1 @@ -277,11 +99,12 @@ public end end - loadDepuisIdAllocine(@idsAllocine[tabNoms[choix-1]]) + @@module.load(@choix[tabNoms[choix-1]], self) return true end + # Ajoute un fichier comme faisant partie du film def addFichier(fichier) if !@fichiers.include?(fichier) @fichiers << fichier @@ -290,16 +113,7 @@ public # Charge les informations du films à partir d'allocine.fr # ret [Film] - def loadData - unless LOAD_DATA - @titre = @fichiers[0] - return self - end - - @id = Film::getNewId - - connexionHttp = Net::HTTP::new('www.allocine.fr') - + def loadData #extrait le nom à partir du nom du fichier /^.*?([^\/]*?)\.(.{3,4})$/ =~ @fichiers[0] #remplace undescores et points par des espaces @@ -314,119 +128,24 @@ public @titre = titre.dup - donneesHtml = nil - begin - begin - reponse, donneesHtml = connexionHttp.get("/recherche/?motcle=#{CGI::escape(titre.unpack("U*").pack("C*"))}") - rescue Exception => e - p e - puts "[!] Connexion lost, retry.." - retry - end - - #convertit le code latin-1 en UTF8 - donneesHtml = donneesHtml.unpack("C*").pack("U*") - - #si pas trouvé alors on enlève un mot à la fin - if /.*?Pas de résultats.*?/ =~ donneesHtml || ! donneesHtml.include?("

Films

") - /(.*?)[^ ]+?$/ =~ titre.strip - titre = $1 - titre.strip! - else - break; - end - end while not titre.nil? and not titre.empty? + unless LOAD_DATA + return self + end - unless titre.nil? or titre.empty? - - #// =~ donneesHtml - #r = donneesHtml.scan(/(.*?)<\/a>/) - r = donneesHtml.scan(/(.*?)<\/a>(?:<\/h4>
 (.*?)<\/h5>){0,1}/) - - if r.length > 1 - @aPlusieursReponses = true - r.each{|f| - @idsAllocine[f[1].virerBalisesHTML + (f[2] != nil ? " " + f[2].virerBalisesHTML : "")] = f[0] - } - elsif r.length == 1 - loadDepuisIdAllocine(r[0][0], connexionHttp) - else - puts "[!] Movie not found : #{@titre} (#{@fichier})" - end + reponses = @@module.rechercherFilm(titre) + @nbReponses = reponses.size + + if @nbReponses == 1 + @@module.load(reponses.values[0], self) + else + @choix = reponses end + self end -private - def loadDepuisIdAllocine(id, connexionHttp = nil) - if (connexionHttp == nil) - connexionHttp = Net::HTTP::new('www.allocine.fr') - end - - r, ficheHtml = connexionHttp.get("/film/fichefilm_gen_cfilm=#{id}.html") - - #convertit le code latin-1 en UTF8 - ficheHtml = ficheHtml.unpack("C*").pack("U*") - - #url - @url = "http://www.allocine.fr/film/fichefilm_gen_cfilm=#{id}.html" - - # Titre - /(.*?)<\/title>/ =~ ficheHtml - @titre = $1 unless $1.nil? - - puts "Movie found : #{@titre} (#{@fichiers[0]})" - - # Année - /<h4>Année de production : (\d+)<\/h4>/ =~ ficheHtml - @annee = $1.to_i unless $1.nil? - - # Réalisateurs - /<h4>Réalisé par(.*?)<\/h4>/ =~ ficheHtml - $1.scan(/<a class="link1" href=".*?">(.*?)<\/a>/m){|a| - @realisateurs << Personne::ajouter(a[0]) unless a[0].nil? - } unless $1.nil? - - # Acteurs - /<h4>Avec(.*?)<\/h4>/ =~ ficheHtml - $1.scan(/<a class="link1" href="\/personne\/fichepersonne_gen_cpersonne=\d+\.html">(.+?)<\/a>/m){|a| - @acteurs << Personne::ajouter(a[0]) unless a[0].nil? - } unless $1.nil? - - - # Pays - /<h4>Film (.*?)\. <\/h4>/ =~ ficheHtml - $1.split(',').each{|pays| - @pays << Pays::ajouter(pays) unless pays.nil? - } unless $1.nil? - - # Duree - /<h4>Durée : (\d+)h (\d+)min./ =~ ficheHtml - @duree = $1.nil? ? $2.to_i : $1.to_i * 60 + $2.to_i - - # Critiques presse et spectateur - /Presse.*etoile_([012345]).*Spectateurs.*etoile_([012345])"/m =~ ficheHtml - @critiquePresse = $1.to_i unless $1.nil? - @critiqueSpectateur = $2.to_i unless $2.nil? - - # Genre - /<h4>Genre : (.*?)<\/h4>/ =~ ficheHtml - $1.scan(/<a href="\/film\/alaffiche_genre_gen_genre=.*?" class="link1">(.+?)<\/a>/m){|g| - @genres << Genre::ajouter(g[0]) unless g[0].nil? - } unless $1.nil? - - # Synopsis - /Synopsis.*?<h4>(.+?)<\/h4>/m =~ ficheHtml - @synopsis = $1 unless $1.nil? - - # Budget - /Budget<\/b> : (.+?) millions d'euros<\/h4>/ =~ ficheHtml - @budget = $1.to_i unless $1.nil? - end - -public - - # Renvoie un film sous la forme d'un élément XML de type REXML + # Renvoie un film sous la forme d'un élément XML . + # ret [REXML::Element] : un element xml <film> def getXml racine = REXML::Element::new('film') @@ -439,7 +158,7 @@ public racine.add(fichiers) racine.add(REXML::Element::new('titre').add_text(@titre)) - racine.add(REXML::Element::new('annee').add_text(@annee.to_s)) + racine.add(REXML::Element::new('annee').add_text(@annee)) realisateurs = REXML::Element::new('realisateurs') @realisateurs.each{|r| @@ -461,8 +180,8 @@ public racine.add(REXML::Element::new('duree').add_text(@duree.to_s)) - racine.add(REXML::Element::new('critiquePresse').add_text(@critiquePresse.to_s)) - racine.add(REXML::Element::new('critiqueSpectateur').add_text(@critiqueSpectateur.to_s)) + racine.add(REXML::Element::new('critiquePresse').add_text(@critiquePresse)) + racine.add(REXML::Element::new('critiqueSpectateur').add_text(@critiqueSpectateur)) genres = REXML::Element::new('genres') @genres.each{|g| @@ -470,9 +189,17 @@ public } racine.add(genres) - racine.add(REXML::Element::new('synopsis').add_text(@synopsis)) + synopsisElement = REXML::Element::new('synopsis') + unless @synopsis.nil? + @synopsis.split("\n").each{|s| + next if s =~ /^\s*$/ + synopsisElement.add(REXML::Element::new('p').add_text(s)) + } + end + racine.add(synopsisElement) + budgetElement = REXML::Element::new('budget') - budgetElement.add_text(@budget.to_s) + budgetElement.add_text(@budget) budgetElement.add_attribute('unite', @budgetUnite) racine.add(budgetElement)