source: TOOLS/ConsoGENCI/trunk/bin/init_alloc_tbl.py @ 2783

#!/usr/bin/env python
# -*- coding: utf-8 -*-

# ==================================================================== #
# Author: Sonia Labetoulle                                             #
# Contact: sonia.labetoulle _at_ ipsl.jussieu.fr                       #
# Created: 2016                                                        #
# History:                                                             #
# Modification:                                                        #
# ==================================================================== #

# ==================================================================== #
# ssh readonly@prodiguer-test-db.ipsl.upmc.fr                          #
# psql -U prodiguer_db_user prodiguer                                  #
#                                                                      #
# ssh readonly@prodiguer-test-db.ipsl.upmc.fr -L 5432:localhost:5432   #
# ==================================================================== #

# this must come first
from __future__ import print_function, unicode_literals, division

# standard library imports

# Application library imports
import libconsodb as cdb
import db_data


#######################################################################
if __name__ == "__main__":

  project_list = [
    {
      "name": "gencmip6",
      "centre": "tgcc",
      "machine": "curie",
      "node": "thin/standard",
      "alloc": 5000000,
      "start": "2015-01-01 00:00:00",
      "end": "2015-06-30 23:59:59",
    },
    {
      "name": "gencmip6",
      "centre": "tgcc",
      "machine": "curie",
      "node": "thin/standard",
      "alloc": 10000000,
      "start": "2015-07-01 00:00:00",
      "end": "2015-12-31 23:59:59",
    },
    {
      "name": "gencmip6",
      "centre": "tgcc",
      "machine": "curie",
      "node": "thin/standard",
      "alloc": 25000000,
      "start": "2016-01-01 00:00:00",
      "end": "2016-06-30 23:59:59",
    },
    {
      "name": "gencmip6",
      "centre": "tgcc",
      "machine": "curie",
      "node": "thin/standard",
      "alloc": 50000000,
      "start": "2016-07-01 00:00:00",
      "end": "2016-12-31 23:59:59",
    },
    {
      "name": "rgzi",
      "centre": "idris",
      "machine": "ada",
      "node": "standard",
      "alloc": 2000000,
      "start": "2015-01-01 00:00:00",
      "end": "2015-12-31 23:59:59",
    },
    {
      "name": "rgzi",
      "centre": "idris",
      "machine": "ada",
      "node": "standard",
      "alloc": 4000000,
      "start": "2016-01-01 00:00:00",
      "end": "2016-12-31 23:59:59",
    },
    {
      "name": "gen0826",
      "centre": "tgcc",
      "machine": "curie",
      "node": "fat/large",
      "alloc": 114000,
      "start": "2015-01-01 00:00:00",
      "end": "2015-12-31 23:59:59",
    },
    {
      "name": "gen0826",
      "centre": "tgcc",
      "machine": "curie",
      "node": "hybrid",
      "alloc": 25000,
      "start": "2015-01-01 00:00:00",
      "end": "2015-12-31 23:59:59",
    },
    {
      "name": "gen0826",
      "centre": "tgcc",
      "machine": "curie",
      "node": "thin/standard",
      "alloc": 114000,
      "start": "2015-01-01 00:00:00",
      "end": "2015-12-31 23:59:59",
    },
  ]

  conn, cursor = cdb.connect_db(
    db_data.db_host,
    db_data.db_name,
    db_data.db_user,
    db_data.db_pwd,
  )

  # Build request
  table_name = "conso.tbl_allocation"
  request = (
    "INSERT INTO " + table_name + " ("
    "project, centre, machine, node_type, total_hrs, "
    "start_date, end_date, row_create_date"
    ") VALUES "
  )

  lines_req = [
    (
      "('{name}', "
      "'{centre}', "
      "'{machine}', "
      "'{node}', "
      "{alloc}, "
      "'{start}', "
      "'{end}', "
      "{create})"
    ) .format(
      name=line["name"],
      centre=line["centre"],
      machine=line["machine"],
      node=line["node"],
      alloc=line["alloc"],
      start=line["start"],
      end=line["end"],
      create="CURRENT_TIMESTAMP",
    ) for line in project_list
  ]

  print(", ".join(lines_req))
  request = request + ", ".join(lines_req)
  print(request)

  cdb.insert_db(cursor, request)

  cdb.commit_db(conn)

  cdb.close_db(conn)

  exit(0)