Working_with__Regular_Expressions.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "provenance": [],
      "authorship_tag": "ABX9TyMKFi8Yshpd8ZWMk0kvHEav",
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/luisosmx/Best-README-Template/blob/master/Working_with__Regular_Expressions.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "id": "OXxJXWZKA7am"
      },
      "outputs": [],
      "source": [
        "#!/usr/bin/env python3\n",
        "\n",
        "#Import libraries\n",
        "import re\n",
        "import csv\n",
        "\n",
        "def contains_domain(address, domain):\n",
        "  \"\"\"Returns True if the email address contains the given,domain,in the domain position, false if not.\"\"\"\n",
        "  domain = r'[\\w\\.-]+@'+domain+'$'\n",
        "  if re.match(domain,address):\n",
        "    return True\n",
        "  return False\n",
        "\n",
        "\n",
        "def replace_domain(address, old_domain, new_domain):\n",
        "  \"\"\"Replaces the old domain with the new domain in the received address.\"\"\"\n",
        "  old_domain_pattern = r'' + old_domain + '$'\n",
        "  address = re.sub(old_domain_pattern, new_domain, address)\n",
        "  return address\n",
        "\n",
        "def main():\n",
        "  \"\"\"Processes the list of emails, replacing any instances of the old domain with the new domain.\"\"\"\n",
        "  old_domain, new_domain = 'abc.edu', 'xyz.edu'\n",
        "  csv_file_location = '../data/user_emails.csv'\n",
        "  report_file = '../data' + '/updated_user_emails.csv'\n",
        "  user_email_list = []\n",
        "  old_domain_email_list = []\n",
        "  new_domain_email_list = []\n",
        "\n",
        "  with open(csv_file_location, 'r') as f:\n",
        "    user_data_list = list(csv.reader(f))\n",
        "    user_email_list = [data[1].strip() for data in user_data_list[1:]]\n",
        "\n",
        "    for email_address in user_email_list:\n",
        "      if contains_domain(email_address, old_domain):\n",
        "        old_domain_email_list.append(email_address)\n",
        "        replaced_email = replace_domain(email_address,old_domain,new_domain)\n",
        "        new_domain_email_list.append(replaced_email)\n",
        "\n",
        "    email_key = ' ' + 'Email Address'\n",
        "    email_index = user_data_list[0].index(email_key)\n",
        "\n",
        "    for user in user_data_list[1:]:\n",
        "      for old_domain, new_domain in zip(old_domain_email_list, new_domain_email_list):\n",
        "        if user[email_index] == ' ' + old_domain:\n",
        "          user[email_index] = ' ' + new_domain\n",
        "  f.close()\n",
        "\n",
        "  with open(report_file, 'w+') as output_file:\n",
        "    writer = csv.writer(output_file)\n",
        "    writer.writerows(user_data_list)\n",
        "    output_file.close()\n",
        "\n",
        "main()"
      ]
    },
    {
      "cell_type": "code",
      "source": [],
      "metadata": {
        "id": "A9uNL_guBGlD"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}