X-Git-Url: http://git.euphorik.ch/?p=pompage.git;a=blobdiff_plain;f=src%2Ffilm.rb;h=24e37b8776a163994b3ed1a553fa0c5bf24f8326;hp=b79584129237556386e5841c34548651734be1ba;hb=29dd69679e2fa7c1b9d5a28265ad03aeac0f13b9;hpb=f67e80719e42ae1e95511df44d65d493a3ead62f diff --git a/src/film.rb b/src/film.rb index b795841..24e37b8 100644 --- a/src/film.rb +++ b/src/film.rb @@ -1,150 +1,39 @@ -require 'rexml/document' -require 'net/http' -require 'thread' -require 'thwait' -require 'cgi' +# coding: utf-8 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 = {} +require 'modules/allocine.rb' - @@mutex = Mutex::new - @@threadsWait = ThreadsWait::new - @@nbConn = 0 - - - # 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) - 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 +# Représente un Film. +# Permet de charger des données depuis allocine.fr +# Permet d'effectuer des recherches depuis allocine.fr +class Film + + # toutes les données membres sont accessibles par défaut (écriture/lecture) + attr_accessor :id, :titre, :url, :fichiers, :mod, :annee, :realisateurs, :acteurs, :pays, :duree, :critiquePresse, :critiqueSpectateur, :genres, :synopsis, :budget - - def initialize(fichier) - @fichier = fichier + # Constructeur. N'entreprend aucune action (chargement), crée juste un film vide. + # p1 [String] : le fichier correspondant au film + def initialize(fichier, mod) + @fichiers = [] # le chemin des fichiers est relatif au repertoire de base + @fichiers << fichier if fichier != nil && fichier != '' + + @mod = mod + + @id = 0 @titre = '' @annee = nil @realisateurs = [] @@ -157,136 +46,233 @@ private @synopsis = nil @budget = nil @budgetUnite = 'euro' + @url = nil + + @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 + + def titre=(t) + if t == nil + @titre = "" + else + @titre = t end + 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 + + # Demande à l'utilisateur de faire un choix. + # ret : 1 si le conflit à été résolu, 2 si le film est à ignorer, 3 si tout les films sont à ignorer + 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 "______" + puts "A. Passer et l'ajouter" + puts "B. Ignorer" + puts "C. Tout ignorer" + choix = STDIN.gets + + if /A/i =~ choix + return 1 + elsif /B/i =~ choix + return 2 + elsif /C/i =~ choix + return 3 + end + + choix = choix.to_i + if choix > 0 && choix <= tabNoms.length + break; + else + puts + puts "Choix pas bon !!" + end + end + + @mod.load(@choix[tabNoms[choix-1]], self) + + return 1 + end - #extrait le nom à partire du nom du fichier - /^.*?([^\/]*?)\.(.{3,4})$/ =~ @fichier + # 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!(/\[.*?\]/,' ') + 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 = @mod.rechercherFilm(titre) + @nbReponses = reponses.size + + if @nbReponses == 1 + @mod.load(reponses.values[0], self) + else + @choix = reponses end + self end - def getXml + def cheminImage + return $repBase + "/" + $repAffichette + @id.to_s + ".jpg"; + end + + def setImage(image) + f = File::new(cheminImage, 'w').binmode + f.write(image) + f.close + end + + def possedeImage? + return File::exist?(cheminImage) + end + + # Met à jour un ou plusieurs champs correspondant à 'champ'. + def update(champ) + return if @url == nil || @url == "" + @mod.setChamp(champ) + @mod.loadURL(@url, self) + end + + # 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 + + def to_s + acc = "'#{@titre}'\n" + + " id = #{@id}\n" + + " annee = #{@annee}\n" + + " réalisateurs :\n" + @realisateurs.each{|r| + acc += " - " + r.to_s + } + acc += " acteurs :\n" + @acteurs.each{|a| + acc += " - " + a.to_s + } + acc += " pays :\n" + @pays.each{|p| + acc += " - " + p.to_s + } + acc += " genres :\n" + @genres.each{|g| + acc += " - " + g.to_s + } + acc += " fichiers :\n" + @fichiers.each{|f| + acc += " - " + f + "\n" + } + + acc += " durée : #{@duree}\n" + acc += " critique presse : #{@critiquePresse}\n" + acc += " critique specatateur : #{@critiqueSpectateur}\n" + acc += " synopsis : #{@synopsis}\n" + acc += " budget : #{@budget} #{@budgetUnite}\n" + acc += " url : #{@url}\n" + + return acc + end end - \ No newline at end of file +