storage/demo: add demo script parser

gcloud/demo.py

@@ -0,0 +1,93 @@
+from code import interact
+import os.path
+import sys
+import time
+
+from gcloud import storage
+
+
+__all__ = ['CLIENT_EMAIL', 'PRIVATE_KEY_PATH', 'PROJECT_NAME',
+ 'get_connection', 'main']
+
+
+CLIENT_EMAIL = '606734090113-6ink7iugcv89da9sru7lii8bs3i0obqg@developer.gserviceaccount.com'
+PRIVATE_KEY_PATH = os.path.join(os.path.dirname(__file__), 'demo.key')
+PROJECT_NAME = 'gcloud-storage-demo'
+
+extra_newline = False
+code_globals, code_locals = globals(), locals()
+
+
+def get_storage_connection():
+ return storage.get_connection(PROJECT_NAME, CLIENT_EMAIL, PRIVATE_KEY_PATH)
+
+
+def write(*strings):
+ # Add an extra newline if necessary.
+ global extra_newline
+ if extra_newline:
+ print
+
+ for string in strings:
+ print string
+ raw_input()
+
+ # We don't need an extra newline after this.
+ extra_newline = False
+
+
+def code(string, comment=None):
+ keypress_time = 0.05
+
+ print '>>> ',
+ for char in string:
+ time.sleep(keypress_time)
+ sys.stdout.write(char)
+ sys.stdout.flush()
+
+ if comment:
+ sys.stdout.write(' # %s' % comment)
+
+ # Wait for an enter key before continuing...
+ raw_input()
+
+ # Yes, this is crazy unsafe... but it's demo code.
+ # Globalize these so our imports hang around...
+ global code_globals
+ global code_locals
+ exec(string, code_globals, code_locals)
+
+ # In the next non-code piece, we need an extra newline.
+ global extra_newline
+ extra_newline = True
+
+def is_empty(line): return not bool(line.strip())
+def is_comment(line): return line.startswith('#')
+def is_code(line): return not is_empty(line) and not is_comment(line)
+
+def consume(is_a, lines):
+ result = []
+ for l in lines:
+ if not is_a(l):
+ break
+ result.append(l)
+ return (result, lines[len(result):])
+
+def trim_until_main(lines):
+ main_tag = 'if __name__ == \'__main__\':'
+ return lines[:lines.index(main_tag)] if main_tag in lines else lines
+
+def run(script):
+ with open(script) as f:
+ lines = trim_until_main([l.strip() for l in f])
+ while len(lines):
+ comments, lines = consume(is_comment, lines)
+ if comments:
+ write(*comments)
+ loc, lines = consume(is_code, lines)
+ for c in loc:
+ code(c)
+ _, lines = consume(is_empty, lines)
+
+ global code_locals
+ interact('(Hit CTRL-D to exit...)', local=code_locals)

gcloud/storage/demo/__main__.py

@@ -0,0 +1,4 @@
+from gcloud import demo
+import os.path
+
+demo.run(os.path.join(os.path.dirname(__file__), 'script.py'))

gcloud/storage/demo/script.py

@@ -0,0 +1,39 @@
+# Welcome to the gCloud Storage Demo! (hit enter)
+
+# We're going to walk through some of the basics...,
+# Don't worry though. You don't need to do anything, just keep hitting enter...
+
+# Let's start by importing the demo module and getting a connection:
+from gcloud import demo
+connection = demo.get_storage_connection()
+
+# OK, now let's look at all of the buckets...
+print connection.get_all_buckets() # This might take a second...
+
+# Now let's create a new bucket...
+import time
+bucket_name = ("bucket-%s" % time.time()).replace(".", "") # Get rid of dots...
+print bucket_name
+bucket = connection.create_bucket(bucket_name)
+print bucket
+
+# Let's look at all of the buckets again...
+print connection.get_all_buckets()
+
+# How about we create a new key inside this bucket.
+key = bucket.new_key("my-new-file.txt")
+
+# Now let's put some data in there.
+key.set_contents_from_string("this is some data!")
+
+# ... and we can read that data back again.
+print key.get_contents_as_string()
+
+# Now let's delete that key.
+print key.delete()
+
+# And now that we're done, let's delete that bucket...
+print bucket.delete()
+
+# Alright! That's all!
+# Here's an interactive prompt for you now...