Source code for superpyrate.db_setup
"""Sets up the tables in a newly created database, ready for data ingest
"""
from pyrate.repositories.aisdb import AISdb
from superpyrate.pipeline import get_environment_variable
[docs]def make_options():
options = {}
options['host'] = get_environment_variable('DBHOSTNAME')
options['db'] = get_environment_variable('DBNAME')
options['user'] = get_environment_variable('DBUSER')
options['pass'] = get_environment_variable('DBUSERPASS')
return options
[docs]def main():
options = make_options()
db = AISdb(options)
with db:
db.create()
db.clean.drop_indices()
if __name__ == '__main__':
main()