/[xmltv_parser]/trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs
ViewVC logotype

Contents of /trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs

Parent Directory Parent Directory | Revision Log Revision Log


Revision 169 - (show annotations) (download)
Sat Mar 16 14:04:31 2013 UTC (6 years, 5 months ago) by william
File size: 24444 byte(s)

1 //#define SQLLITE_CONNECTION_TEST
2 using System;
3 using System.Collections.Generic;
4 using System.Linq;
5 using System.Text;
6 using System.Data.SQLite;
7 using System.Diagnostics;
8 using Enterprise.Logging;
9 using libxmltv;
10 using libxmltv.Interfaces;
11 using libxmltv.Core;
12 namespace GBPVRProgramDatabaseFixer
13 {
14 public interface ISQLLITE
15 {
16 List<SQLLITE.IPROGRAMME> Programs { get; }
17 List<SQLLITE.IRECORDING_SCHEDULE> Recordings { get; }
18 List<SQLLITE.ICHANNEL> Channels { get; }
19 IDateTimeRange GetProgramsDateRange(List<SQLLITE.IPROGRAMME> programs);
20 List<SQLLITE.IPROGRAMME> CreateUpdatedProgramsList(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> removed_programs);
21
22 }
23 public class SQLLITE : ISQLLITE
24 {
25 public static ISQLLITE Create(string database, EventHandler<EventArgs> OnInstanceCreated)
26 {
27 return new SQLLITE(database, OnInstanceCreated);
28 }
29
30 #region DATABASE DEFINITIONS
31
32 public interface ICHANNEL
33 {
34 Int64 oid { get; }
35 String name { get; }
36 String channelID { get; }
37 Int64 channel_number { get; }
38 String favourite_channel { get; }
39 String display_name { get; }
40 }
41 private class CHANNEL : ICHANNEL
42 {
43 public CHANNEL()
44 {
45 BaseDatabaseDefinition<CHANNEL>.CreateDefault(this);
46 }
47 //public RECORDING_SCHEDULE(SQLiteDataReader r, int index) { BaseDatabaseDefinition<RECORDING_SCHEDULE>.Create(this, r, index); }
48
49 public static void Create(ref CHANNEL instance, SQLiteDataReader r, int index)
50 {
51 BaseDatabaseDefinition<CHANNEL>.Create(ref instance, r, index);
52 }
53 #region ICHANNEL members
54 public Int64 oid { get; set; }
55 public String name { get; set; }
56 public String channelID { get; set; }
57 public Int64 channel_number { get; set; }
58 public String favourite_channel { get; set; }
59 public String display_name { get; set; }
60 #endregion
61 }
62 public interface IRECORDING_SCHEDULE
63 {
64 Int64 oid { get; }
65 Int64 programme_oid { get; }
66 Int64 capture_source_oid { get; }
67 Int16 status { get; }
68 String filename { get; }
69 Int64 recording_type { get; }
70 Int64 recording_group { get; }
71 DateTime manual_start_time { get; }
72 DateTime manual_end_time { get; }
73 Int64 manual_channel_oid { get; }
74 Int64 quality_level { get; }
75 Int64 pre_pad_minutes { get; }
76 Int64 post_pad_minutes { get;}
77 Int32 priority { get; }
78 String conversion_profile { get; }
79 }
80
81 private static class BaseDatabaseDefinition<T>
82 {
83 public static void CreateDefault(T instance)
84 {
85 try
86 {
87 Type t = typeof(T);
88 var props = t.GetProperties();
89 foreach (var prop in props)
90 {
91 Type prop_type = prop.PropertyType;
92 object field_value = null;
93 try
94 {
95 if (prop_type == typeof(string))
96 {
97 field_value = string.Empty;
98 }
99 else
100 {
101 field_value = Activator.CreateInstance(prop_type);
102 }
103 }
104 catch (Exception ex)
105 {
106 throw ex;
107 }
108 prop.SetValue(instance, field_value, null);
109 }
110 }
111 catch (Exception ex)
112 {
113 throw ex;
114 }
115 }
116 public static void Create(ref T instance, SQLiteDataReader r, int index)
117 {
118 string field_name = r.GetName(index);
119 Type field_type = r.GetFieldType(index);
120 object field_value = r.GetValue(index);
121 //gLog.Verbose.Debug.WriteLine("Name: '{0}' Type: '{1}' Value: '{2}'", field_name, field_type.Name, field_value == null ? "null" : field_value.ToString());
122
123 Type t = typeof(T);
124 var props = t.GetProperties();
125 foreach (var prop in props)
126 {
127 if (prop.Name.ToLower() == field_name.ToLower())
128 {
129 if (prop.PropertyType == field_type)
130 {
131 Type db_type = field_value.GetType();
132 try
133 {
134 if (db_type == typeof(System.DBNull))
135 {
136 prop.SetValue(instance, null, null);
137 }
138 else
139 {
140 prop.SetValue(instance, field_value, null);
141 }
142
143 }
144 catch (Exception ex)
145 {
146 throw ex;
147 }
148 }
149 else
150 {
151 gLog.Verbose.Debug.WriteLine("Found Property: {0} but there was a type mismatch. Property Type: '{1}' Expected: '{2}'", prop.Name, prop.PropertyType.Name, field_type.Name);
152 throw new InvalidOperationException(string.Format("Found Property: {0} but there was a type mismatch. Property Type: '{1}' Expected: '{2}'", prop.Name, prop.PropertyType.Name, field_type.Name));
153 }
154 }
155 }
156
157 }
158 }
159
160 private class RECORDING_SCHEDULE : IRECORDING_SCHEDULE
161 {
162 public RECORDING_SCHEDULE()
163 {
164 BaseDatabaseDefinition<RECORDING_SCHEDULE>.CreateDefault(this);
165 }
166 //public RECORDING_SCHEDULE(SQLiteDataReader r, int index) { BaseDatabaseDefinition<RECORDING_SCHEDULE>.Create(this, r, index); }
167
168 public static void Create(ref RECORDING_SCHEDULE instance, SQLiteDataReader r, int index)
169 {
170 BaseDatabaseDefinition<RECORDING_SCHEDULE>.Create(ref instance, r, index);
171 }
172
173 #region IRECORDING_SCHEDULE members
174 public Int64 oid { get; set;}
175 public Int64 programme_oid { get; set;}
176 public Int64 capture_source_oid { get; set;}
177 public Int16 status { get; set; }
178 public String filename { get; set; }
179 public Int64 recording_type { get;set; }
180 public Int64 recording_group { get; set;}
181 public DateTime manual_start_time { get; set;}
182 public DateTime manual_end_time { get; set; }
183 public Int64 manual_channel_oid { get; set; }
184 public Int64 quality_level { get; set; }
185 public Int64 pre_pad_minutes { get; set; }
186 public Int64 post_pad_minutes { get; set; }
187 public Int32 priority { get; set; }
188 public String conversion_profile { get; set; }
189 #endregion
190 }
191
192 public interface IPROGRAMME
193 {
194 Int64 oid { get; set; }
195 String name { get; set; }
196 String sub_title { get; set; }
197 String description { get; set; }
198 DateTime start_time { get; set; }
199 DateTime end_time { get; set; }
200 Int64 channel_oid { get; set; }
201 String unique_identifier { get; set; }
202 String rating { get; set; }
203
204 IProgramDefinition AsXMLTVProgramDefinition(ISQLLITE sqllite);
205 }
206 private class PROGRAMME : IPROGRAMME
207 {
208 public PROGRAMME()
209 {
210 BaseDatabaseDefinition<PROGRAMME>.CreateDefault(this);
211 }
212 //public PROGRAMME(SQLiteDataReader r, int index) : base(r, index) { }
213 public static void Create(ref PROGRAMME instance, SQLiteDataReader r, int index)
214 {
215 BaseDatabaseDefinition<PROGRAMME>.Create(ref instance, r, index);
216 }
217 #region IPROGRAMME members
218 public Int64 oid { get; set; }
219 public String name { get; set; }
220 public String sub_title { get; set; }
221 public String description { get; set; }
222 public DateTime start_time { get; set; }
223 public DateTime end_time { get; set; }
224 public Int64 channel_oid { get; set; }
225 public String unique_identifier { get; set; }
226 public String rating { get; set; }
227 public IProgramDefinition AsXMLTVProgramDefinition(ISQLLITE sqllite)
228 {
229 ProgramList.ProgramDefintion definition = new ProgramList.ProgramDefintion();
230 var channel = sqllite.Channels.Find(s => s.oid == this.channel_oid);
231 definition.ChannelName = channel.display_name;
232 definition.ChannelNumber = (int)channel.channel_number;
233 definition.Description = this.description;
234 definition.Start = this.start_time.ToDateTimeString();
235 definition.Stop = this.end_time.ToDateTimeString();
236 definition.SubTitle = this.sub_title;
237 definition.Title = this.name;
238 return definition;
239 }
240 #endregion
241 }
242 #endregion
243 private static class TABLES
244 {
245 public const string RECORDING_SCHEDULE = "RECORDING_SCHEDULE";
246 public const string PROGRAMME = "PROGRAMME";
247 public const string CHANNEL = "CHANNEL";
248 }
249
250 //public SQLLite() { }
251 protected SQLLITE(string database, EventHandler<EventArgs> OnInstanceCreated)
252 {
253 this.OnInstanceCreated = OnInstanceCreated;
254 //CreateConnection(database);
255 this.Database = database;
256 if (!CreateDatabaseBackup())
257 {
258 gLog.Error.WriteLine("Failed to backup database.");
259 return;
260 }
261 ConnectionTest();
262 ReadChannelData();
263 ReadRecodringScheduleData();
264 ReadProgrammeData();
265 OnCreatedInstance(this, new EventArgs());
266 }
267
268 [NonSerialized]
269 private EventHandler<EventArgs> _OnInstanceCreated;
270 private EventHandler<EventArgs> OnInstanceCreated { get { return _OnInstanceCreated; } set { _OnInstanceCreated = value; } }
271
272 private void OnCreatedInstance(object sender, EventArgs e)
273 {
274 if (OnInstanceCreated != null) { OnInstanceCreated.Invoke(sender, e); }
275 }
276
277 private string Database;
278 #region ISQLLITE members
279 public List<IPROGRAMME> Programs { get; private set; }
280 public List<IRECORDING_SCHEDULE> Recordings { get; private set; }
281 public List<ICHANNEL> Channels { get; private set; }
282
283 public IDateTimeRange GetProgramsDateRange(List<SQLLITE.IPROGRAMME> programs)
284 {
285 var list = new List<IPROGRAMME>(programs.ToArray());
286 DateTime first = new DateTime();
287 DateTime last = new DateTime();
288 first = list.OrderBy(s => s.start_time).ToList().First().start_time;
289 last = list.OrderBy(s => s.start_time).ToList().Last().start_time;
290 gLog.Verbose.Debug.WriteLine("\tFirst: {0} = ({1})", first.ToString("yyyy/MM/dd HH:mm:ss.fffffff"), first.ToDateTimeString());
291 gLog.Verbose.Debug.WriteLine("\tLast: {0} = ({1})", last.ToString("yyyy/MM/dd HH:mm:ss.fffffff"), last.ToDateTimeString());
292 var range = DateTimeRange.Create(first, last);
293 return range;
294 }
295
296 public List<SQLLITE.IPROGRAMME> CreateUpdatedProgramsList(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> source_invalid)
297 {
298 source_invalid = new List<IPROGRAMME>();
299 gbpvr_programs = gbpvr_programs.OrderBy(s => s.start_time).ToList();
300 List<SQLLITE.IPROGRAMME> source_valid = new List<IPROGRAMME>();
301 //if (range == null)
302 //{
303 // gLog.Warn.WriteLine("The DateTimeRange passed in is null...returning the original program list");
304 // return list;
305 //}
306 //gLog.Warn.WriteLine("FilterProgramsByDateRange has not been implemented");
307
308 double total = gbpvr_programs.Count;
309 double index = 0;
310 double progress = 0;
311 foreach (var program in gbpvr_programs)
312 {
313 progress = 100.0 * (index / total);
314 gLog.ReportProgress(this, new ReportProgressEventArgs((int)progress, string.Format("Filtering GBPVR Programs: {0:00}%", (int)progress)));
315 var channel_oid = program.channel_oid;
316 var channel = this.Channels.Find(s => s.oid == channel_oid);
317 var start_date = program.start_time;
318 var xmltv_entry = xmltv_programs.Find(s => s.ChannelNumber == channel.channel_number && s.Start == start_date.ToDateTimeString());
319 if (xmltv_entry == null)
320 {
321 source_invalid.Add(program);
322 }
323 else
324 {
325 var updated_program = program;
326 updated_program.description = xmltv_entry.Description;
327 updated_program.end_time = DateTime.Parse(xmltv_entry.Stop);
328 updated_program.name = xmltv_entry.Title;
329 updated_program.start_time = DateTime.Parse(xmltv_entry.Start);
330 updated_program.sub_title = xmltv_entry.SubTitle;
331 source_valid.Add(updated_program);
332 }
333 index++;
334 }
335
336 //for (int i = 0; i < programs.Count(); i++)
337 //{
338 // var gbpvr_entry = gbpvr[i];
339 // var xmltv_entry = gbpvr_entry.AsXMLTVProgramDefinition(sqlite);
340 // if (!xmltv[i].Equals(xmltv_entry))
341 // {
342 // gLog.Warn.WriteLine("Warning GBPVR Program oid: {0} might be invalid", gbpvr_entry.oid);
343 // }
344 //}
345
346 source_valid = source_valid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
347 source_invalid = source_invalid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
348
349 gLog.Info.WriteLine("Total XMLTV Programs: 0x{0:x8}", xmltv_programs.Count);
350 gLog.Info.WriteLine("Updated: 0x{0:x8} GB-PVR Programs", source_valid.Count);
351 gLog.Info.WriteLine("Removed: 0x{0:x8} GB-PVR Programs", source_invalid.Count);
352 gLog.Info.WriteLine("Total GB-PVR Programs (Updated & Removed): 0x{0:x8}", source_valid.Count + source_invalid.Count);
353
354 return source_valid;
355 }
356 #endregion
357
358
359 private string CreateConnectionString()
360 {
361 string connection_string = string.Format("Data Source={0}", this.Database);
362 return connection_string;
363 }
364
365 private SQLiteConnection CreateConnection() { SQLiteConnection connection = new SQLiteConnection(CreateConnectionString()); return connection; }
366
367 [Conditional("SQLLITE_CONNECTION_TEST")]
368 private void ConnectionTest()
369 {
370 try
371 {
372 using (SQLiteConnection con = CreateConnection())
373 {
374 con.Open();
375 string command_text = string.Format("select * from {0};", TABLES.CHANNEL);
376 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
377 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
378 {
379 using (SQLiteDataReader r = cmd.ExecuteReader())
380 {
381 if (!r.HasRows)
382 {
383 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
384 }
385 else
386 {
387 while (r.Read())
388 {
389 for (int i = 0; i < r.FieldCount; i++)
390 {
391 string field_name = r.GetName(i);
392 Type field_type = r.GetFieldType(i);
393 object field_value = r.GetValue(i);
394 gLog.Verbose.Debug.WriteLine("Name: '{0}' Type: '{1}' Value: '{2}'", field_name, field_type.Name, field_value == null ? "null" : field_value.ToString());
395 }
396 break;
397 }
398 }
399 }
400 }
401 con.Clone();
402 }
403 OnCreatedInstance(this, new EventArgs());
404 }
405 catch (Exception ex)
406 {
407 gLog.Error.WriteLine(ex.ToString());
408 }
409 }
410
411
412 private bool CreateDatabaseBackup()
413 {
414 try
415 {
416 string backup_file = string.Format("{0}.{1}", this.Database, DateTime.Now.ToString("yyyyMMddHHmmss"));
417 gLog.Info.WriteLine("Creating Database backup...");
418 gLog.Info.WriteLine("\tSource: {0}", this.Database);
419 gLog.Info.WriteLine("\tDestination: {0}", backup_file);
420
421 System.IO.File.Copy(this.Database, backup_file);
422 return true;
423 }
424 catch (Exception ex)
425 {
426 gLog.Error.WriteLine(ex.ToString());
427 return false;
428 }
429 }
430 private void ReadChannelData()
431 {
432 try
433 {
434 List<ICHANNEL> channels = new List<ICHANNEL>();
435 using (SQLiteConnection con = CreateConnection())
436 {
437 con.Open();
438 string command_text = string.Format("select * from {0};", TABLES.CHANNEL);
439 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
440 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
441 {
442 using (SQLiteDataReader r = cmd.ExecuteReader())
443 {
444 if (!r.HasRows)
445 {
446 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
447 }
448 else
449 {
450 while (r.Read())
451 {
452 CHANNEL channel = new CHANNEL();
453 for (int i = 0; i < r.FieldCount; i++)
454 {
455 CHANNEL.Create(ref channel, r, i);
456 }
457 channels.Add(channel);
458 }
459 }
460 }
461 }
462 con.Clone();
463 }
464 this.Channels = channels;
465 }
466 catch (Exception ex)
467 {
468 gLog.Error.WriteLine(ex.ToString());
469 }
470 }
471 private void ReadProgrammeData()
472 {
473 try
474 {
475 List<IPROGRAMME> programs = new List<IPROGRAMME>();
476 using (SQLiteConnection con = CreateConnection())
477 {
478 con.Open();
479 string command_text = string.Format("select * from {0};", TABLES.PROGRAMME);
480 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
481 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
482 {
483 using (SQLiteDataReader r = cmd.ExecuteReader())
484 {
485 if (!r.HasRows)
486 {
487 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
488 }
489 else
490 {
491 while (r.Read())
492 {
493 PROGRAMME program = new PROGRAMME();
494 for (int i = 0; i < r.FieldCount; i++)
495 {
496 PROGRAMME.Create(ref program, r, i);
497 }
498 programs.Add(program);
499 }
500 }
501 }
502 }
503 con.Clone();
504 }
505 this.Programs = programs;
506 }
507 catch (Exception ex)
508 {
509 gLog.Error.WriteLine(ex.ToString());
510 }
511 }
512 private void ReadRecodringScheduleData()
513 {
514 try
515 {
516 List<IRECORDING_SCHEDULE> recordings = new List<IRECORDING_SCHEDULE>();
517 using (SQLiteConnection con = CreateConnection())
518 {
519 con.Open();
520 string command_text = string.Format("select * from {0};", TABLES.RECORDING_SCHEDULE);
521 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
522 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
523 {
524 using (SQLiteDataReader r = cmd.ExecuteReader())
525 {
526 if (!r.HasRows)
527 {
528 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
529 }
530 else
531 {
532 while (r.Read())
533 {
534 RECORDING_SCHEDULE recording = new RECORDING_SCHEDULE();
535 for (int i = 0; i < r.FieldCount; i++)
536 {
537 RECORDING_SCHEDULE.Create(ref recording, r, i);
538 }
539 recordings.Add(recording);
540 }
541 }
542 }
543 }
544 con.Clone();
545 }
546 this.Recordings = recordings;
547 }
548 catch (Exception ex)
549 {
550 gLog.Error.WriteLine(ex.ToString());
551 }
552 }
553
554 }
555 }

  ViewVC Help
Powered by ViewVC 1.1.22