def test_get_and_put_cached(self):
storage = StorageByKeyName(CredentialsModel,
'foo',
'credentials',
cache=memcache)
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsModel.get_by_key_name('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
# Now remove the item from the cache.
memcache.delete('foo')
# Check that getting refreshes the cache.
credentials = storage.get()
self.assertEqual('bar', credentials.access_token)
self.assertNotEqual(None, memcache.get('foo'))
# Deleting should clear the cache.
storage.delete()
credentials = storage.get()
self.assertEqual(None, credentials)
self.assertEqual(None, memcache.get('foo'))
def test_get_and_put_set_store_on_cache_retrieval(self):
storage = StorageByKeyName(
CredentialsModel, 'foo', 'credentials', cache=memcache)
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
storage.put(self.credentials)
# Pre-bug 292 old_creds wouldn't have storage, and the _refresh
# wouldn't be able to store the updated cred back into the storage.
old_creds = storage.get()
self.assertEqual(old_creds.access_token, 'foo')
old_creds.invalid = True
old_creds._refresh(_http_request)
new_creds = storage.get()
self.assertEqual(new_creds.access_token, 'bar')
def test_delete_ndb(self):
# Start empty
storage = StorageByKeyName(CredentialsNDBModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
# Add credentials to model with storage, and check equivalent
# w/o storage
storage.put(self.credentials)
credmodel = CredentialsNDBModel.get_by_id('foo')
self.assertEqual(credmodel.credentials.to_json(),
self.credentials.to_json())
# Delete and make sure empty
storage.delete()
self.assertEqual(None, storage.get())
def test_delete_ndb(self):
# Start empty
storage = StorageByKeyName(
CredentialsNDBModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
# Add credentials to model with storage, and check equivalent
# w/o storage
storage.put(self.credentials)
credmodel = CredentialsNDBModel.get_by_id('foo')
self.assertEqual(credmodel.credentials.to_json(),
self.credentials.to_json())
# Delete and make sure empty
storage.delete()
self.assertEqual(None, storage.get())
def test_get_and_put_set_store_on_cache_retrieval(self):
storage = StorageByKeyName(CredentialsModel,
'foo',
'credentials',
cache=memcache)
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
storage.put(self.credentials)
# Pre-bug 292 old_creds wouldn't have storage, and the _refresh
# wouldn't be able to store the updated cred back into the storage.
old_creds = storage.get()
self.assertEqual(old_creds.access_token, 'foo')
old_creds.invalid = True
old_creds._refresh(_http_request)
new_creds = storage.get()
self.assertEqual(new_creds.access_token, 'bar')
def test_get_and_put_simple(self):
storage = StorageByKeyName(CredentialsModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsModel.get_by_key_name('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
def test_get_and_put_simple(self):
storage = StorageByKeyName(
CredentialsModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsModel.get_by_key_name('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
def test_get_and_put_ndb(self):
# Start empty
storage = StorageByKeyName(CredentialsNDBModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
# Refresh storage and retrieve without using storage
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsNDBModel.get_by_id('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
self.assertEqual(credmodel.credentials.to_json(),
self.credentials.to_json())
def test_delete_db_ndb_mixed(self):
# Start empty
storage_ndb = StorageByKeyName(
CredentialsNDBModel, 'foo', 'credentials')
storage = StorageByKeyName(
CredentialsModel, 'foo', 'credentials')
# First DB, then NDB
self.assertEqual(None, storage.get())
storage.put(self.credentials)
self.assertNotEqual(None, storage.get())
storage_ndb.delete()
self.assertEqual(None, storage.get())
# First NDB, then DB
self.assertEqual(None, storage_ndb.get())
storage_ndb.put(self.credentials)
storage.delete()
self.assertNotEqual(None, storage_ndb.get())
# NDB uses memcache and an instance cache (Context)
ndb.get_context().clear_cache()
memcache.flush_all()
self.assertEqual(None, storage_ndb.get())
def test_get_and_put_ndb(self):
# Start empty
storage = StorageByKeyName(
CredentialsNDBModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
# Refresh storage and retrieve without using storage
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsNDBModel.get_by_id('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
self.assertEqual(credmodel.credentials.to_json(),
self.credentials.to_json())
def test_get_and_put_mixed_db_storage_ndb_get(self):
# Start empty
storage = StorageByKeyName(CredentialsModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
# Set DB store and refresh to add to storage
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
# Retrieve same key from NDB model to confirm mixing works
credmodel = CredentialsNDBModel.get_by_id('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
self.assertEqual(self.credentials.to_json(),
credmodel.credentials.to_json())
def test_get_and_put_mixed_db_storage_ndb_get(self):
# Start empty
storage = StorageByKeyName(
CredentialsModel, 'foo', 'credentials')
self.assertEqual(None, storage.get())
# Set DB store and refresh to add to storage
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
# Retrieve same key from NDB model to confirm mixing works
credmodel = CredentialsNDBModel.get_by_id('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
self.assertEqual(self.credentials.to_json(),
credmodel.credentials.to_json())
def test_get_and_put_cached(self):
storage = StorageByKeyName(
CredentialsModel, 'foo', 'credentials', cache=memcache)
self.assertEqual(None, storage.get())
self.credentials.set_store(storage)
self.credentials._refresh(_http_request)
credmodel = CredentialsModel.get_by_key_name('foo')
self.assertEqual('bar', credmodel.credentials.access_token)
# Now remove the item from the cache.
memcache.delete('foo')
# Check that getting refreshes the cache.
credentials = storage.get()
self.assertEqual('bar', credentials.access_token)
self.assertNotEqual(None, memcache.get('foo'))
# Deleting should clear the cache.
storage.delete()
credentials = storage.get()
self.assertEqual(None, credentials)
self.assertEqual(None, memcache.get('foo'))
def test_delete_db_ndb_mixed(self):
# Start empty
storage_ndb = StorageByKeyName(CredentialsNDBModel, 'foo',
'credentials')
storage = StorageByKeyName(CredentialsModel, 'foo', 'credentials')
# First DB, then NDB
self.assertEqual(None, storage.get())
storage.put(self.credentials)
self.assertNotEqual(None, storage.get())
storage_ndb.delete()
self.assertEqual(None, storage.get())
# First NDB, then DB
self.assertEqual(None, storage_ndb.get())
storage_ndb.put(self.credentials)
storage.delete()
self.assertNotEqual(None, storage_ndb.get())
# NDB uses memcache and an instance cache (Context)
ndb.get_context().clear_cache()
memcache.flush_all()
self.assertEqual(None, storage_ndb.get())
from google.appengine.api import users from oauth2client.contrib.appengine import StorageByKeyName user = users.get_current_user() storage = StorageByKeyName(CredentialsModel, user.user_id(), 'credentials') credentials = storage.get() # hi dewdadssddsaad
def get_credentials(self):
storage = StorageByKeyName(TokenizedCredentialsModel, self.token,
'credentials')
return storage.get()
def post(self):
"""Entry point for the worker."""
# channel_id refers to the channel that the bot is using
# video_id refers to the live stream we want to chat in
channel_id = self.request.get("channel_id")
live_chat_id = self.request.get("live_chat_id")
storage = StorageByKeyName(CredentialsModel, channel_id, 'credentials')
credential = storage.get()
http = credential.authorize(httplib2.Http())
youtube = build('youtube', 'v3', http=http)
in_chat = memcache.get("{}:in_chat".format(live_chat_id))
if in_chat is None:
say("Hello, I've joined the chat!", live_chat_id, youtube)
next_page = memcache.get("{}:nextpage".format(live_chat_id))
# This is our loop control. If we want the bot to gracefully exit,
# we can just set this to false.
remain_in_channel = True
try:
while remain_in_channel:
memcache.set("{}:in_chat".format(live_chat_id), True,
MEMCACHE_CHAT_PING_EXPIRY_TIME)
messages = youtube.liveChatMessages().list(liveChatId=live_chat_id,
part="id,snippet,authorDetails",
pageToken=next_page).execute()
if messages is None:
break
for message in messages['items']:
message_id = message['id']
message_type = message['snippet']['type']
# Keep track of every message we process, so we
# don't accidentally do it more than once.
already_processed = memcache.get("{}:processed".format(message_id))
if already_processed:
continue
else:
memcache.set("{}:processed".format(message_id), True)
# Here, we're doing the real work of the bot.
# Check the message type and respond appropriately.
# Message types are documented here:
# https://developers.google.com/youtube/v3/live/docs/liveChatMessages#snippet.type
# Before we proceed, let's pull out the author details, so we can
# personalize our response and do some basic permission checking.
author_channel_name = message['authorDetails']['displayName']
author_is_moderator = message['authorDetails']['isChatModerator']
author_is_owner = message['authorDetails']['isChatOwner']
if message_type == "textMessageEvent":
message_text = message['snippet']['textMessageDetails']['messageText']
if message_text == ".hi":
say("Well hello there, {}!".format(author_channel_name), live_chat_id,
youtube)
elif message_text == ".leave":
# We only want moderators or the channel owner to be able to
# tell the bot to leave. Let's ensure that's the case.
if author_is_moderator or author_is_owner:
say("Okay {}, I'm leaving the channel!".format(author_channel_name),
live_chat_id, youtube)
remain_in_channel = False
elif type == "chatEndedEvent":
remain_in_channel = False
break
next_page = messages['nextPageToken']
memcache.set("{}:nextpage".format(live_chat_id), next_page)
time.sleep(messages['pollingIntervalMillis']/1000)
except DeadlineExceededError:
# App Engine is terminating our task, so we need to re-queue it.
# Tasks in Task Queues have deadlines. To learn more:
# https://cloud.google.com/appengine/docs/standard/python/taskqueue/push/creating-handlers
if remain_in_channel:
taskqueue.add(url='/spawnbot', target='worker', params=
{'channel_id':channel_id,
'live_chat_id':live_chat_id})
else:
# TODO: Don't duplicate the cleanup code.
memcache.delete("{}:nextpage".format(live_chat_id))
memcache.delete("{}:in_chat".format(live_chat_id))
return
memcache.delete("{}:nextpage".format(live_chat_id))
memcache.delete("{}:in_chat".format(live_chat_id))
