chore(bigtable): add bigtable samples

packages/google-cloud-bigtable/samples/AUTHORING_GUIDE.md

@@ -0,0 +1 @@
+See https://github.com/GoogleCloudPlatform/python-docs-samples/blob/main/AUTHORING_GUIDE.md

packages/google-cloud-bigtable/samples/CONTRIBUTING.md

@@ -0,0 +1 @@
+See https://github.com/GoogleCloudPlatform/python-docs-samples/blob/main/CONTRIBUTING.md

packages/google-cloud-bigtable/samples/README.md

@@ -0,0 +1,24 @@
+[//]: # "This README.md file is auto-generated, all changes to this file will be lost."
+[//]: # "To regenerate it, use `python -m synthtool`."
+
+## Python Samples for Cloud Bigtable
+
+This directory contains samples for Cloud Bigtable, which may be used as a refererence for how to use this product. 
+
+## Additional Information
+
+You can read the documentation for more details on API usage and use GitHub
+to <a href="https://github.com/googleapis/python-bigtable">browse the source</a> and [report issues][issues].
+
+### Contributing
+View the [contributing guidelines][contrib_guide], the [Python style guide][py_style] for more information.
+
+[authentication]: https://cloud.google.com/docs/authentication/getting-started
+[enable_billing]:https://cloud.google.com/apis/docs/getting-started#enabling_billing
+[client_library_python]: https://googlecloudplatform.github.io/google-cloud-python/
+[issues]: https://github.com/GoogleCloudPlatform/google-cloud-python/issues
+[contrib_guide]: https://github.com/googleapis/google-cloud-python/blob/main/CONTRIBUTING.rst
+[py_style]: http://google.github.io/styleguide/pyguide.html
+[cloud_sdk]: https://cloud.google.com/sdk/docs
+[gcloud_shell]: https://cloud.google.com/shell/docs
+[gcloud_shell]: https://cloud.google.com/shell/docs

packages/google-cloud-bigtable/samples/beam/hello_world_write.py

@@ -0,0 +1,70 @@
+# Copyright 2020 Google Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import datetime
+
+import apache_beam as beam
+from apache_beam.io.gcp.bigtableio import WriteToBigTable
+from apache_beam.options.pipeline_options import PipelineOptions
+
+from google.cloud.bigtable import row
+
+
+class BigtableOptions(PipelineOptions):
+    @classmethod
+    def _add_argparse_args(cls, parser):
+        parser.add_argument(
+            "--bigtable-project",
+            help="The Bigtable project ID, this can be different than your "
+            "Dataflow project",
+            default="bigtable-project",
+        )
+        parser.add_argument(
+            "--bigtable-instance",
+            help="The Bigtable instance ID",
+            default="bigtable-instance",
+        )
+        parser.add_argument(
+            "--bigtable-table",
+            help="The Bigtable table ID in the instance.",
+            default="bigtable-table",
+        )
+
+
+class CreateRowFn(beam.DoFn):
+    def process(self, key):
+        direct_row = row.DirectRow(row_key=key)
+        direct_row.set_cell(
+            "stats_summary", b"os_build", b"android", datetime.datetime.now()
+        )
+        return [direct_row]
+
+
+def run(argv=None):
+    """Build and run the pipeline."""
+    options = BigtableOptions(argv)
+    with beam.Pipeline(options=options) as p:
+        (
+            p
+            | beam.Create(["phone#4c410523#20190501", "phone#4c410523#20190502"])
+            | beam.ParDo(CreateRowFn())
+            | WriteToBigTable(
+                project_id=options.bigtable_project,
+                instance_id=options.bigtable_instance,
+                table_id=options.bigtable_table,
+            )
+        )
+
+
+if __name__ == "__main__":
+    run()

packages/google-cloud-bigtable/samples/beam/hello_world_write_test.py

@@ -0,0 +1,48 @@
+# Copyright 2020 Google Inc.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import os
+import uuid
+
+import pytest
+
+from ..utils import create_table_cm
+from . import hello_world_write
+
+PROJECT = os.environ["GOOGLE_CLOUD_PROJECT"]
+BIGTABLE_INSTANCE = os.environ["BIGTABLE_INSTANCE"]
+TABLE_ID = f"mobile-time-series-beam-{str(uuid.uuid4())[:16]}"
+
+
+@pytest.fixture(scope="module", autouse=True)
+def table():
+    with create_table_cm(
+        PROJECT, BIGTABLE_INSTANCE, TABLE_ID, {"stats_summary": None}
+    ) as table:
+        yield table
+
+
+def test_hello_world_write(table):
+    hello_world_write.run(
+        [
+            "--bigtable-project=%s" % PROJECT,
+            "--bigtable-instance=%s" % BIGTABLE_INSTANCE,
+            "--bigtable-table=%s" % TABLE_ID,
+        ]
+    )
+
+    rows = table.read_rows()
+    count = 0
+    for _ in rows:
+        count += 1
+    assert count == 2