forked from GetStream/stream-python
-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy path__init__.py
More file actions
53 lines (47 loc) · 1.43 KB
/
__init__.py
File metadata and controls
53 lines (47 loc) · 1.43 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import re
import os
__author__ = "Thierry Schellenbach"
__copyright__ = "Copyright 2014, Stream.io, Inc"
__credits__ = ["Thierry Schellenbach, mellowmorning.com, @tschellenbach"]
__license__ = "BSD-3-Clause"
__version__ = "3.1.1"
__maintainer__ = "Thierry Schellenbach"
__status__ = "Production"
def connect(
api_key=None,
api_secret=None,
app_id=None,
version="v1.0",
timeout=3.0,
location=None,
base_url=None,
):
"""
Returns a Client object
:param api_key: your api key or heroku url
:param api_secret: the api secret
:param app_id: the app id (used for listening to feed changes)
"""
from stream.client import StreamClient
stream_url = os.environ.get("STREAM_URL")
# support for the heroku STREAM_URL syntax
if stream_url and not api_key:
pattern = re.compile(
r"https\:\/\/(\w+)\:(\w+)\@([\w-]*).*\?app_id=(\d+)", re.IGNORECASE
)
result = pattern.match(stream_url)
if result and len(result.groups()) == 4:
api_key, api_secret, location, app_id = result.groups()
location = None if location in ("getstream", "stream-io-api") else location
else:
raise ValueError("Invalid api key or heroku url")
return StreamClient(
api_key,
api_secret,
app_id,
version,
timeout,
location=location,
base_url=base_url,
)