ranger/orm/services/region_manager/revert_csv2db.py
Nicholas Jones 7667a7b9af Fix pep8 errors
Fixes existing pep8 errors and reenables the corresponding tox rules

Change-Id: I4168a90c40173e4c35c9d75030cd592ace657508
2017-08-07 15:46:47 -05:00

32 lines
719 B
Python

import csv
import logging
import config
from rms.storage.base_data_manager import SQLDBError
from rms.storage.my_sql.data_manager import DataManager
logger = logging.getLogger(__name__)
def revert_csv2db(data_manager):
logger.info('revert csv to db..')
try:
with open('rms/resources/regions.csv') as csvfile:
reader = csv.DictReader(csvfile)
for row in reader:
data_manager.delete_region(row["region_id"])
except SQLDBError as e:
logger.error("SQL error raised {}".format(e.message))
def main():
db_url = config.database['url']
data_manager = DataManager(db_url)
revert_csv2db(data_manager)
if __name__ == "__main__":
main()