ranger/orm/services/region_manager/revert_csv2db.py
hosingh000 87d644675e Merge Python3 branch
This change is going to upgrade ranger to use Python 3.x

Change-Id: I563661e071c56c2df7e0e1a6e365aecd4158b6cd
2019-10-29 10:36:00 -05:00

32 lines
728 B
Python

import csv
import logging
from . 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()