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

Diff of /trunk/GBPVRProgramDatabaseFixer/SQLLITE.cs

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 157 by william, Thu Mar 14 19:36:01 2013 UTC revision 169 by william, Sat Mar 16 14:04:31 2013 UTC
# Line 6  using System.Text; Line 6  using System.Text;
6  using System.Data.SQLite;  using System.Data.SQLite;
7  using System.Diagnostics;  using System.Diagnostics;
8  using Enterprise.Logging;  using Enterprise.Logging;
9    using libxmltv;
10    using libxmltv.Interfaces;
11    using libxmltv.Core;
12  namespace GBPVRProgramDatabaseFixer  namespace GBPVRProgramDatabaseFixer
13  {  {
14      internal class SQLLITE      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          #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          public interface IRECORDING_SCHEDULE
63          {          {
64              Int64 oid { get; }              Int64 oid { get; }
# Line 144  namespace GBPVRProgramDatabaseFixer Line 191  namespace GBPVRProgramDatabaseFixer
191    
192          public interface IPROGRAMME          public interface IPROGRAMME
193          {          {
194              Int64 oid { get; }              Int64 oid { get; set; }
195              String name { get; }              String name { get; set; }
196              String sub_title { get; }              String sub_title { get; set; }
197              String description { get; }              String description { get; set; }
198              DateTime start_time { get; }              DateTime start_time { get; set; }
199              DateTime end_time { get; }              DateTime end_time { get; set; }
200              Int64 channel_oid { get; }              Int64 channel_oid { get; set; }
201              String unique_identifier { get; }              String unique_identifier { get; set; }
202              String rating { get; }              String rating { get; set; }
203    
204                IProgramDefinition AsXMLTVProgramDefinition(ISQLLITE sqllite);
205          }          }
206          private class PROGRAMME : IPROGRAMME          private class PROGRAMME : IPROGRAMME
207          {          {
# Line 175  namespace GBPVRProgramDatabaseFixer Line 224  namespace GBPVRProgramDatabaseFixer
224              public Int64 channel_oid { get; set; }              public Int64 channel_oid { get; set; }
225              public String unique_identifier { get; set; }              public String unique_identifier { get; set; }
226              public String rating { get; set; }              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              #endregion
241          }          }
242          #endregion          #endregion
# Line 182  namespace GBPVRProgramDatabaseFixer Line 244  namespace GBPVRProgramDatabaseFixer
244          {          {
245              public const string RECORDING_SCHEDULE = "RECORDING_SCHEDULE";              public const string RECORDING_SCHEDULE = "RECORDING_SCHEDULE";
246              public const string PROGRAMME = "PROGRAMME";              public const string PROGRAMME = "PROGRAMME";
247                public const string CHANNEL = "CHANNEL";
248          }          }
249    
250          //public SQLLite() { }          //public SQLLite() { }
251          public SQLLITE(string database, EventHandler<EventArgs> OnInstanceCreated)          protected SQLLITE(string database, EventHandler<EventArgs> OnInstanceCreated)
252          {          {
253              this.OnInstanceCreated = OnInstanceCreated;              this.OnInstanceCreated = OnInstanceCreated;
254              //CreateConnection(database);              //CreateConnection(database);
# Line 196  namespace GBPVRProgramDatabaseFixer Line 259  namespace GBPVRProgramDatabaseFixer
259                  return;                  return;
260              }              }
261              ConnectionTest();              ConnectionTest();
262                ReadChannelData();
263              ReadRecodringScheduleData();              ReadRecodringScheduleData();
264              ReadProgrammeData();              ReadProgrammeData();
265              OnCreatedInstance(this, new EventArgs());              OnCreatedInstance(this, new EventArgs());
# Line 210  namespace GBPVRProgramDatabaseFixer Line 274  namespace GBPVRProgramDatabaseFixer
274              if (OnInstanceCreated != null) { OnInstanceCreated.Invoke(sender, e); }              if (OnInstanceCreated != null) { OnInstanceCreated.Invoke(sender, e); }
275          }          }
276    
         #region SQLLite  
277          private string Database;          private string Database;
278            #region ISQLLITE members        
279          public List<IPROGRAMME> Programs { get; private set; }          public List<IPROGRAMME> Programs { get; private set; }
280          public List<IRECORDING_SCHEDULE> Recordings { get; private set; }          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          #endregion
357    
358    
# Line 233  namespace GBPVRProgramDatabaseFixer Line 372  namespace GBPVRProgramDatabaseFixer
372                  using (SQLiteConnection con = CreateConnection())                  using (SQLiteConnection con = CreateConnection())
373                  {                  {
374                      con.Open();                      con.Open();
375                      string command_text = string.Format("select * from {0};", TABLES.PROGRAMME);                      string command_text = string.Format("select * from {0};", TABLES.CHANNEL);
376                      gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);                      gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
377                      using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))                      using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
378                      {                      {
# Line 288  namespace GBPVRProgramDatabaseFixer Line 427  namespace GBPVRProgramDatabaseFixer
427                  return false;                  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()          private void ReadProgrammeData()
472          {          {
473              try              try
# Line 357  namespace GBPVRProgramDatabaseFixer Line 536  namespace GBPVRProgramDatabaseFixer
536                                      {                                      {
537                                          RECORDING_SCHEDULE.Create(ref recording, r, i);                                          RECORDING_SCHEDULE.Create(ref recording, r, i);
538                                      }                                      }
   
539                                      recordings.Add(recording);                                      recordings.Add(recording);
540                                  }                                  }
541                              }                              }
# Line 372  namespace GBPVRProgramDatabaseFixer Line 550  namespace GBPVRProgramDatabaseFixer
550                  gLog.Error.WriteLine(ex.ToString());                  gLog.Error.WriteLine(ex.ToString());
551              }              }
552          }          }
553    
554      }      }
555  }  }

Legend:
Removed from v.157  
changed lines
  Added in v.169

  ViewVC Help
Powered by ViewVC 1.1.22