karaolidis/fbchat
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added get_url_parameters util method

Browse Source
This commit is contained in:
Kacper Ziubryniewicz
2018-09-15 11:21:35 +02:00
committed by GitHub
parent 940a65954c
commit 90813c959d
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
fbchat/utils.py
View File

@@ -14,10 +14,10 @@ import requests
from .models import * from .models import *
try: try:
from urllib.parse import urlencode from urllib.parse import urlencode, parse_qs, urlparse
basestring = (str, bytes) basestring = (str, bytes)
except ImportError: except ImportError:
from urllib import urlencode from urllib import urlencode, parse_qs, urlparse
basestring = basestring basestring = basestring
# Python 2's `input` executes the input, whereas `raw_input` just returns the input # Python 2's `input` executes the input, whereas `raw_input` just returns the input
@@ -297,3 +297,10 @@ def get_files_from_paths(filenames):
yield files yield files
for fn, fp, ft in files: for fn, fp, ft in files:
fp.close() fp.close()
def get_url_parameters(url, *args):
params = parse_qs(urlparse(url).query)
return [params[arg][0] for arg in args if params.get(arg)]
def get_url_parameter(url, param):
return get_url_parameters(url, param)[0]