X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=src%2Ffilm.rb;h=21c27cbe33c91fb792d5ea051413e5c7f9ea0108;hb=e34f06962c1450efb1aab6d814540c415466a01c;hp=a7bcc54d2a2a2ee2ee51be7dc7c21cc6879e0f65;hpb=868d7866eee089c6e9f9a9a4304b8ab55a05fca1;p=pompage.git diff --git a/src/film.rb b/src/film.rb index a7bcc54..21c27cb 100644 --- a/src/film.rb +++ b/src/film.rb @@ -1,3 +1,5 @@ +# Attention, le but de ce fichier n'est pas d'être le plus clair possible ^-^´´ + require 'rexml/document' require 'net/http' require 'thread' @@ -10,25 +12,67 @@ require 'personne.rb' require 'constantes.rb' +class String + def virerBalisesHTML + return gsub(/<(.*?)>/, '') + end + def virerBalisesHTML! + gsub!(/<(.*?)>/, '') + end +end + class Film - attr_accessor :titre, :fichier, :annee, :realisateurs, :acteurs, :pays, :duree, :critiquePresse, :critiqueSpectateur, :genres, :synopsis, :budget + attr_accessor :id, :titre, :fichiers, :annee, :realisateurs, :acteurs, :pays, :duree, :critiquePresse, :critiqueSpectateur, :genres, :synopsis, :budget + + # repertoire de base, par exemple C:/Divx/ + @@repBase = '' # Les films indexés par leur titre @@films = {} - # Les films indexés par leur nom de fichier + # 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) + @@repBase = r + repPrecedant = Dir::getwd + Dir::chdir(r) + + Film::litRepertoireR('.') + # 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 + i = 1 + @@filmsPlusieursReponses.each{|f| + puts + puts "Plop, ya un conflit #{i} / #{@@filmsPlusieursReponses.length} :" + if f.reglerConflitPlusieursReponses + Film::ajouterFilm(f) + end + i += 1 + } + + Dir::chdir(repPrecedant) end # Charge les films contenus dans un fichier XML. @@ -37,30 +81,38 @@ class Film 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 + 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('realisateur') - acteurs = e.get_elements('acteur') - pays = e.get_elements('pays') - genres = e.get_elements('genres') + 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 = Film::new(fichier.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? - realisateurs.each{|e| - film.realisateurs << Personne::ajouter(e.get_text.value) - } acteurs.each{|e| film.acteurs << Personne::ajouter(e.get_text.value) } @@ -71,13 +123,12 @@ class Film 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.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 - @@filmsFichier[film.fichier] = film - } + } end # Renvoie tous les films sous la forme d'un document XML. @@ -86,10 +137,12 @@ class Film 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| @@ -103,61 +156,86 @@ class Film private def Film::filmsFactory(fichier) - /^.*?\.(.{3,4})$/ =~ fichier - if FILMS_EXTENSIONS.include?($1) - Film::new(fichier).loadData - end - end + Film::new(fichier).loadData + end def Film::litRepertoireR(r) Dir::foreach(r){|f| - next if f == '.' or f == '..' - fichier = r + "/" + f + next if f[0,1] == '.' + fichier = (r == '.' ? '' : r + "/") + f if File::directory?(fichier) litRepertoireR(fichier) else - - #si le film n'existe pas déjà dans la liste + + # 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} (#{film.fichier})" + puts "[i] Already exists in DB : #{film.titre} (#{fichier})" next end + #p fichier + film = nil + if @@nbConn >= NB_CONN_MAX + @@threadsWait.next_wait + end + @@nbConn += 1 @@threadsWait.join_nowait( - Thread::new{ - begin - @@mutex.lock if @@nbConn >= NB_CONN_MAX - 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 + Thread::new{ + begin + film = Film::filmsFactory(fichier) + unless film.nil? # le film a été correctement construit + Film::ajouterFilm(film) end - @@nbConn -= 1 - #p @@nbConn - @@mutex.unlock - rescue Exception => e + rescue Exception => e puts e.message puts e.backtrace - end + end + @@nbConn -= 1 } ) 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 + def initialize(fichier) - #convertit le code latin-1 en UTF8 - @fichier = fichier.unpack("C*").pack("U*") + @fichiers = [fichier] + @id = 0 @titre = '' @annee = nil @realisateurs = [] @@ -171,22 +249,78 @@ private @budget = nil @budgetUnite = 'euro' @url + + @aPlusieursReponses = false + # mémorise les tuples {nom => id} dans le cas ou il y a plusieurs réponses + @idsAllocine = {} end public + def plusieursReponses? + return @aPlusieursReponses + end + + # demande à l'utilisateur de faire un choix + # ret : true si le conflit à été résolu sinon false + def reglerConflitPlusieursReponses + + @aPlusieursReponses = false # pour faire les choses bien + + puts " -> " + @fichiers[0] + puts "Fais ton choix jeune padawan (un caractère et pas plus)" + tabNoms = @idsAllocine.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 + + loadDepuisIdAllocine(@idsAllocine[tabNoms[choix-1]]) + + return true + end + + def addFichier(fichier) + if !@fichiers.include?(fichier) + @fichiers << fichier + end + end + # Charge les informations du films à partir d'allocine.fr # ret [Film] def loadData unless LOAD_DATA - @titre = @fichier + @titre = @fichiers[0] return self end + + @id = Film::getNewId - connexionHttp = Net::HTTP::new('www.allocine.fr'); + connexionHttp = Net::HTTP::new('www.allocine.fr') #extrait le nom à partir du nom du fichier - /^.*?([^\/]*?)\.(.{3,4})$/ =~ @fichier + /^.*?([^\/]*?)\.(.{3,4})$/ =~ @fichiers[0] #remplace undescores et points par des espaces titre = $1.gsub(/[_\.]/, ' ') #remplace les suites d'espaces par un seul @@ -201,13 +335,19 @@ public donneesHtml = nil begin - reponse, donneesHtml = connexionHttp.get("/recherche/?motcle=#{CGI::escape(titre)}") + 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 + if /.*?Pas de résultats.*?/ =~ donneesHtml || ! donneesHtml.include?("