return None if __name__ == '__main__': rid = int(sys.argv[1]) if Player.objects.filter(sc2c_id=rid).count() > 0: print 'Did not import new player, already exists.' sys.exit(1) soup = BeautifulSoup(get_url(_base_url % rid)) tag = soup.findAll('h1')[0].contents[0].strip() try: country = soup.findAll(lambda t: (u'class', u'content_flag') in t.attrs)[0]['src'].split('/')[-1].split('.')[0].upper() except: country = '' race = sys.argv[2] print '------ From %s' % (_base_url % rid) k = raw_input('------ Add new player: %s (%s, %s)? ' % (tag, country, race)) if k.isdigit(): p = Player.objects.get(id=int(k)) p.sc2c_id = rid p.save() elif k.upper() == 'Y': p = Player() p.tag = tag p.country = country p.race = race p.sc2c_id = rid p.save()