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

Contents of /trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs

Parent Directory Parent Directory | Revision Log Revision Log


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

  ViewVC Help
Powered by ViewVC 1.1.22