hsine.py

#!/usr/bin/python

from distance import Distance
from file import File
import argparse
import sys, os

# mean earth radius - https://en.wikipedia.org/wiki/Earth_radius#Mean_radius
EARTH = 6371.0088
OFFICE = (53.339428, -6.257664)
URL="https://s3.amazonaws.com/intercom-take-home-test/customers.txt"

class Hsine:

    def __init__(self, earth, office, url):
        self.earth = earth
        self.office = office
        self.url = url

    #Sorts filtered guest list
    def sort_list(self,l):
        l.sort(key = lambda x : x["user_id"])
        return l

    #Returns sorted by ID and filtered by distance guest list
    def make_glist(self,k,l):
        guest_list = []
        for i in range(len(l)):
            km = k.get_km(l[i]["latitude"],l[i]["longitude"])
            if km <= 100:
                s={"name":l[i]["name"],"user_id":l[i]["user_id"]} #,"K": km}
                guest_list.append(s)
        return self.sort_list(guest_list)

    #Save Glist to output.txt
    def save_file(self,f):
        with open('output.txt','w') as writeme:
            for l in range(len(f)):
                writeme.write('%s\n' % f[l])

    def main(self):
        print("Start \n")
        k = Distance(self.earth,self.office)
        f = File(self.url)
        list = f.parse_file()
        guest_list = self.make_glist(k,list)
        self.save_file(guest_list)
        for i in range(len(guest_list)):
            print(guest_list[i])
        print("\nFinish \nPlease check output.txt file too.")

if __name__ == "__main__":

    def getOptions(args=sys.argv[1:]):
        parser = argparse.ArgumentParser(description="Hsine Menu")
        parser.add_argument("-t", "--test", dest='t', action="store_true", help="Runs Unit Test")
        opt = parser.parse_args()
        return opt

    try:
        opt = getOptions(sys.argv[1:])
        if opt.t:
            print("UniTest Module")
            os.system('python3 test_hsine.py')
        else:
            a = Hsine(EARTH, OFFICE, URL)
            a.main()
    except KeyboardInterrupt:
        print("Sajonara!")
        sys.exit(0)