#!/usr/bin/env python

import sys

from oslo.config import cfg

from billingstack.openstack.common import log as logging

from billingstack import service

from billingstack.samples import get_samples

from billingstack.storage import get_connection

from billingstack.openstack.common.context import get_admin_context

from billingstack.storage.impl_sqlalchemy import models

cfg.CONF.import_opt('storage_driver', 'billingstack.central',

group='service:central')

cfg.CONF.import_opt('database_connection', 'billingstack.storage.impl_sqlalchemy',

group='storage:sqlalchemy')

SAMPLES = get_samples()

def get_fixture(name, fixture=0, values={}):

f = SAMPLES[name][fixture].copy()

f.update(values)

return f

if __name__ == '__main__':

service.prepare_service(sys.argv)

conn = get_connection()

samples = get_samples()

ctxt = get_admin_context()

currencies = {}

for c in samples['currency']:

currencies[c['name']] = conn.currency_add(ctxt, c)

languages = {}

for l in samples['language']:

languages[l['name']] = conn.language_add(ctxt, l)

for method in samples['pg_method']:

conn.pg_method_add(ctxt, method)

country_data = {

"currency_name": currencies['nok']['name'],

"language_name": languages['nor']['name']}

merchant = conn.merchant_add(

ctxt, get_fixture('merchant', values=country_data))

customer = conn.customer_add(

ctxt, merchant['id'], get_fixture('customer', values=country_data))

contact_info = get_fixture('contact_info')

merchant_user = get_fixture('user')

merchant_user['username'] = 'demo_merchant'

merchant_user['contact_info'] = contact_info

merchant_user = conn.user_add(

ctxt, merchant['id'], merchant_user)

customer_user = get_fixture('user')

customer_user['username'] = 'demo_customer'

customer_user['contact_info'] = contact_info

customer_user['customer_id'] = customer['id']

customer_user = conn.user_add(

ctxt,

merchant['id'],

customer_user)

products = {}

for p in samples['product']:

products[p['name']] = conn.product_add(ctxt, merchant['id'], p)

values = {

'plan_items': [

{'product_id': products['memory']},

{'product_id': products['vcpus']},

{'product_id': products['root_disk_size']},

{'product_id': products['network.incoming.bytes']},

{'product_id': products['network.outgoing.bytes']}

]}

plan = get_fixture('plan', values=values)

conn.plan_add(ctxt, merchant['id'], get_fixture('plan'))