import pandas as pd
import mysql.connector
import re
import time
import os
import sys
from bs4 import BeautifulSoup as Soup

config = {
  'user': 'root',
  'password': 'Ilovekitty1',
  'host': '127.0.0.1',
  'database': 'comecondo',
  'raise_on_warnings': True
}

os.chdir('/home/gavin/workspace/comecondo/python/')
if len(sys.argv) != 4:
    print("Invalid Parameters")
else:
    target_page = sys.argv[1]
    tag         = sys.argv[2]
    content     = sys.argv[3]
    print("Page:"+target_page+" Tag:"+tag)
    df     = pd.read_csv(r'page_links.csv')
    tag_df = df.loc[df['tag'] == tag]
#    print (tag_df['page'])
    
    if content=='wordpress':
        cnx = mysql.connector.connect(host="localhost", user="root", password="Ilovekitty1", database="comecondo")
        dbcur = cnx.cursor()
        # Update wp_post
        query = "select post_content from wp_posts where id="+target_page
        dbcur.execute(query)
        result = dbcur.fetchall()
        for row in result:
#           print (row[0])
            new_content = row[0]
            soup = Soup(new_content, 'html.parser')
            ul = soup.find('ul')
            for link in tag_df['page']:
#               print (link)
                html_link = Soup(link,'html.parser')
                ul.insert(0,html_link)
#               print (ul)
#               print(soup)
            with open('new_file.html', 'w') as f:
                f.write(str(soup))
            update_query = "update wp_posts set post_content='"+str(soup)+"' where id="+target_page+";"
            print (update_query)
            dbcur.execute(update_query)
        cnx.commit()
        cnx.close()
    
    if content=='elementor':
        cnx = mysql.connector.connect(host="localhost", user="root", password="Ilovekitty1", database="comecondo")
        dbcur = cnx.cursor()
        query = "select meta_value from wp_postmeta where meta_key='_elementor_data' and post_id="+target_page+";"
        dbcur.execute(query)
        result = dbcur.fetchall()
        for row in result:
#           print (row[0])
            new_content = row[0]
            soup = Soup(new_content, 'html.parser')
            ul = soup.find('ul')
            for link in tag_df['page']:
#                html_link = Soup(link.encode('unicode-escape'),'html.parser')
                html_link = Soup(link,'html.parser')
#                print (html_link)
                ul.insert(0,html_link)
#               print (ul)
#               print(soup)
            update_query = "update wp_postmeta set meta_value='"+str(soup).replace("'","''")+"' where meta_key='_elementor_data' and post_id="+target_page+";"
            print (update_query)
            dbcur.execute(update_query)
        cnx.commit()
        cnx.close()