git-svn-id: svn://euphorik.ch/pompage@35 02bbb61a-6d21-0410-aba0-cb053bdfd66a
[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 def Film::setModule(m)
26 @@module = m
27 end
28
29 # toutes les données membres sont accessibles par défaut (écriture/lecture)
30 attr_accessor :id, :titre, :url, :fichiers, :annee, :realisateurs, :acteurs, :pays, :duree, :critiquePresse, :critiqueSpectateur, :genres, :synopsis, :budget
31
32 # Constructeur. N'entreprend aucune action (chargement), crée juste un film vide.
33 # p1 [String] : le fichier correspondant au film
34 def initialize(fichier)
35 @fichiers = [fichier] # le chemin des fichiers est relatif au repertoire de base
36
37 @id = 0
38 @titre = ''
39 @annee = nil
40 @realisateurs = []
41 @acteurs = []
42 @pays = []
43 @duree = nil
44 @critiquePresse = nil
45 @critiqueSpectateur = nil
46 @genres = []
47 @synopsis = nil
48 @budget = nil
49 @budgetUnite = 'euro'
50 @url
51
52 @nbReponses = 0
53
54 # mémorise les tuples {nom => id} dans le cas ou il y a plusieurs choix de films après une recherche
55 @choix = {}
56 end
57
58 # Est-ce qu'il y a eu plusieurs réponses pour ce film lors de la cherche sur le net ?
59 def plusieursReponses?
60 return @nbReponses > 1
61 end
62
63 # Est-ce qu'il y a eu plusieurs réponses pour ce film lors de la cherche sur le net ?
64 def nbReponses
65 return @nbReponses
66 end
67
68 # Demande à l'utilisateur de faire un choix.
69 # ret : true si le conflit à été résolu sinon false
70 def reglerConflitPlusieursReponses
71 @nbReponses = 1
72
73 puts " -> " + @fichiers[0]
74 puts "Fais ton choix jeune padawan (un caractère et pas plus)"
75 tabNoms = @choix.keys
76 choix = 1
77 loop do
78 i = 1
79 tabNoms.each{|n|
80 puts "#{i}. #{n}"
81 i += 1
82 }
83 puts "A. Passer et l'ajouter"
84 puts "B. Ignorer"
85 choix = STDIN.gets
86
87 if /A/i =~ choix
88 return true
89 elsif /B/i =~ choix
90 return false
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 @@module.load(@choix[tabNoms[choix-1]], self)
103
104 return true
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 = @@module.rechercherFilm(titre)
136 @nbReponses = reponses.size
137
138 if @nbReponses == 1
139 @@module.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