X-Git-Url: http://git.euphorik.ch/?a=blobdiff_plain;f=src%2Ffilm.rb;h=b56ad072cd78573f0fdb122fe065df198d608f1b;hb=f2d0f55dc2e3dc77561c26703e382b1917498242;hp=1eaa33f9ed97f307d68e3098dcd58cbad26602b9;hpb=44c2f215b80fe438522f2daaa0bd33f1a60c619f;p=pompage.git

diff --git a/src/film.rb b/src/film.rb
index 1eaa33f..b56ad07 100644
--- a/src/film.rb
+++ b/src/film.rb
@@ -1,204 +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'
 
-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
+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
-      
-      # 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()
-       
-         @@idDisponible = id + 1         
-         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
-      }
+# ajout de deux méthodes à la classe String
+class String
+   def virerBalisesHTML
+      return gsub(/<(.*?)>/, '')
    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 virerBalisesHTML!
+      gsub!(/<(.*?)>/, '')
    end
-   
-private
+end
 
-   def Film::filmsFactory(fichier)
-      /^.*?\.(.{3,4})$/ =~ fichier
-      if FILMS_EXTENSIONS.include?($1)
-         #convertit le code latin-1 en UTF8
-         Film::new(fichier.unpack("C*").pack("U*")).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.fichiers[0]})"
-               next
-            end
-            
-            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
+class Film
    
-   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.fichier})"
-         end
-      else
-         puts "[i] movie added : #{film.titre} (#{film.fichiers[0]})"
-         @@films[film.titre] = film
-         @@filmsFichier[film.fichiers[0]] = film
-      end
+   def Film::setModule(m)
+      @@module = m
    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)
-      @fichiers = [fichier]
+      @fichiers = [fichier] # le chemin des fichiers est relatif au repertoire de base
            
       @id = 0
       @titre = ''
@@ -214,44 +48,72 @@ 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
-      puts "prout"
+      @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
       
-      return false
+      @@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))
+      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 = @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
@@ -266,114 +128,24 @@ public
       
       @titre = titre.dup
       
-      donneesHtml = nil
-      begin
-         begin
-            reponse, donneesHtml = connexionHttp.get("/recherche/?motcle=#{CGI::escape(titre)}")
-         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?("<h3><b>Films <h4>")
-            /(.*?)[^ ]+?$/ =~ 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?
-            
-         #/<a href="\/film\/fichefilm_gen_cfilm=(\d+)\.html" class="link1">/ =~ donneesHtml
-         r = donneesHtml.scan(/<a href="\/film\/fichefilm_gen_cfilm=(\d+)\.html" class="link1">(.*?)<\/a>/)
-         
-         if r.length > 1
-            @aPlusieursReponses = true
-            r.each{|f|
-               @idsAllocine[f[1].gsub(/<(.*?)>/, '')] = 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)
-      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>(.*?)<\/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
-      /<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 (.*?)\.&nbsp;<\/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')
@@ -386,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|
@@ -408,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|
@@ -417,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)