Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 4 additions & 1 deletion .gitignore
Original file line number Diff line number Diff line change
@@ -1,7 +1,10 @@
# Byte-compiled / optimized / DLL files
# Byte-compiled / optimized / DLL / Data files
__pycache__/
*.py[cod]
*$py.class
*.txt
*.xlsx
data/oews/raw_data

# C extensions
*.so
Expand Down
21 changes: 21 additions & 0 deletions data/oews/loader/loader.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
from data.onet.models.mapping import *
from data.oews.loader.mapping import *

from motor.motor_asyncio import AsyncIOMotorClient
from beanie import init_beanie
import asyncio


async def load_mapping():
await load_job_salary()


async def load_oews_data():
client = AsyncIOMotorClient('mongodb+srv://uofteceelcano:Dd8TbR5VaPkzuFVU@elcanocluster.bwllguc.mongodb.net/?retryWrites=true&w=majority')
await init_beanie(database=client['lighthouse'], document_models=[Mapping])

await load_mapping()


if __name__ == "__main__":
asyncio.run(load_oews_data())
29 changes: 29 additions & 0 deletions data/oews/loader/mapping.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
import pandas as pd
from data.oews.models.mapping import *

async def load_job_salary():
base = 'data/oews/raw_data/'
df = pd.read_excel(base + 'oesm22all/all_data_M_2022.xlsx')
df.replace(['*', '#', '**', '~'], None, inplace=True)

items = []
for _, row in df.iterrows():
# there are repetitioins in the OCC_TITLE column with only different O_GROUP
try:
if row['OCC_TITLE'] == items[-1].job_title:
continue
except:
pass

item = Mapping(
src_element_id=row['OCC_CODE'],
job_title=row['OCC_TITLE'],
hourly_salary=row['H_MEAN'],
annual_salary=row['A_MEAN'],
hourly_median=row['H_MEDIAN'],
annual_median=row['A_MEDIAN'],
total_employment=row['TOT_EMP']
)
items.append(item)

await Mapping.insert_many(items)
18 changes: 18 additions & 0 deletions data/oews/models/mapping.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,18 @@
from beanie import Document
from typing import Optional

class Mapping(Document):
class Settings:
name = "job_salary"

src_element_id: str
job_title: str
hourly_salary: Optional[float]
annual_salary: Optional[int]
hourly_median: Optional[float]
annual_median: Optional[int]
total_employment: Optional[int]


def get_mapping_models():
return [Mapping]