Skip to content

Changed output file location and added encoding='utf-8' on file write. #8

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 4 commits into from
Apr 21, 2024
Merged

Changed output file location and added encoding='utf-8' on file write. #8

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
5e84cdd
REFACTOR: made output files to be created in separater folder
sebakthapa Apr 20, 2024
8574ced
ADD: added gitignore to ignore the output directory
sebakthapa Apr 20, 2024
f6f983f
REFACTOR: added utf-8 mode to write files
sebakthapa Apr 20, 2024
5e5fdf7
Merge branch 'main' into refactor
sebakthapa Apr 21, 2024
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: 5 additions & 0 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -99,3 +99,8 @@ venv_*/
.vimrc

mypy.ini
# Output Directories
output
result
results
outputs
15 changes: 14 additions & 1 deletion school_center.py
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
OUTPUT_DIR = 'results/'

PREF_DISTANCE_THRESHOLD = 2 # Preferred threshold distance in kilometers
ABS_DISTANCE_THRESHOLD = 7 # Absolute threshold distance in kilometers
MIN_STUDENT_IN_CENTER = 10 # minimum number of students from a school to be assigned to a center in normal circumstances
Expand All @@ -8,9 +10,18 @@
import csv
import random
import argparse
import os
from typing import Dict, List


def create_dir(dirPath:str):
"""
Create the given directory if it doesn't exists
- Creates all the directories needed to resolve to the provided directory path
"""
if not os.path.exists(dirPath):
os.makedirs(dirPath)

def haversine_distance(lat1, lon1, lat2, lon2):
"""
Calculate the great circle distance between two points
Expand Down Expand Up @@ -142,7 +153,9 @@ def is_allocated(scode1: str, scode2:str) -> bool:
remaining = 0 # stores count of non allocated students
allocations = {} # to track mutual allocations

with open('school-center-distance.tsv', 'w') as intermediate_file, open(args.output, 'w') as a_file:
create_dir(OUTPUT_DIR) # Create the output directory if not exists
with open('{}school-center-distance.tsv'.format(OUTPUT_DIR), 'w', encoding='utf-8') as intermediate_file, \
open(OUTPUT_DIR + args.output, 'w', encoding='utf-8') as a_file:
writer = csv.writer(intermediate_file, delimiter="\t")
writer.writerow(["scode", "s_count", "school_name", "school_lat", "school_long", "cscode", "center_name", "center_address", "center_capacity", "distance_km"])

Expand Down