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

Contents of /trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs

Parent Directory Parent Directory | Revision Log Revision Log


Revision 167 - (show annotations) (download)
Sat Mar 16 13:26:09 2013 UTC (6 years, 10 months ago) by william
File size: 24015 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> GetValidPrograms(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; }
195 String name { get; }
196 String sub_title { get; }
197 String description { get; }
198 DateTime start_time { get; }
199 DateTime end_time { get; }
200 Int64 channel_oid { get; }
201 String unique_identifier { get; }
202 String rating { get; }
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.ToString("");
235 definition.Stop = this.end_time.ToString("");
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> GetValidPrograms(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 List<SQLLITE.IPROGRAMME> source = new List<IPROGRAMME>(gbpvr_programs.ToArray());
300 source = source.OrderBy(s => s.start_time).ToList();
301 List<SQLLITE.IPROGRAMME> source_valid = new List<IPROGRAMME>();
302 //if (range == null)
303 //{
304 // gLog.Warn.WriteLine("The DateTimeRange passed in is null...returning the original program list");
305 // return list;
306 //}
307 //gLog.Warn.WriteLine("FilterProgramsByDateRange has not been implemented");
308
309 double total = source.Count;
310 double index = 0;
311 double progress = 0;
312 foreach (var program in source)
313 {
314 //if (program.start_time >= range.Start &&
315 // program.start_time <= range.End)
316 //{
317 // list2.Add(program);
318 //}
319 //else
320 //{
321 // removed_programs.Add(program);
322 //}
323 progress = 100.0 * (index / total);
324 gLog.ReportProgress(this, new ReportProgressEventArgs((int)progress, string.Format("Filtering GBPVR Programs: {0:00}%", (int)progress)));
325 var xmltv_entry = program.AsXMLTVProgramDefinition(this);
326 if (xmltv_programs.Contains(xmltv_entry))
327 {
328 source_valid.Add(program);
329 }
330 else
331 {
332 source_invalid.Add(program);
333 }
334 index++;
335 }
336
337 //for (int i = 0; i < programs.Count(); i++)
338 //{
339 // var gbpvr_entry = gbpvr[i];
340 // var xmltv_entry = gbpvr_entry.AsXMLTVProgramDefinition(sqlite);
341 // if (!xmltv[i].Equals(xmltv_entry))
342 // {
343 // gLog.Warn.WriteLine("Warning GBPVR Program oid: {0} might be invalid", gbpvr_entry.oid);
344 // }
345 //}
346
347 source_valid = source_valid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
348 source_invalid = source_invalid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
349
350 gLog.Info.WriteLine("Total XMLTV Programs: 0x{0:x8}", xmltv_programs.Count);
351 gLog.Info.WriteLine("Found: 0x{0:x8} valid GB-PVR Programs", source_valid.Count);
352 gLog.Info.WriteLine("Found: 0x{0:x8} invalid GB-PVR Programs", source_invalid.Count);
353 gLog.Info.WriteLine("Total GB-PVR Programs: 0x{0:x8}", source_valid.Count + source_invalid.Count);
354
355 return source_valid;
356 }
357 #endregion
358
359
360 private string CreateConnectionString()
361 {
362 string connection_string = string.Format("Data Source={0}", this.Database);
363 return connection_string;
364 }
365
366 private SQLiteConnection CreateConnection() { SQLiteConnection connection = new SQLiteConnection(CreateConnectionString()); return connection; }
367
368 [Conditional("SQLLITE_CONNECTION_TEST")]
369 private void ConnectionTest()
370 {
371 try
372 {
373 using (SQLiteConnection con = CreateConnection())
374 {
375 con.Open();
376 string command_text = string.Format("select * from {0};", TABLES.CHANNEL);
377 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
378 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
379 {
380 using (SQLiteDataReader r = cmd.ExecuteReader())
381 {
382 if (!r.HasRows)
383 {
384 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
385 }
386 else
387 {
388 while (r.Read())
389 {
390 for (int i = 0; i < r.FieldCount; i++)
391 {
392 string field_name = r.GetName(i);
393 Type field_type = r.GetFieldType(i);
394 object field_value = r.GetValue(i);
395 gLog.Verbose.Debug.WriteLine("Name: '{0}' Type: '{1}' Value: '{2}'", field_name, field_type.Name, field_value == null ? "null" : field_value.ToString());
396 }
397 break;
398 }
399 }
400 }
401 }
402 con.Clone();
403 }
404 OnCreatedInstance(this, new EventArgs());
405 }
406 catch (Exception ex)
407 {
408 gLog.Error.WriteLine(ex.ToString());
409 }
410 }
411
412
413 private bool CreateDatabaseBackup()
414 {
415 try
416 {
417 string backup_file = string.Format("{0}.{1}", this.Database, DateTime.Now.ToString("yyyyMMddHHmmss"));
418 gLog.Info.WriteLine("Creating Database backup...");
419 gLog.Info.WriteLine("\tSource: {0}", this.Database);
420 gLog.Info.WriteLine("\tDestination: {0}", backup_file);
421
422 System.IO.File.Copy(this.Database, backup_file);
423 return true;
424 }
425 catch (Exception ex)
426 {
427 gLog.Error.WriteLine(ex.ToString());
428 return false;
429 }
430 }
431 private void ReadChannelData()
432 {
433 try
434 {
435 List<ICHANNEL> channels = new List<ICHANNEL>();
436 using (SQLiteConnection con = CreateConnection())
437 {
438 con.Open();
439 string command_text = string.Format("select * from {0};", TABLES.CHANNEL);
440 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
441 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
442 {
443 using (SQLiteDataReader r = cmd.ExecuteReader())
444 {
445 if (!r.HasRows)
446 {
447 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
448 }
449 else
450 {
451 while (r.Read())
452 {
453 CHANNEL channel = new CHANNEL();
454 for (int i = 0; i < r.FieldCount; i++)
455 {
456 CHANNEL.Create(ref channel, r, i);
457 }
458 channels.Add(channel);
459 }
460 }
461 }
462 }
463 con.Clone();
464 }
465 this.Channels = channels;
466 }
467 catch (Exception ex)
468 {
469 gLog.Error.WriteLine(ex.ToString());
470 }
471 }
472 private void ReadProgrammeData()
473 {
474 try
475 {
476 List<IPROGRAMME> programs = new List<IPROGRAMME>();
477 using (SQLiteConnection con = CreateConnection())
478 {
479 con.Open();
480 string command_text = string.Format("select * from {0};", TABLES.PROGRAMME);
481 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
482 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
483 {
484 using (SQLiteDataReader r = cmd.ExecuteReader())
485 {
486 if (!r.HasRows)
487 {
488 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
489 }
490 else
491 {
492 while (r.Read())
493 {
494 PROGRAMME program = new PROGRAMME();
495 for (int i = 0; i < r.FieldCount; i++)
496 {
497 PROGRAMME.Create(ref program, r, i);
498 }
499 programs.Add(program);
500 }
501 }
502 }
503 }
504 con.Clone();
505 }
506 this.Programs = programs;
507 }
508 catch (Exception ex)
509 {
510 gLog.Error.WriteLine(ex.ToString());
511 }
512 }
513 private void ReadRecodringScheduleData()
514 {
515 try
516 {
517 List<IRECORDING_SCHEDULE> recordings = new List<IRECORDING_SCHEDULE>();
518 using (SQLiteConnection con = CreateConnection())
519 {
520 con.Open();
521 string command_text = string.Format("select * from {0};", TABLES.RECORDING_SCHEDULE);
522 gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
523 using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
524 {
525 using (SQLiteDataReader r = cmd.ExecuteReader())
526 {
527 if (!r.HasRows)
528 {
529 gLog.Warn.WriteLine("Query: '{0}' returned no rows.", cmd.CommandText);
530 }
531 else
532 {
533 while (r.Read())
534 {
535 RECORDING_SCHEDULE recording = new RECORDING_SCHEDULE();
536 for (int i = 0; i < r.FieldCount; i++)
537 {
538 RECORDING_SCHEDULE.Create(ref recording, r, i);
539 }
540 recordings.Add(recording);
541 }
542 }
543 }
544 }
545 con.Clone();
546 }
547 this.Recordings = recordings;
548 }
549 catch (Exception ex)
550 {
551 gLog.Error.WriteLine(ex.ToString());
552 }
553 }
554
555 }
556 }

  ViewVC Help
Powered by ViewVC 1.1.22