/[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 159 by william, Thu Mar 14 20:26:05 2013 UTC revision 179 by william, Sat Mar 16 17:13:23 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> FixGBPVRProgramsDatabase(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            private struct OldNewProgram
26            {
27                public IPROGRAMME OldProgram;
28                public IPROGRAMME NewProgram;
29            }
30            public static ISQLLITE Create(string database, EventHandler<EventArgs> OnInstanceCreated)
31            {
32                return new SQLLITE(database, OnInstanceCreated);
33            }
34    
35          #region DATABASE DEFINITIONS          #region DATABASE DEFINITIONS
36    
37          public interface ICHANNEL          public interface ICHANNEL
# Line 175  namespace GBPVRProgramDatabaseFixer Line 196  namespace GBPVRProgramDatabaseFixer
196    
197          public interface IPROGRAMME          public interface IPROGRAMME
198          {          {
199              Int64 oid { get; }              Int64 oid { get; set; }
200              String name { get; }              String name { get; set; }
201              String sub_title { get; }              String sub_title { get; set; }
202              String description { get; }              String description { get; set; }
203              DateTime start_time { get; }              DateTime start_time { get; set; }
204              DateTime end_time { get; }              DateTime end_time { get; set; }
205              Int64 channel_oid { get; }              Int64 channel_oid { get; set; }
206              String unique_identifier { get; }              String unique_identifier { get; set; }
207              String rating { get; }              String rating { get; set; }
208    
209                IProgramDefinition AsXMLTVProgramDefinition(ISQLLITE sqllite);
210          }          }
211          private class PROGRAMME : IPROGRAMME          private class PROGRAMME : IPROGRAMME
212          {          {
# Line 206  namespace GBPVRProgramDatabaseFixer Line 229  namespace GBPVRProgramDatabaseFixer
229              public Int64 channel_oid { get; set; }              public Int64 channel_oid { get; set; }
230              public String unique_identifier { get; set; }              public String unique_identifier { get; set; }
231              public String rating { get; set; }              public String rating { get; set; }
232                public IProgramDefinition AsXMLTVProgramDefinition(ISQLLITE sqllite)
233                {
234                    ProgramList.ProgramDefintion definition = new ProgramList.ProgramDefintion();
235                    var channel = sqllite.Channels.Find(s => s.oid == this.channel_oid);
236                    definition.ChannelName = channel.display_name;
237                    definition.ChannelNumber = (int)channel.channel_number;
238                    definition.Description = this.description;
239                    definition.Start = this.start_time.ToDateTimeString();
240                    definition.Stop = this.end_time.ToDateTimeString();
241                    definition.SubTitle = this.sub_title;
242                    definition.Title = this.name;
243                    return definition;
244                }
245              #endregion              #endregion
246          }          }
247          #endregion          #endregion
# Line 217  namespace GBPVRProgramDatabaseFixer Line 253  namespace GBPVRProgramDatabaseFixer
253          }          }
254    
255          //public SQLLite() { }          //public SQLLite() { }
256          public SQLLITE(string database, EventHandler<EventArgs> OnInstanceCreated)          protected SQLLITE(string database, EventHandler<EventArgs> OnInstanceCreated)
257          {          {
258              this.OnInstanceCreated = OnInstanceCreated;              this.OnInstanceCreated = OnInstanceCreated;
259              //CreateConnection(database);              //CreateConnection(database);
# Line 243  namespace GBPVRProgramDatabaseFixer Line 279  namespace GBPVRProgramDatabaseFixer
279              if (OnInstanceCreated != null) { OnInstanceCreated.Invoke(sender, e); }              if (OnInstanceCreated != null) { OnInstanceCreated.Invoke(sender, e); }
280          }          }
281    
         #region SQLLite  
282          private string Database;          private string Database;
283            #region ISQLLITE members        
284          public List<IPROGRAMME> Programs { get; private set; }          public List<IPROGRAMME> Programs { get; private set; }
285          public List<IRECORDING_SCHEDULE> Recordings { get; private set; }          public List<IRECORDING_SCHEDULE> Recordings { get; private set; }
286          public List<ICHANNEL> Channels { get; private set; }          public List<ICHANNEL> Channels { get; private set; }
287    
288            public IDateTimeRange GetProgramsDateRange(List<SQLLITE.IPROGRAMME> programs)
289            {
290                var list = new List<IPROGRAMME>(programs.ToArray());
291                DateTime first = new DateTime();
292                DateTime last = new DateTime();
293                first = list.OrderBy(s => s.start_time).ToList().First().start_time;            
294                last = list.OrderBy(s => s.start_time).ToList().Last().start_time;
295                gLog.Verbose.Debug.WriteLine("\tFirst: {0} = ({1})", first.ToString("yyyy/MM/dd HH:mm:ss.fffffff"), first.ToDateTimeString());
296                gLog.Verbose.Debug.WriteLine("\tLast: {0} = ({1})", last.ToString("yyyy/MM/dd HH:mm:ss.fffffff"), last.ToDateTimeString());
297                var range = DateTimeRange.Create(first, last);
298                return range;
299            }
300    
301            public List<SQLLITE.IPROGRAMME> FixGBPVRProgramsDatabase(List<SQLLITE.IPROGRAMME> gbpvr_programs, List<libxmltv.Interfaces.IProgramDefinition> xmltv_programs, out List<SQLLITE.IPROGRAMME> source_invalid)
302            {
303                source_invalid = new List<IPROGRAMME>();
304                gbpvr_programs = gbpvr_programs.OrderBy(s => s.start_time).ToList();            
305                List<SQLLITE.IPROGRAMME> source_valid = new List<IPROGRAMME>();
306    
307                List<OldNewProgram> source_update = new List<OldNewProgram>();
308                //if (range == null)
309                //{
310                //    gLog.Warn.WriteLine("The DateTimeRange passed in is null...returning the original program list");
311                //    return list;
312                //}
313                //gLog.Warn.WriteLine("FilterProgramsByDateRange has not been implemented");
314                
315                double total = gbpvr_programs.Count;
316                double index = 0;
317                double progress = 0;
318                foreach (var program in gbpvr_programs)
319                {  
320                    progress = 100.0 * (index / total);
321                    gLog.ReportProgress(this, new ReportProgressEventArgs((int)progress, string.Format("Filtering GBPVR Programs: {0:00}%", (int)progress)));                
322                    var channel_oid = program.channel_oid;
323                    var channel = this.Channels.Find(s => s.oid == channel_oid);
324                    var start_date = program.start_time;
325                    //var xmltv_entry_list = xmltv_programs.FindAll(s => s.ChannelNumber == channel.channel_number && s.Start == start_date.ToDateTimeString());
326                    var query = from c in xmltv_programs
327                                where
328                                c.ChannelNumber == channel.channel_number &&
329                                c.Start == start_date.ToDateTimeString()
330                                select c;
331    
332                    IProgramDefinition xmltv_entry = null;
333                    if (query.Count()-1 > 0)
334                    {
335                        gLog.Verbose.Error.WriteLine("Found more than one entry: Matching channel='{0}' and start='{1}'", channel.channel_number, start_date.ToDateTimeString());
336                        gLog.Verbose.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);
337                        if (Debugger.IsAttached)
338                        {                      
339                            gLog.Error.WriteLine("    Found: {0} matching entries", query.Count());
340                            int k_index = 0;
341                            foreach (var k in query)
342                            {
343                                gLog.Verbose.Error.WriteLine("    query[{0}]: channel='{1}' start='{2}' ('{3}') title='{4}' subtitle='{5}' description='{6}'", k_index, k.ChannelNumber, k.Start, DateTime.Parse(k.Start).ToString("yyyy/MM/dd HH:mm:ss.fffffff"), k.Title, k.SubTitle, k.Description);
344                                k_index++;
345                            }
346                            Debugger.Break();
347                        }
348                    }
349                    else
350                    {
351                        xmltv_entry = query.FirstOrDefault();
352                    }
353    
354                    if (xmltv_entry == null)
355                    {
356                        //gLog.Verbose.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());
357                        source_invalid.Add(program);                  
358                    }
359                    else
360                    {
361                        //gLog.Verbose.Warn.WriteLine("Updating GB-PVR Program (if needed): 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());
362                        var updated_program = program;
363                        bool needsupdate = false;
364                        if (xmltv_entry.Title != program.name)
365                        {
366                            //gLog.Verbose.Warn.WriteLine("    Updating:");
367                            //gLog.Verbose.Warn.WriteLine("        Old Title: {0}", program.name);
368                            //gLog.Verbose.Warn.WriteLine("        New Title: {0}", xmltv_entry.Title);
369                            updated_program.name = xmltv_entry.Title;
370                            needsupdate = true;
371                        }
372                        if (xmltv_entry.SubTitle != program.sub_title)
373                        {
374                            //gLog.Verbose.Warn.WriteLine("    Updating:");
375                            //gLog.Verbose.Warn.WriteLine("        Old SubTile: {0}", program.sub_title);
376                            //gLog.Verbose.Warn.WriteLine("        New SubTile: {0}", xmltv_entry.SubTitle);
377                            updated_program.sub_title = xmltv_entry.SubTitle;
378                            needsupdate = true;
379                        }
380                        if (xmltv_entry.Description != program.description)
381                        {
382                            //gLog.Verbose.Warn.WriteLine("    Updating:");
383                            //gLog.Verbose.Warn.WriteLine("        Old Descption: {0}", program.description);
384                            //gLog.Verbose.Warn.WriteLine("        New Descption: {0}", xmltv_entry.Description);
385                            updated_program.description = xmltv_entry.Description;
386                            needsupdate = true;
387                        }
388                        if (DateTime.Parse(xmltv_entry.Start) != program.start_time)
389                        {
390                            //gLog.Verbose.Warn.WriteLine("    Updating:");
391                            //gLog.Verbose.Warn.WriteLine("        Old StartTime: {0}", program.start_time.ToDateTimeString());
392                            //gLog.Verbose.Warn.WriteLine("        New StartTime: {0}", DateTime.Parse(xmltv_entry.Start).ToDateTimeString());
393                            updated_program.start_time = DateTime.Parse(xmltv_entry.Start);
394                            needsupdate = true;
395                        }
396                        if (DateTime.Parse(xmltv_entry.Stop) != program.end_time)
397                        {
398                            //gLog.Verbose.Warn.WriteLine("    Updating:");
399                            //gLog.Verbose.Warn.WriteLine("        Old EndTime: {0}", program.end_time.ToDateTimeString());
400                            //gLog.Verbose.Warn.WriteLine("        New EndTime: {0}", DateTime.Parse(xmltv_entry.Stop).ToDateTimeString());
401                            updated_program.end_time = DateTime.Parse(xmltv_entry.Stop);
402                            needsupdate = true;
403                        }
404                        
405                        if (needsupdate)
406                        {
407                            OldNewProgram p = new OldNewProgram();
408                            p.OldProgram = program;
409                            p.NewProgram = updated_program;
410                            source_update.Add(p);
411                        }
412                        source_valid.Add(updated_program);
413                    }
414                    index++;
415                }
416    
417    
418                source_valid = source_valid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
419                source_invalid = source_invalid.OrderBy(s => s.channel_oid).ThenBy(s => s.start_time).ToList();
420    
421                if (source_invalid.Count > 0)
422                {
423                    double source_invalid_count = source_valid.Count;
424                    double source_invalid_index = 0;
425                    double source_invalid_progress = 0;
426                    foreach (var old_program in source_invalid)
427                    {
428                        source_invalid_progress = 100.0 * (source_invalid_index / source_invalid_count);
429                        gLog.ReportProgress(this, new ReportProgressEventArgs((int)source_invalid_progress, string.Format("Removing old GBPVR Programs: {0:00}%", (int)source_invalid_progress)));    
430                        // remove database entry
431                        if (!RemoveProgramDatabaseEntry(old_program))
432                        {
433                            gLog.Error.WriteLine("Failed to remove program with oid: '{0}'", old_program.oid);
434                        }
435                        else
436                        {
437                            gLog.Verbose.Info.WriteLine("Removed program with oid: '{0}'", old_program.oid);
438                        }
439                        source_invalid_index++;
440                    }
441                }
442                else
443                {
444                    gLog.Info.WriteLine("No old GB-PVR Programs needed to be removed.");
445                }
446    
447                if (source_update.Count > 0)
448                {
449                    double source_update_count = source_valid.Count;
450                    double source_update_index = 0;
451                    double source_update_progress = 0;
452                    foreach (var p in source_update)
453                    {
454                        source_update_progress = 100.0 * (source_update_index / source_update_count);
455                        gLog.ReportProgress(this, new ReportProgressEventArgs((int)source_update_progress, string.Format("Updating GBPVR Programs: {0:00}%", (int)source_update_progress)));    
456                        // remove database entry
457                        if (!UpdateProgramDatabaseEntry(p.OldProgram,p.NewProgram))
458                        {
459                            gLog.Error.WriteLine("Failed to update program with oid: '{0}'", p.OldProgram.oid);
460                        }
461                        else
462                        {
463                            gLog.Verbose.Info.WriteLine("Upated program with oid: '{0}'", p.OldProgram.oid);
464                        }
465                        source_update_index++;
466                    }
467                }
468                else
469                {
470                    gLog.Info.WriteLine("No GB-PVR Programs needed to be updated.");
471                }
472    
473                gLog.Info.WriteLine("Total XMLTV Programs: 0x{0:x8}", xmltv_programs.Count);
474                gLog.Info.WriteLine("Updated: 0x{0:x8} GB-PVR Programs", source_valid.Count);
475                gLog.Info.WriteLine("Removed: 0x{0:x8} GB-PVR Programs", source_invalid.Count);
476                gLog.Info.WriteLine("Total GB-PVR Programs (Updated & Removed): 0x{0:x8}", source_valid.Count + source_invalid.Count);
477    
478                return source_valid;
479            }
480          #endregion          #endregion
481    
482    
# Line 446  namespace GBPVRProgramDatabaseFixer Line 675  namespace GBPVRProgramDatabaseFixer
675              }              }
676          }          }
677    
678    
679    
680            private bool RemoveProgramDatabaseEntry(IPROGRAMME old_program)
681            {
682                bool result = false;
683                try
684                {                
685                    using (SQLiteConnection con = CreateConnection())
686                    {
687                        try
688                        {
689                            gLog.Verbose.Info.WriteLine("Removing old program with oid: '{0}'", old_program.oid);
690                            con.Open();
691                            string command_text = string.Format(@"DELETE FROM [{0}] WHERE [oid] = @oid;", TABLES.PROGRAMME);
692                            gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
693                            using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
694                            {
695                                cmd.Parameters.Add(new SQLiteParameter("oid", old_program.oid));
696                                int rowsupdated = cmd.ExecuteNonQuery();
697                                gLog.Verbose.Info.WriteLine("Updated '{0}' rows", rowsupdated);
698                            }
699                            con.Clone();
700                            result = true;
701                        }
702                        catch (SQLiteException ex)
703                        {
704                            gLog.Error.WriteLine(ex.ToString());
705                            result = false;
706                        }
707                    }
708                }
709                catch (Exception ex)
710                {
711                    gLog.Error.WriteLine(ex.ToString());
712                    result = false;
713                }
714                return result;
715            }
716            private bool UpdateProgramDatabaseEntry(IPROGRAMME old_program, IPROGRAMME new_program)
717            {
718                bool result = false;
719                try
720                {
721                    using (SQLiteConnection con = CreateConnection())
722                    {
723                        try
724                        {
725                            gLog.Verbose.Info.WriteLine("Updating old program with oid: '{0}'", old_program.oid);
726                            con.Open();
727                            //string command_text = string.Format(@"UPDATE [{0}] SET [name]=@name,[sub_title]=@subtitle, WHERE [OID] = @oid", TABLES.PROGRAMME);
728                            string command_text = string.Empty;
729                            StringBuilder builder = new StringBuilder();
730    
731                            builder.AppendLine(string.Format("UPDATE [{0}]", TABLES.PROGRAMME));
732                            builder.Append("SET ");
733                            builder.AppendFormat("[name]=@name,");
734                            builder.AppendFormat("[sub_title]=@sub_title,");
735                            builder.AppendFormat("[description]=@description,");
736                            builder.AppendFormat("[start_time]=@start_time,");
737                            builder.AppendFormat("[end_time]=@end_time,");
738                            builder.AppendFormat("[channel_oid]=@channel_oid,");
739                            builder.AppendFormat("[unique_identifier]=@unique_identifier,");
740                            builder.AppendFormat("[rating]=@rating");
741                            builder.AppendLine(" WHERE [oid] = @oid");
742    
743                            command_text = builder.ToString();
744                            gLog.Verbose.Debug.WriteLine("Executing Command: '{0}'", command_text);
745                            using (SQLiteCommand cmd = new SQLiteCommand(command_text, con))
746                            {
747                                cmd.Parameters.Add(new SQLiteParameter("oid", old_program.oid));
748                                cmd.Parameters.Add(new SQLiteParameter("name", new_program.name));
749                                cmd.Parameters.Add(new SQLiteParameter("sub_title", new_program.sub_title));
750                                cmd.Parameters.Add(new SQLiteParameter("description", new_program.description));
751                                cmd.Parameters.Add(new SQLiteParameter("start_time", new_program.start_time));
752                                cmd.Parameters.Add(new SQLiteParameter("end_time", new_program.end_time));
753                                cmd.Parameters.Add(new SQLiteParameter("channel_oid", new_program.channel_oid));
754                                cmd.Parameters.Add(new SQLiteParameter("unique_identifier", new_program.unique_identifier));
755                                cmd.Parameters.Add(new SQLiteParameter("rating", new_program.rating));
756                                int rowsupdated = cmd.ExecuteNonQuery();
757                                gLog.Verbose.Info.WriteLine("Updated '{0}' rows", rowsupdated);
758                            }
759                            con.Clone();
760                            result = true;
761                        }
762                        catch (SQLiteException ex)
763                        {
764                            gLog.Error.WriteLine(ex.ToString());
765                            result = false;
766                        }
767                    }
768                }
769                catch (Exception ex)
770                {
771                    gLog.Error.WriteLine(ex.ToString());
772                    result = false;
773                }
774                return result;
775            }
776      }      }
777  }  }

Legend:
Removed from v.159  
changed lines
  Added in v.179

  ViewVC Help
Powered by ViewVC 1.1.22