Add date field to eztv

nova3/engines/eztv.py

@@ -1,10 +1,12 @@
-#VERSION: 1.15
+#VERSION: 1.16
 # AUTHORS: nindogo
 # CONTRIBUTORS: Diego de las Heras ([email protected])
 
+import re
 import urllib.error
 import urllib.parse
 import urllib.request
+from datetime import datetime, timedelta
 from html.parser import HTMLParser
 
 from novaprinter import prettyPrinter
@@ -24,6 +26,14 @@ def __init__(self, url):
  HTMLParser.__init__(self)
  self.url = url
 
+ now = datetime.now()
+ self.date_parsers = {
+ r"(\d+)h\s+(\d+)m": lambda m: now - timedelta(hours=int(m[1]), minutes=int(m[2])),
+ r"(\d+)d\s+(\d+)h": lambda m: now - timedelta(days=int(m[1]), hours=int(m[2])),
+ r"(\d+)\s+weeks?": lambda m: now - timedelta(weeks=int(m[1])),
+ r"(\d+)\s+mo": lambda m: now - timedelta(days=int(m[1]) * 30),
+ r"(\d+)\s+years?": lambda m: now - timedelta(days=int(m[1]) * 365),
+ }
  self.in_table_row = False
  self.current_item = {}
 
@@ -38,6 +48,7 @@ def handle_starttag(self, tag, attrs):
  self.current_item['leech'] = -1
  self.current_item['size'] = -1
  self.current_item['engine_url'] = self.url
+ self.current_item['pub_date'] = -1
 
  if (tag == self.A
  and self.in_table_row and params.get('class') == 'magnet'):
@@ -57,6 +68,13 @@ def handle_data(self, data):
  elif self.in_table_row and data.isnumeric():
  self.current_item['seeds'] = int(data)
 
+ elif self.in_table_row: # Check for a relative time
+ for pattern, calc in self.date_parsers.items():
+ m = re.match(pattern, data)
+ if m:
+ self.current_item["pub_date"] = int(calc(m).timestamp())
+ break
+
  def handle_endtag(self, tag):
  if self.in_table_row and tag == self.TR:
  prettyPrinter(self.current_item)

nova3/engines/versions.txt

@@ -1,4 +1,4 @@
-eztv: 1.15
+eztv: 1.16
 jackett: 4.0
 limetorrents: 4.7
 piratebay: 3.3