C2M Scrape add

This commit is contained in:
Yam ZhengLim
2024-01-27 14:46:57 +08:00
parent e53a9817a6
commit 4ff602f2c6
9 changed files with 280 additions and 36 deletions
+12
View File
@@ -57,6 +57,7 @@ data-management
├─ .gitignore # Ignore the file to push into gitlab
├─ config # Folder store all the config file to use in the code
│ ├─ C2M_CONFIG.json # C2M webscraping configuration
│ ├─ CLOUD-STORAGE-ADMIN-CREDENTIAL.json # Get from GCP Service Account cloud-storage-admin-service-account
│ ├─ CRISP_CONFIG.json # Config for CRISP API key, secret & web-id
│ ├─ EXCHANGE_CONFIG.json # Manually config the MySQL ID, Password, etc...
@@ -64,6 +65,17 @@ data-management
│ ├─ SHIPPING_CONFIG.json # Manually config the MySQL ID, Password, etc...
│ └─ SHIPPING_EXTRACT_SQL.json # SQL use in MYSQL to extract the data we need to save in temp's csv
├─ c2m # C2M webscrapping
│ ├─ scrape # All script use to scrape the competitor data is in here
│ │ ├─ c2m_scrape_app.py # C2M webscrap, call in main.py or seperate run it in your local environment
│ │ ├─ c2m_temp_scrape_files # Temp folder to store all the csv or error.png from webscraper (Auto Create)
│ │ │ └─ .gitkeep # Keep empty folder
│ │ └─ local_run_files # Folder to store all the data or error from webscraper which run at local pc
│ │ └─ error # Folder to store error.png from webscraper which run at local pc
│ │ └─ .gitkeep # Keep empty folder
│ └─ load
│ └─ c2m_load_function.py # C2M load code function to import in main.py
├─ crisp # CRISP-CRM
│ ├─ extract # All crisp extract needed code is in here
│ │ ├─ crisp_extract_function.py # Crisp extract code function to import in main.py
+21
View File
@@ -0,0 +1,21 @@
from google.cloud import storage
def login_cloudstorage_credential(credential):
try:
storage_client = storage.Client.from_service_account_json(credential)
return storage_client
except Exception as e:
print(e)
print("[FUNCTION_ERROR]-login_cloudstorage_credential")
return False
def upload_to_bucket(storage_client, blob_name, file_path):
try:
crisp_bucket = storage_client.get_bucket('exchange_competitor_bucket_raw')
blob = crisp_bucket.blob(blob_name)
blob.upload_from_filename(file_path)
return print("[Upload Complete]",file_path)
except Exception as e:
print(e)
print("[FUNCTION_ERROR]-upload_to_bucket",blob_name)
return False
+152
View File
@@ -0,0 +1,152 @@
from playwright.sync_api import Playwright, sync_playwright
import re
import time
import csv
import json
import os
import pandas as pd
from datetime import datetime
def run(playwright: Playwright, DATA_PATH, ERROR_PATH, CONFIG) -> None:
try:
dt = datetime.now()
date = dt.strftime('%Y-%m-%d')
hour = dt.hour
RANGE = CONFIG['range']
PRODUCT_NAME = CONFIG['product_name']
SITE_NAME = CONFIG['site_name']
browser = playwright.chromium.launch(headless=False)
context = browser.new_context()
page = context.new_page()
page.goto(CONFIG["base_url"])
page.click("text=选择语言")
with page.expect_navigation():
page.click("text=English (US)")
page.click("text=Please login to your account or register an account for free. >> a")
page.click("input[name=\"login_id\"]")
page.fill("input[name=\"login_id\"]", "yamzhenglim@gmail.com")
page.click("input[name=\"login_pass\"]")
page.fill("input[name=\"login_pass\"]", "Abc_123456")
with page.expect_navigation():
page.click("button:has-text(\"Login\")")
page.click(":nth-match(:text(\"Request Service\"), 2)")
time.sleep(2)
page.eval_on_selector('.btn-agree', '$(".btn-agree").button( "option", "disabled", true | false )')
page.click("text=I Agree")
for option in CONFIG['options']: # Not Implemented "ALLINK", "GH", "NH", "OB"
try:
option_selector = page.locator("select[name=\"reload[acc_type]\"]")
option_selector.select_option(option)
data = []
for i in range (RANGE['start'], RANGE['stop']+1, RANGE['step']):
d = {}
myr_selector = '#reload_myr_amount'
myr_locator = page.locator(myr_selector)
myr_locator.clear()
myr_locator.fill(f'{i}')
page.press("text=Transaction Amount", 'Tab')
time.sleep(0.5)
myr = page.evaluate('$("#reload_myr_amount").val()')
cny = page.evaluate('$("#reload_cny_amount").val()')
q_staff = page.query_selector('.lime')
if q_staff:
staff_status = q_staff.inner_text().strip()
else:
staff_status = 'Offline'
orders_selector = page.query_selector('.in-processing')
if orders_selector:
orders = re.findall('\d+', orders_selector.inner_text())
svc = page.evaluate('$("#service_charge").val()')
total_payable = page.evaluate('$("#total_amount").val()')
current_rate = page.query_selector('#ex_rate').text_content()
# today_rate = page.query_selector('.today_rate').text_content()
# last_rate_update = page.query_selector('span:below(.today_rate)').text_content()
last_rate_update = page.query_selector('.text-muted.help-inline').text_content().strip()
d['product_name'] = PRODUCT_NAME
d['site_name'] = SITE_NAME
d['service_type'] = option # need confirmation / add to config
d['pending_orders'] = int(orders[0])
d['CNY'] = cny
d['MYR'] = myr
d['handeling_fee'] = svc
d['payable'] = total_payable
d['current_rate'] = current_rate
# d['day_rate'] = today_rate
d['last_rate_update'] = last_rate_update
d['staff_status'] = staff_status
d['scrape_date'] = datetime.now().strftime('%Y-%m-%d')
d['scrape_hour'] = datetime.now().strftime('%H')
data.append(d)
df = pd.DataFrame(data)
except Exception as e:
print(e)
stamp = datetime.now().strftime('%Y_%m_%d')
page.screenshot(path = f"{ERROR_PATH}/ERROR_{stamp}.png")
df.to_csv(f"{DATA_PATH}/{option}_{date}_{hour}.csv", index=False)
except Exception as e:
print(e)
stamp = datetime.now().strftime('%Y_%m_%d')
page.screenshot(path = f"{ERROR_PATH}/ERROR_{stamp}.png")
if __name__ == '__main__':
# Define a dictionary
conf_dic = {
"base_url": "https://www.c2m.my/reload.php",
"product_name": "Exchange",
"site_name": "c2m",
"range":{
"start":1,
"stop":100001,
"step":1000
},
"options":["TBDF", "ALTR"]
}
# Convert the dictionary to JSON format
conf = json.dumps(conf_dic)
CONFIG = json.loads(conf)
DATA_PATH = "./c2m/scrape/local_run_files/"
ERROR_PATH ="./c2m/scrape/local_run_files/error/"
os.makedirs(DATA_PATH, exist_ok=True)
with sync_playwright() as playwright:
run(playwright, DATA_PATH, ERROR_PATH, CONFIG)
else:
def run_app(C2M_FILES_TEMP_STORAGE_PATH, C2M_CONFIG):
# dt = datetime.now()
# date = dt.strftime('%Y-%m-%d')
# hour = dt.hour
DATA_PATH = C2M_FILES_TEMP_STORAGE_PATH
ERROR_PATH = C2M_FILES_TEMP_STORAGE_PATH
os.makedirs(DATA_PATH, exist_ok=True)
with sync_playwright() as playwright:
run(playwright, DATA_PATH, ERROR_PATH, C2M_CONFIG)
+11
View File
@@ -0,0 +1,11 @@
{
"base_url": "https://www.c2m.my/reload.php",
"product_name": "Exchange",
"site_name": "c2m",
"range":{
"start":1,
"stop":100001,
"step":1000
},
"options":["TBDF", "ALTR"]
}
+7
View File
@@ -22,6 +22,13 @@ def delete_csv_in_path(path):
os.unlink(path)
print("[Delete Temp Complete]", path)
def delete_csv_or_png_in_path(path):
for folder, subfolders, files in os.walk(path):
for file in files:
if file.endswith('.csv') or file.endswith('.png'):
path = os.path.join(folder, file)
os.unlink(path)
print("[Delete Temp Complete]", path)
def change_to_current_directory():
script_dir = os.path.dirname(os.path.realpath(__file__))
+75 -35
View File
@@ -8,7 +8,8 @@ import shipping.extract.shipping_extract_function as shipping_extract_func
import shipping.load.shipping_load_function as shipping_load_func
import crisp.extract.crisp_extract_function as crisp_extract_func
import crisp.load.crisp_load_function as crisp_load_func
import c2m.scrape.c2m_scrape_app as c2m_scrape_app
import c2m.load.c2m_load_function as c2m_load_func
def exchange_extract(EXCHANGE_CSV_TEMP_STORAGE_PATH):
#Config file name
@@ -127,52 +128,91 @@ def crisp_load_to_cloudstorage(CRISP_CSV_TEMP_STORAGE_PATH):
file_path = os.path.join(dirpath,file_name)
crisp_load_func.upload_to_bucket(cirsp_storage_client, file_name, file_path)
def c2m_scrape(C2M_FILES_TEMP_STORAGE_PATH):
C2M_CONFIG_FILE = "C2M_CONFIG.json"
C2M_FILES_TEMP_STORAGE_PATH = C2M_FILES_TEMP_STORAGE_PATH
#Read config file
C2M_CONFIG = general_function.read_json_file("./config/",C2M_CONFIG_FILE)
#Run the c2m_scrape_app
c2m_scrape_app.run_app(C2M_FILES_TEMP_STORAGE_PATH, C2M_CONFIG)
def c2m_load_to_cloudstorage(C2M_FILES_TEMP_STORAGE_PATH):
#Config file name
CLOUDSTORAGE_CREDENTIAL_FILE = "./config/CLOUD_STORAGE_ADMIN_CREDENTIAL.json"
#Connect to cloudstorage
c2m_storage_client = c2m_load_func.login_cloudstorage_credential(CLOUDSTORAGE_CREDENTIAL_FILE)
#Get the filename and path in temp csv storage
temp_file_list = []
for (dirpath, dirnames, filenames) in walk(C2M_FILES_TEMP_STORAGE_PATH):
temp_file_list.extend(filenames)
#Loop all the file and upload to cloudstorage
for file_name in temp_file_list:
file_path = os.path.join(dirpath,file_name)
c2m_load_func.upload_to_bucket(c2m_storage_client, file_name, file_path)
def main():
print("RUNNING MAIN_PY")
# Exchange
#Make sure cron in the file directory
general_function.change_to_current_directory()
#Extract Load Exchange Data
EXCHANGE_CSV_TEMP_STORAGE_PATH = "./exchange/extract/exchange_temp_extract_csv/"
print("STARTING EXCHANGE EXTRACT")
exchange_extract(EXCHANGE_CSV_TEMP_STORAGE_PATH)
print("STARTING EXCHANGE LOAD")
exchange_load_to_cloudstorage(EXCHANGE_CSV_TEMP_STORAGE_PATH)
#Delete Temp CSV
print("DELETING EXCHANGE CSV")
general_function.delete_csv_in_path(EXCHANGE_CSV_TEMP_STORAGE_PATH)
# # Exchange
# #Make sure cron in the file directory
# general_function.change_to_current_directory()
# #Extract Load Exchange Data
# EXCHANGE_CSV_TEMP_STORAGE_PATH = "./exchange/extract/exchange_temp_extract_csv/"
# print("STARTING EXCHANGE EXTRACT")
# exchange_extract(EXCHANGE_CSV_TEMP_STORAGE_PATH)
# print("STARTING EXCHANGE LOAD")
# exchange_load_to_cloudstorage(EXCHANGE_CSV_TEMP_STORAGE_PATH)
# #Delete Temp CSV
# print("DELETING EXCHANGE CSV")
# general_function.delete_csv_in_path(EXCHANGE_CSV_TEMP_STORAGE_PATH)
# Shipping
# # Shipping
# #Make sure cron in the file directory
# general_function.change_to_current_directory()
# #Extract Load SHIPPING Data
# SHIPPING_CSV_TEMP_STORAGE_PATH = "./shipping/extract/shipping_temp_extract_csv/"
# print("STARTING SHIPPING EXTRACT")
# shipping_extract(SHIPPING_CSV_TEMP_STORAGE_PATH)
# print("STARTING SHIPPING LOAD")
# shipping_load_to_cloudstorage(SHIPPING_CSV_TEMP_STORAGE_PATH)
# #Delete Temp CSV
# print("DELETING SHIPPING CSV")
# general_function.delete_csv_in_path(SHIPPING_CSV_TEMP_STORAGE_PATH)
# C2M
#Make sure cron in the file directory
general_function.change_to_current_directory()
#Extract Load SHIPPING Data
SHIPPING_CSV_TEMP_STORAGE_PATH = "./shipping/extract/shipping_temp_extract_csv/"
print("STARTING SHIPPING EXTRACT")
shipping_extract(SHIPPING_CSV_TEMP_STORAGE_PATH)
print("STARTING SHIPPING LOAD")
shipping_load_to_cloudstorage(SHIPPING_CSV_TEMP_STORAGE_PATH)
#Scrape C2M Data
C2M_FILES_TEMP_STORAGE_PATH = "./c2m/scrape/c2m_temp_scrape_files/"
print("STARTING C2M SCRAPE")
c2m_scrape(C2M_FILES_TEMP_STORAGE_PATH)
print("STARTING C2M LOAD")
c2m_load_to_cloudstorage(C2M_FILES_TEMP_STORAGE_PATH)
#Delete Temp CSV
print("DELETING SHIPPING CSV")
general_function.delete_csv_in_path(SHIPPING_CSV_TEMP_STORAGE_PATH)
general_function.delete_csv_or_png_in_path(C2M_FILES_TEMP_STORAGE_PATH)
# CRISP
#Make sure cron in the file directory
general_function.change_to_current_directory()
#Extract Load Exchange Data
CRISP_CSV_TEMP_STORAGE_PATH = "./crisp/extract/crisp_temp_extract_csv/"
print("STARTING CRISP EXTRACT")
crisp_extract(CRISP_CSV_TEMP_STORAGE_PATH)
print("STARTING CRISP LOAD")
crisp_load_to_cloudstorage(CRISP_CSV_TEMP_STORAGE_PATH)
#Delete Temp CSV
print("DELETING CRISP CSV")
general_function.delete_csv_in_path(CRISP_CSV_TEMP_STORAGE_PATH)
# # CRISP
# #Make sure cron in the file directory
# general_function.change_to_current_directory()
# #Extract Load Exchange Data
# CRISP_CSV_TEMP_STORAGE_PATH = "./crisp/extract/crisp_temp_extract_csv/"
# print("STARTING CRISP EXTRACT")
# crisp_extract(CRISP_CSV_TEMP_STORAGE_PATH)
# print("STARTING CRISP LOAD")
# crisp_load_to_cloudstorage(CRISP_CSV_TEMP_STORAGE_PATH)
# #Delete Temp CSV
# print("DELETING CRISP CSV")
# general_function.delete_csv_in_path(CRISP_CSV_TEMP_STORAGE_PATH)
print("END MAIN_PY")
print("------------------------------------------------------------------------------------------------")
# print("END MAIN_PY")
# print("------------------------------------------------------------------------------------------------")
if True:
main()
+2 -1
View File
@@ -1,4 +1,5 @@
mysql-connector-python==8.0.29
google-cloud-storage==2.5.0
crisp-api==1.1.13
pandas==1.5.2
pandas==1.5.2
playwright==1.34.0