import argparse

from pybamboo.connection import Connection
from pybamboo.dataset import Dataset


DEV_BAMBOO_URL = 'http://dev.bamboo.io/'


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('-d', '--dataset', help='The dataset ID to migrate')
    args = parser.parse_args()


def main():
    dataset_url = "http://bamboo.io/datasets/%s.csv" % args.dataset

    dev_connect = Connection(url=DEV_BAMBOO_URL)

    dataset = Dataset(url=dataset_url, connection=dev_connect)
    print dataset.id


main()

@@ -14,51 +14,8 @@
 se%0A%0A
-from pybamboo.connection import Connection%0A
 from
@@ -49,27 +49,27 @@
 taset%0A%0A%0A
-DEV_
 BAMBOO_
+DEV_
 URL = 'h
@@ -303,25 +303,10 @@
  = %22
-http://bamboo.io/
+%25s
 data
@@ -324,69 +324,37 @@
 %22 %25 
-args.dataset%0A%0A    dev_connect = Connection(url=DEV_BAMBOO_URL
+(BAMBOO_DEV_URL, args.dataset
 )%0A%0A 
@@ -393,32 +393,8 @@
 _url
-, connection=dev_connect
 )%0A  

update script for encoded on prod