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

Contents of /trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs

Parent Directory Parent Directory | Revision Log Revision Log


Revision 171 - (show annotations) (download)
Sat Mar 16 14:31:24 2013 UTC (6 years, 5 months ago) by william
File size: 25925 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_list = xmltv_programs.FindAll(s => s.ChannelNumber == channel.channel_number && s.Start == start_date.ToDateTimeString());
319 IProgramDefinition xmltv_entry = null;
320 if (xmltv_entry_list.Count > 0)
321 {
322 gLog.Error.WriteLine("Found more than one entry: Matching channel='{0}' and start='{1}'", channel.channel_number, start_date.ToDateTimeString());
323 gLog.Error.WriteLine(" GB-PVR Program Data: oid='{0}' channel_oid='{1}' name='{2}' sub_title='{3}' description='{4}'", program.oid, program.channel_oid, program.name, program.sub_title, program.description);
324 if (Debugger.IsAttached)
325 {
326 Debugger.Break();
327 }
328 }
329 else
330 {
331 xmltv_entry = xmltv_entry_list.FirstOrDefault();
332 }
333
334 if (xmltv_entry == null)
335 {
336 gLog.Warn.WriteLine("Invalidating GB-PVR Program: oid='{0}' channel_oid='{1}' title='{2}' start='{3}'", program.oid, program.channel_oid, program.name, program.start_time.ToDateTimeString());
337 source_invalid.Add(program);
338 }
339 else
340 {
341 var updated_program = program;
342 updated_program.description = xmltv_entry.Description;
343 updated_program.end_time = DateTime.Parse(xmltv_entry.Stop);
344 updated_program.name = xmltv_entry.Title;
345 updated_program.start_time = DateTime.Parse(xmltv_entry.Start);
346 updated_program.sub_title = xmltv_entry.SubTitle;
347 gLog.Warn.WriteLine("Updating GB-PVR Program: oid='{0}' channel_oid='{1}' title='{2}' subtitle='{3}' start='{4}'", program.oid, program.channel_oid, program.name,program.sub_title, program.start_time.ToDateTimeString());
348 gLog.Warn.WriteLine(" Updated with Program: title='{0}' subtitle='{1}' start='{2}'", updated_program.name,updated_program.sub_title, program.start_time.ToDateTimeString());
349 source_valid.Add(updated_program);
350 }
351 index++;
352 }
353
354 //for (int i = 0; i < programs.Count(); i++)
355 //{
356 // var gbpvr_entry = gbpvr[i];
357 // var xmltv_entry = gbpvr_entry.AsXMLTVProgramDefinition(sqlite);
358 // if (!xmltv[i].Equals(xmltv_entry))
359 // {
360 // gLog.Warn.WriteLine("Warning GBPVR Program oid: {0} might be invalid", gbpvr_entry.oid);
361 // }
362 //}
363
364 source_valid = source_valid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
365 source_invalid = source_invalid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
366
367 gLog.Info.WriteLine("Total XMLTV Programs: 0x{0:x8}", xmltv_programs.Count);
368 gLog.Info.WriteLine("Updated: 0x{0:x8} GB-PVR Programs", source_valid.Count);
369 gLog.Info.WriteLine("Removed: 0x{0:x8} GB-PVR Programs", source_invalid.Count);
370 gLog.Info.WriteLine("Total GB-PVR Programs (Updated & Removed): 0x{0:x8}", source_valid.Count + source_invalid.Count);
371
372 return source_valid;
373 }
374 #endregion
375
376
377 private string CreateConnectionString()
378 {
379 string connection_string = string.Format("Data Source={0}", this.Database);
380 return connection_string;
381 }
382
383 private SQLiteConnection CreateConnection() { SQLiteConnection connection = new SQLiteConnection(CreateConnectionString()); return connection; }
384
385 [Conditional("SQLLITE_CONNECTION_TEST")]
386 private void ConnectionTest()
387 {
388 try
389 {
390 using (SQLiteConnection con = CreateConnection())
391 {
392 con.Open();
393 string command_text = string.Format("select * from {0};", TABLES.CHANNEL);
394 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
395 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
396 {
397 using (SQLiteDataReader r = cmd.ExecuteReader())
398 {
399 if (!r.HasRows)
400 {
401 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
402 }
403 else
404 {
405 while (r.Read())
406 {
407 for (int i = 0; i < r.FieldCount; i++)
408 {
409 string field_name = r.GetName(i);
410 Type field_type = r.GetFieldType(i);
411 object field_value = r.GetValue(i);
412 gLog.Verbose.Debug.WriteLine("Name: '{0}' Type: '{1}' Value: '{2}'", field_name, field_type.Name, field_value == null ? "null" : field_value.ToString());
413 }
414 break;
415 }
416 }
417 }
418 }
419 con.Clone();
420 }
421 OnCreatedInstance(this, new EventArgs());
422 }
423 catch (Exception ex)
424 {
425 gLog.Error.WriteLine(ex.ToString());
426 }
427 }
428
429
430 private bool CreateDatabaseBackup()
431 {
432 try
433 {
434 string backup_file = string.Format("{0}.{1}", this.Database, DateTime.Now.ToString("yyyyMMddHHmmss"));
435 gLog.Info.WriteLine("Creating Database backup...");
436 gLog.Info.WriteLine("\tSource: {0}", this.Database);
437 gLog.Info.WriteLine("\tDestination: {0}", backup_file);
438
439 System.IO.File.Copy(this.Database, backup_file);
440 return true;
441 }
442 catch (Exception ex)
443 {
444 gLog.Error.WriteLine(ex.ToString());
445 return false;
446 }
447 }
448 private void ReadChannelData()
449 {
450 try
451 {
452 List<ICHANNEL> channels = new List<ICHANNEL>();
453 using (SQLiteConnection con = CreateConnection())
454 {
455 con.Open();
456 string command_text = string.Format("select * from {0};", TABLES.CHANNEL);
457 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
458 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
459 {
460 using (SQLiteDataReader r = cmd.ExecuteReader())
461 {
462 if (!r.HasRows)
463 {
464 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
465 }
466 else
467 {
468 while (r.Read())
469 {
470 CHANNEL channel = new CHANNEL();
471 for (int i = 0; i < r.FieldCount; i++)
472 {
473 CHANNEL.Create(ref channel, r, i);
474 }
475 channels.Add(channel);
476 }
477 }
478 }
479 }
480 con.Clone();
481 }
482 this.Channels = channels;
483 }
484 catch (Exception ex)
485 {
486 gLog.Error.WriteLine(ex.ToString());
487 }
488 }
489 private void ReadProgrammeData()
490 {
491 try
492 {
493 List<IPROGRAMME> programs = new List<IPROGRAMME>();
494 using (SQLiteConnection con = CreateConnection())
495 {
496 con.Open();
497 string command_text = string.Format("select * from {0};", TABLES.PROGRAMME);
498 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
499 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
500 {
501 using (SQLiteDataReader r = cmd.ExecuteReader())
502 {
503 if (!r.HasRows)
504 {
505 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
506 }
507 else
508 {
509 while (r.Read())
510 {
511 PROGRAMME program = new PROGRAMME();
512 for (int i = 0; i < r.FieldCount; i++)
513 {
514 PROGRAMME.Create(ref program, r, i);
515 }
516 programs.Add(program);
517 }
518 }
519 }
520 }
521 con.Clone();
522 }
523 this.Programs = programs;
524 }
525 catch (Exception ex)
526 {
527 gLog.Error.WriteLine(ex.ToString());
528 }
529 }
530 private void ReadRecodringScheduleData()
531 {
532 try
533 {
534 List<IRECORDING_SCHEDULE> recordings = new List<IRECORDING_SCHEDULE>();
535 using (SQLiteConnection con = CreateConnection())
536 {
537 con.Open();
538 string command_text = string.Format("select * from {0};", TABLES.RECORDING_SCHEDULE);
539 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
540 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
541 {
542 using (SQLiteDataReader r = cmd.ExecuteReader())
543 {
544 if (!r.HasRows)
545 {
546 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
547 }
548 else
549 {
550 while (r.Read())
551 {
552 RECORDING_SCHEDULE recording = new RECORDING_SCHEDULE();
553 for (int i = 0; i < r.FieldCount; i++)
554 {
555 RECORDING_SCHEDULE.Create(ref recording, r, i);
556 }
557 recordings.Add(recording);
558 }
559 }
560 }
561 }
562 con.Clone();
563 }
564 this.Recordings = recordings;
565 }
566 catch (Exception ex)
567 {
568 gLog.Error.WriteLine(ex.ToString());
569 }
570 }
571
572 }
573 }

  ViewVC Help
Powered by ViewVC 1.1.22