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

Annotation of /trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs

Parent Directory Parent Directory | Revision Log Revision Log


Revision 170 - (hide annotations) (download)
Sat Mar 16 14:28:56 2013 UTC (6 years, 10 months ago) by william
File size: 25693 byte(s)

1 william 156 //#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 william 163 using libxmltv;
10 william 165 using libxmltv.Interfaces;
11     using libxmltv.Core;
12 william 156 namespace GBPVRProgramDatabaseFixer
13     {
14 william 163 public interface ISQLLITE
15 william 156 {
16 william 163 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 william 168 List<SQLLITE.IPROGRAMME> CreateUpdatedProgramsList(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> removed_programs);
21 william 163
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 william 156 #region DATABASE DEFINITIONS
31 william 159
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 william 156 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 william 168 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 william 165
204     IProgramDefinition AsXMLTVProgramDefinition(ISQLLITE sqllite);
205 william 156 }
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 william 165 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 william 168 definition.Start = this.start_time.ToDateTimeString();
235     definition.Stop = this.end_time.ToDateTimeString();
236 william 165 definition.SubTitle = this.sub_title;
237     definition.Title = this.name;
238     return definition;
239     }
240 william 156 #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 william 159 public const string CHANNEL = "CHANNEL";
248 william 156 }
249    
250     //public SQLLite() { }
251 william 163 protected SQLLITE(string database, EventHandler<EventArgs> OnInstanceCreated)
252 william 156 {
253     this.OnInstanceCreated = OnInstanceCreated;
254     //CreateConnection(database);
255     this.Database = database;
256 william 157 if (!CreateDatabaseBackup())
257     {
258     gLog.Error.WriteLine("Failed to backup database.");
259     return;
260     }
261 william 156 ConnectionTest();
262 william 159 ReadChannelData();
263 william 156 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 william 163 #region ISQLLITE members
279 william 156 public List<IPROGRAMME> Programs { get; private set; }
280     public List<IRECORDING_SCHEDULE> Recordings { get; private set; }
281 william 159 public List<ICHANNEL> Channels { get; private set; }
282 william 163
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 william 167 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 william 163 var range = DateTimeRange.Create(first, last);
293     return range;
294     }
295    
296 william 169 public List<SQLLITE.IPROGRAMME> CreateUpdatedProgramsList(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> source_invalid)
297 william 163 {
298 william 169 source_invalid = new List<IPROGRAMME>();
299 william 168 gbpvr_programs = gbpvr_programs.OrderBy(s => s.start_time).ToList();
300 william 166 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 william 163 //gLog.Warn.WriteLine("FilterProgramsByDateRange has not been implemented");
307 william 170
308 william 168 double total = gbpvr_programs.Count;
309 william 166 double index = 0;
310     double progress = 0;
311 william 168 foreach (var program in gbpvr_programs)
312 william 169 {
313     progress = 100.0 * (index / total);
314     gLog.ReportProgress(this, new ReportProgressEventArgs((int)progress, string.Format("Filtering GBPVR Programs: {0:00}%", (int)progress)));
315 william 168 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 william 170 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 william 168 if (xmltv_entry == null)
334 william 163 {
335 william 170 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 william 169 source_invalid.Add(program);
337 william 163 }
338 william 164 else
339     {
340 william 168 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 william 170 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 william 168 source_valid.Add(updated_program);
349 william 164 }
350 william 169 index++;
351 william 163 }
352 william 166
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 william 169 source_invalid = source_invalid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
365 william 167
366     gLog.Info.WriteLine("Total XMLTV Programs: 0x{0:x8}", xmltv_programs.Count);
367 william 168 gLog.Info.WriteLine("Updated: 0x{0:x8} GB-PVR Programs", source_valid.Count);
368 william 169 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 william 167
371 william 166 return source_valid;
372 william 163 }
373 william 156 #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 william 159 string command_text = string.Format("select * from {0};", TABLES.CHANNEL);
393 william 156 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 william 157 con.Clone();
419 william 156 }
420     OnCreatedInstance(this, new EventArgs());
421     }
422     catch (Exception ex)
423     {
424     gLog.Error.WriteLine(ex.ToString());
425     }
426     }
427    
428 william 157
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 william 159 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 william 156 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 william 159
571 william 156 }
572     }

  ViewVC Help
Powered by ViewVC 1.1.22