vcs.maemo.org Git - cinaest/blob - src/imdb/imdb-line-parser.vala

   1 abstract class LineParser {
   2         internal unowned IMDbSqlite sqlite;
   3
   4         public LineParser (IMDbSqlite _sqlite) {
   5                 sqlite = _sqlite;
   6         }
   7
   8         public abstract void parse_line (string line);
   9
  10         internal bool skip_title (string title) {
  11                 if (title.has_suffix ("(TV)")) {
  12                         return true;
  13                 }
  14                 if (title.has_suffix ("(V)")) {
  15                         return true;
  16                 }
  17                 if (title.has_suffix ("(VG)")) {
  18                         return true;
  19                 }
  20                 return false;
  21         }
  22 }
  23
  24 class MovieLineParser : LineParser {
  25         Regex re_movie;
  26
  27         public MovieLineParser (IMDbSqlite _sqlite) {
  28                 base (_sqlite);
  29                 try {
  30                         re_movie = new Regex ("^([^\t]+)\t+([0-9]+)$");
  31                 } catch (RegexError e) {
  32                         critical ("Failed to initialize regex: %s\n", e.message);
  33                 }
  34         }
  35
  36         public override void parse_line (string line) {
  37                 MatchInfo matchinfo;
  38
  39                 // Skip series episodes
  40                 if (line[0] == '"')
  41                         return;
  42
  43                 if (!re_movie.match(line, 0, out matchinfo))
  44                         return;
  45
  46                 string title;
  47                 string year = matchinfo.fetch (2);
  48                 try {
  49                         title = convert(matchinfo.fetch (1), -1, "utf-8", "latin1");
  50                 } catch (ConvertError e) {
  51                         return;
  52                 }
  53
  54                 if (skip_title (title))
  55                         return;
  56
  57                 sqlite.add_movie (title, year.to_int ());
  58         }
  59 }
  60
  61 class GenreLineParser : LineParser {
  62         Regex re_genre;
  63
  64         public GenreLineParser (IMDbSqlite _sqlite) {
  65                 base (_sqlite);
  66                 try {
  67                         re_genre = new Regex ("^([^\t]+)\t+([A-Za-z-]+)$");
  68                 } catch (RegexError e) {
  69                         critical ("Failed to initialize regex: %s\n", e.message);
  70                 }
  71         }
  72
  73         public override void parse_line (string line) {
  74                 MatchInfo matchinfo;
  75
  76                 // Skip series episodes
  77                 if (line[0] == '"')
  78                         return;
  79
  80                 if (!re_genre.match(line, 0, out matchinfo))
  81                         return;
  82
  83                 string title;
  84                 string genre = matchinfo.fetch (2);
  85                 try {
  86                         title = convert(matchinfo.fetch (1), -1, "utf-8", "latin1");
  87                 } catch (ConvertError e) {
  88                         return;
  89                 }
  90
  91                 sqlite.movie_add_genre (title, genre);
  92         }
  93 }
  94
  95 class RatingLineParser : LineParser {
  96         Regex re_rating;
  97
  98         public RatingLineParser (IMDbSqlite _sqlite) {
  99                 base (_sqlite);
 100                 try {
 101                         re_rating = new Regex ("^      .+ +([0-9]+) +([0-9.]+) +(.+)$");
 102                 } catch (RegexError e) {
 103                         critical ("Failed to initialize regex: %s\n", e.message);
 104                 }
 105         }
 106
 107         public override void parse_line (string line) {
 108                 MatchInfo matchinfo;
 109
 110                 // Skip series episodes
 111                 if (line[0] == '"')
 112                         return;
 113
 114                 if (!re_rating.match(line, 0, out matchinfo))
 115                         return;
 116
 117                 string title;
 118                 string votes = matchinfo.fetch (1);
 119                 string rating = matchinfo.fetch (2);
 120                 try {
 121                         title = convert(matchinfo.fetch (3), -1, "utf-8", "latin1");
 122                 } catch (ConvertError e) {
 123                         return;
 124                 }
 125
 126                 // Skip series episodes
 127                 if (title[0] == '"')
 128                         return;
 129
 130                 if (skip_title (title))
 131                         return;
 132
 133                 sqlite.movie_set_rating (title, (int) (rating.to_double () * 10), votes.to_int ());
 134         }
 135 }