erjemin/instaloader
erjemin/instaloader
1
0
Fork 0
Code Issues Wiki Activity

Support datetime objects in filter strings

Browse Source 
Now --only-if="date_utc<datetime(2018,1,1,hour=12)" is possible.
This commit is contained in:
Alexander Graf 2018-04-19 15:42:26 +02:00
parent 7d2830b8b2
commit 57ce51d4ef
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
instaloader/__main__.py
View File

@ -1,6 +1,7 @@
"""Download pictures (or videos) along with their captions and other metadata from Instagram.""" """Download pictures (or videos) along with their captions and other metadata from Instagram."""
import ast import ast
import datetime
import os import os
import sys import sys
from argparse import ArgumentParser, SUPPRESS from argparse import ArgumentParser, SUPPRESS
@ -35,6 +36,8 @@ def filterstr_to_filterfunc(filter_str: str, item_type: type):
# pylint:disable=no-self-use # pylint:disable=no-self-use
if not isinstance(node.ctx, ast.Load): if not isinstance(node.ctx, ast.Load):
raise InvalidArgumentException("Invalid filter: Modifying variables ({}) not allowed.".format(node.id)) raise InvalidArgumentException("Invalid filter: Modifying variables ({}) not allowed.".format(node.id))
if node.id == "datetime":
return node
if not hasattr(item_type, node.id): if not hasattr(item_type, node.id):
raise InvalidArgumentException("Invalid filter: {} not a {} attribute.".format(node.id, raise InvalidArgumentException("Invalid filter: {} not a {} attribute.".format(node.id,
item_type.__name__)) item_type.__name__))
@ -48,7 +51,7 @@ def filterstr_to_filterfunc(filter_str: str, item_type: type):
def filterfunc(item) -> bool: def filterfunc(item) -> bool:
# pylint:disable=eval-used # pylint:disable=eval-used
return bool(eval(compiled_filter, {'item': item})) return bool(eval(compiled_filter, {'item': item, 'datetime': datetime.datetime}))
return filterfunc return filterfunc