X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=src%2Ffilm.rb;h=b56ad072cd78573f0fdb122fe065df198d608f1b;hb=f2d0f55dc2e3dc77561c26703e382b1917498242;hp=b79584129237556386e5841c34548651734be1ba;hpb=f67e80719e42ae1e95511df44d65d493a3ead62f;p=pompage.git diff --git a/src/film.rb b/src/film.rb index b795841..b56ad07 100644 --- a/src/film.rb +++ b/src/film.rb @@ -1,150 +1,40 @@ -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' -class Film - attr_accessor :titre, :fichier, :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 - @@filmsFichier = {} - - @@mutex = Mutex::new - @@threadsWait = ThreadsWait::new - @@nbConn = 0 +require 'modules/allocine.rb' - - # Lit un repertoire de manière recursive - def Film::litRepertoire(r) - Film::litRepertoireR(r) - # on attends que les threads se terminent - @@threadsWait.all_waits +# ajout de deux méthodes à la classe String +class String + def virerBalisesHTML + return gsub(/<(.*?)>/, '') 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| - fichier = e.get_elements('fichier')[0].get_text - next if fichier == nil - titre = e.get_elements('titre')[0].get_text - 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('realisateur') - acteurs = e.get_elements('acteur') - pays = e.get_elements('pays') - genres = e.get_elements('genres') - - - film = Film::new(fichier.value) - film.titre = titre.value unless titre.nil? - film.annee = annee.value unless annee.nil? - realisateurs.each{|e| - film.realisateurs << Personne::ajouter(e.get_text.value) - } - 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) - } - film.synopsis = synopsis.value unless synopsis.nil? - film.budget = budget.value unless budget.nil? - @@films[film.titre] = film - @@filmsFichier[film.fichier] = film - } + def virerBalisesHTML! + gsub!(/<(.*?)>/, '') end +end - # Renvoie tous les films sous la forme d'un élement XML. - def Film::getFilmsXml - racine = REXML::Element::new('filmographie') - @@films.each{|nom, f| - racine.add(f.getXml) - } - return REXML::Document::new.add(racine) +class Film + + def Film::setModule(m) + @@module = m end -private - - def Film::filmsFactory(fichier) - /^.*?\.(.{3,4})$/ =~ fichier - if FILMS_EXTENSIONS.include?($1) - Film::new(fichier).loadData - end - end - - def Film::litRepertoireR(r) - Dir::foreach(r){|f| - next if f == '.' or f == '..' - fichier = r + "/" + f - if File::directory?(fichier) - litRepertoireR(fichier) - else - - #si le film n'existe pas déjà dans la liste - if film = @@filmsFichier[fichier] - puts "[i] Already exists in DB : #{film.titre} (#{film.fichier})" - next - end - - film = nil - @@nbConn += 1 - @@threadsWait.join_nowait( - Thread::new{ - begin - @@mutex.lock if @@nbConn >= NB_CONN_MAX - #p @@nbConn - film = Film::filmsFactory(fichier) - unless film.nil? - if @@films.has_key?(film.titre) - puts "[!] Duplicate movie : #{film.titre} (#{film.fichier})" - else - puts "[i] movie added : #{film.titre} (#{film.fichier})" - @@films[film.titre] = film - @@filmsFichier[film.fichier] = film - end - end - @@nbConn -= 1 - #p @@nbConn - @@mutex.unlock - rescue Exception => e - puts e.message - puts e.backtrace - end - } - ) - end - } - end + # 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 - + # Constructeur. N'entreprend aucune action (chargement), crée juste un film vide. + # p1 [String] : le fichier correspondant au film def initialize(fichier) - @fichier = fichier + @fichiers = [fichier] # le chemin des fichiers est relatif au repertoire de base + + @id = 0 @titre = '' @annee = nil @realisateurs = [] @@ -157,134 +47,163 @@ private @synopsis = nil @budget = nil @budgetUnite = 'euro' + @url + + @nbReponses = 0 + + # mémorise les tuples {nom => id} dans le cas ou il y a plusieurs choix de films après une recherche + @choix = {} end -public - #charge les informations du films à partir d'allocine.fr - def loadData - unless LOAD_DATA - @titre = @fichier - return self - end + # Est-ce qu'il y a eu plusieurs réponses pour ce film lors de la cherche sur le net ? + def plusieursReponses? + return @nbReponses > 1 + end - connexionHttp = Net::HTTP::new('www.allocine.fr'); + # 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 - #extrait le nom à partire du nom du fichier - /^.*?([^\/]*?)\.(.{3,4})$/ =~ @fichier + # Demande à l'utilisateur de faire un choix. + # ret : true si le conflit à été résolu sinon false + def reglerConflitPlusieursReponses + @nbReponses = 1 + + puts " -> " + @fichiers[0] + puts "Fais ton choix jeune padawan (un caractère et pas plus)" + tabNoms = @choix.keys + choix = 1 + loop do + i = 1 + tabNoms.each{|n| + puts "#{i}. #{n}" + i += 1 + } + puts "A. Passer et l'ajouter" + puts "B. Ignorer" + choix = STDIN.gets + + if /A/i =~ choix + return true + elsif /B/i =~ choix + return false + end + + choix = choix.to_i + if choix > 0 && choix <= tabNoms.length + break; + else + puts + puts "Choix pas bon !!" + end + end + + @@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 + end + end + + # Charge les informations du films à partir d'allocine.fr + # ret [Film] + def loadData + #extrait le nom à partir du nom du fichier + /^.*?([^\/]*?)\.(.{3,4})$/ =~ @fichiers[0] #remplace undescores et points par des espaces titre = $1.gsub(/[_\.]/, ' ') - #vire les espaces au début et à la fin - titre.strip! #remplace les suites d'espaces par un seul titre.gsub!(/ {2,}/,' ') titre.gsub!(/\[.*?\]/,'') titre.gsub!(/\(.*?\)/,'') + titre.gsub!(/\{.*?\}/,'') + #vire les espaces au début et à la fin + titre.strip! + @titre = titre.dup - donneesHtml = nil - begin - reponse, donneesHtml = connexionHttp.get("/recherche/?motcle=#{CGI::escape(titre)}") - #si pas trouvé alors on enlève un mot à la fin - if /.*?Pas de résultats.*?/ =~ donneesHtml - /(.*?)[^ ]+?$/ =~ 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 - if $1 - r, ficheHtml = connexionHttp.get("/film/fichefilm_gen_cfilm=#{$1}.html") - - # Titre - /(.*?)<\/title>/ =~ ficheHtml - @titre = $1 unless $1.nil? - - #puts "Movie found : #{@titre} (#{@fichier})" - - # Année - /<h4>Année de production : (\d+)<\/h4>/ =~ ficheHtml - @annee = $1.to_i unless $1.nil? - - # Réalisateurs - /Réalisé par <a class="link1" href=".*?" target="">(.*?)<\/a>/ =~ ficheHtml - @realisateurs << Personne::ajouter($1) unless $1.nil? - - # Acteurs - /Avec(.*)/ =~ ficheHtml - $1.scan(/<a class="link1" href="\/personne\/fichepersonne_gen_cpersonne=\d+\.html" target="">(.+?)<\/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.<\/h4>/ =~ ficheHtml - @duree = $1.nil? ? $2.to_i : $1.to_i * 60 + $2.to_i - - # Critique presse - /Presse<\/a> <img src=".*?etoile_(\d)\.gif"/ =~ ficheHtml - @critiquePresse = $1.to_i unless $1.nil? - - # Critique spectateur - /Spectateurs<\/a> <img src=".*?etoile_(\d)\.gif"/ =~ ficheHtml - @critiqueSpectateur = $1.to_i unless $1.nil? - - # Genre - /Genre : (.*)/ =~ 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? - 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 - def getXml + # 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') - racine.add(REXML::Element::new('fichier').add_text(@fichier)) + racine.add_attribute('id', @id.to_s) + + fichiers = REXML::Element::new('fichiers') + @fichiers.each{|f| + fichiers.add(REXML::Element::new('fichier').add_text(f)) + } + 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| - racine.add(REXML::Element::new('realisateur').add_text(r.nom)) + realisateurs.add(REXML::Element::new('realisateur').add_text(r.nom)) } + racine.add(realisateurs) + acteurs = REXML::Element::new('acteurs') @acteurs.each{|a| - racine.add(REXML::Element::new('acteur').add_text(a.nom)) + acteurs.add(REXML::Element::new('acteur').add_text(a.nom)) } + racine.add(acteurs) + + lespays = REXML::Element::new('lespays') @pays.each{|p| - racine.add(REXML::Element::new('pays').add_text(p.nom)) + lespays.add(REXML::Element::new('pays').add_text(p.nom)) } + racine.add(lespays) + 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| - racine.add(REXML::Element::new('genre').add_text(g.nom)) + genres.add(REXML::Element::new('genre').add_text(g.nom)) } - racine.add(REXML::Element::new('synopsis').add_text(@synopsis)) + racine.add(genres) + + 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) + + racine.add(REXML::Element::new('url').add_text(@url)) racine end