def test_get_customers_to_invite(self):
expected = [
{"user_id": 2, "name": "b"},
{"user_id": 3, "name": "c"}
]
inviter = Inviter()
customer_test = ['{"latitude": "53.339428", "user_id": 2, "name": "b", "longitude": "-6.257664"}', # constants.DUBLIN_OFFICE_COORDINATES
'{"latitude": "53.339280", "user_id": 3, "name": "c", "longitude": "-6.281314"}', # Facebook Dublin
'{"latitude": "51.886170", "user_id": 1, "name": "a", "longitude": "-8.402208"}', # EMC Cork
'{"latitude": "-900", "user_id": 1, "name": "a", "longitude": ""}', # Bad Data]
'zzz{"latitude": "53.339428", "user_id": 2, "name": "b", "longitude": "-6.257664"}'] # Bad Data
with tempfile.TemporaryDirectory() as temp_dir:
f_path = os.path.join(temp_dir, 'customers.txt')
with open(f_path, 'w') as f:
f.write('\n'.join(customer_test))
actual = inviter.get_customers_to_invite(f_path)
self.assertEqual(actual, expected)
import argparse
import logging
from intercom_inviter.inviter import Inviter
logger = logging.getLogger(__name__)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--customers',
help='Path to text file of json formatted customers.',
type=str,
required=True)
args = parser.parse_args()
inviter = Inviter()
print(
'The following users are within 100k of the Intercom Dublin Office (Ordered by user_id).:'
)
for invitee in inviter.get_customers_to_invite(args.customers):
print(' ', end='')
print(invitee)