309f68629491d598824cc5c7b99fd119a676b574
[pompage.git] / src / film.rb
1 # Représente un Film.
2 # Permet de charger des données depuis allocine.fr
3 # Permet d'effectuer des recherches depuis allocine.fr
4 # TODO : rendre la classe indépendant de la source de donnée pour pouvoir utiliser d'autres sites comme par exemple imdb.com
5
6 require 'pays.rb'
7 require 'genre.rb'
8 require 'personne.rb'
9 require 'constantes.rb'
10
11 require 'modules/allocine.rb'
12
13 # ajout de deux méthodes à la classe String
14 class String
15 def virerBalisesHTML
16 return gsub(/<(.*?)>/, '')
17 end
18 def virerBalisesHTML!
19 gsub!(/<(.*?)>/, '')
20 end
21 end
22
23 class Film
24
25 # toutes les données membres sont accessibles par défaut (écriture/lecture)
26 attr_accessor :id, :titre, :url, :fichiers, :mod, :annee, :realisateurs, :acteurs, :pays, :duree, :critiquePresse, :critiqueSpectateur, :genres, :synopsis, :budget
27
28 # Constructeur. N'entreprend aucune action (chargement), crée juste un film vide.
29 # p1 [String] : le fichier correspondant au film
30 def initialize(fichier, mod)
31 @fichiers = [fichier] # le chemin des fichiers est relatif au repertoire de base
32 @mod = mod
33
34 @id = 0
35 @titre = ''
36 @annee = nil
37 @realisateurs = []
38 @acteurs = []
39 @pays = []
40 @duree = nil
41 @critiquePresse = nil
42 @critiqueSpectateur = nil
43 @genres = []
44 @synopsis = nil
45 @budget = nil
46 @budgetUnite = 'euro'
47 @url
48
49 @nbReponses = 0
50
51 # mémorise les tuples {nom => id} dans le cas ou il y a plusieurs choix de films après une recherche
52 @choix = {}
53 end
54
55 # Est-ce qu'il y a eu plusieurs réponses pour ce film lors de la cherche sur le net ?
56 def plusieursReponses?
57 return @nbReponses > 1
58 end
59
60 # Est-ce qu'il y a eu plusieurs réponses pour ce film lors de la cherche sur le net ?
61 def nbReponses
62 return @nbReponses
63 end
64
65 # Demande à l'utilisateur de faire un choix.
66 # ret : 1 si le conflit à été résolu, 2 si le film est à ignorer, 3 si tout les films sont à ignorer
67 def reglerConflitPlusieursReponses
68 @nbReponses = 1
69
70 puts " -> " + @fichiers[0]
71 puts "Fais ton choix jeune padawan (un caractère et pas plus)"
72 tabNoms = @choix.keys
73 choix = 1
74 loop do
75 i = 1
76 tabNoms.each{|n|
77 puts "#{i}. #{n}"
78 i += 1
79 }
80 puts "A. Passer et l'ajouter"
81 puts "B. Ignorer"
82 puts "C. Tout ignorer"
83 choix = STDIN.gets
84
85 if /A/i =~ choix
86 return 1
87 elsif /B/i =~ choix
88 return 2
89 elsif /C/i =~ choix
90 return 3
91 end
92
93 choix = choix.to_i
94 if choix > 0 && choix <= tabNoms.length
95 break;
96 else
97 puts
98 puts "Choix pas bon !!"
99 end
100 end
101
102 @mod.load(@choix[tabNoms[choix-1]], self)
103
104 return 1
105 end
106
107 # Ajoute un fichier comme faisant partie du film
108 def addFichier(fichier)
109 if !@fichiers.include?(fichier)
110 @fichiers << fichier
111 end
112 end
113
114 # Charge les informations du films à partir d'allocine.fr
115 # ret [Film]
116 def loadData
117 #extrait le nom à partir du nom du fichier
118 /^.*?([^\/]*?)\.(.{3,4})$/ =~ @fichiers[0]
119 #remplace undescores et points par des espaces
120 titre = $1.gsub(/[_\.]/, ' ')
121 #remplace les suites d'espaces par un seul
122 titre.gsub!(/ {2,}/,' ')
123 titre.gsub!(/\[.*?\]/,'')
124 titre.gsub!(/\(.*?\)/,'')
125 titre.gsub!(/\{.*?\}/,'')
126 #vire les espaces au début et à la fin
127 titre.strip!
128
129 @titre = titre.dup
130
131 unless LOAD_DATA
132 return self
133 end
134
135 reponses = @mod.rechercherFilm(titre)
136 @nbReponses = reponses.size
137
138 if @nbReponses == 1
139 @mod.load(reponses.values[0], self)
140 else
141 @choix = reponses
142 end
143
144 self
145 end
146
147 # Renvoie un film sous la forme d'un élément XML .
148 # ret [REXML::Element] : un element xml <film>
149 def getXml
150
151 racine = REXML::Element::new('film')
152 racine.add_attribute('id', @id.to_s)
153
154 fichiers = REXML::Element::new('fichiers')
155 @fichiers.each{|f|
156 fichiers.add(REXML::Element::new('fichier').add_text(f))
157 }
158 racine.add(fichiers)
159
160 racine.add(REXML::Element::new('titre').add_text(@titre))
161 racine.add(REXML::Element::new('annee').add_text(@annee))
162
163 realisateurs = REXML::Element::new('realisateurs')
164 @realisateurs.each{|r|
165 realisateurs.add(REXML::Element::new('realisateur').add_text(r.nom))
166 }
167 racine.add(realisateurs)
168
169 acteurs = REXML::Element::new('acteurs')
170 @acteurs.each{|a|
171 acteurs.add(REXML::Element::new('acteur').add_text(a.nom))
172 }
173 racine.add(acteurs)
174
175 lespays = REXML::Element::new('lespays')
176 @pays.each{|p|
177 lespays.add(REXML::Element::new('pays').add_text(p.nom))
178 }
179 racine.add(lespays)
180
181 racine.add(REXML::Element::new('duree').add_text(@duree.to_s))
182
183 racine.add(REXML::Element::new('critiquePresse').add_text(@critiquePresse))
184 racine.add(REXML::Element::new('critiqueSpectateur').add_text(@critiqueSpectateur))
185
186 genres = REXML::Element::new('genres')
187 @genres.each{|g|
188 genres.add(REXML::Element::new('genre').add_text(g.nom))
189 }
190 racine.add(genres)
191
192 synopsisElement = REXML::Element::new('synopsis')
193 unless @synopsis.nil?
194 @synopsis.split("\n").each{|s|
195 next if s =~ /^\s*$/
196 synopsisElement.add(REXML::Element::new('p').add_text(s))
197 }
198 end
199 racine.add(synopsisElement)
200
201 budgetElement = REXML::Element::new('budget')
202 budgetElement.add_text(@budget)
203 budgetElement.add_attribute('unite', @budgetUnite)
204 racine.add(budgetElement)
205
206 racine.add(REXML::Element::new('url').add_text(@url))
207
208 racine
209 end
210 end
211