def test_company_model_creation_2(self): c = Company.Company(*TestModels.test_company) db.session.add(c) db.session.commit() result = Company.find_by_id(100) # company was found assert (len(result) > 0)
def test_callingGetEmployeesReturnsListOfDicts(self): Employee.objects.filter = mock.MagicMock() company = Company() company.pk = 98 employee_one = Employee() employee_one.pk = 12 employee_one.user = User() employee_one.user.first_name = 'Fornavn' employee_one.user.last_name = 'Efternavn' employee_one.company = company employee_two = Employee() employee_two.pk = 13 employee_two.user = User() employee_two.user.first_name = 'Fornavn' employee_two.user.last_name = 'Efternavn' employee_two.company = company Employee.objects.filter.return_value = [employee_one, employee_two] employee_one.getMyEmployees = mock.MagicMock() employee_one.getMyEmployees.return_value = [] employee_list = employee_one.getEmployees() self.assertEquals(2, len(employee_list)) self.assertTrue(isinstance(employee_list[0], dict)) self.assertTrue(isinstance(employee_list[0]['manager'], Employee)) self.assertEquals(employee_list[0]['manager'].pk, 12) self.assertTrue(isinstance(employee_list[0]['employees'], list)) self.assertEquals(employee_list[0]['manager'].user.first_name, 'Fornavn') self.assertEquals(employee_list[0]['manager'].user.last_name, 'Efternavn')
def setUp(self): database_name = 'capstone_test' database_path = 'postgresql://jaimeaznar@{}/{}'.format( 'localhost:5432', database_name) self.app = create_app() self.app.config['WTF_CSRF_ENABLED'] = False self.client = self.app.test_client setup_db(self.app, database_path) db.session.close() db.drop_all() db.create_all() # mock company self.company = Company(name='test company', city='test city', state='test state', address='test address', phone='12345') self.company.insert() # mock product self.product = Product(name='test product', image='static/img/tomahawk.jpg', description='test description', company_id=1) self.product.insert()
def test_f_expression_update_attribute(self): # F expressions can be used to update attributes on single objects test_gmbh = Company.objects.get(name='Test GmbH') self.assertEqual(test_gmbh.num_employees, 32) test_gmbh.num_employees = F('num_employees') + 4 test_gmbh.save() test_gmbh = Company.objects.get(pk=test_gmbh.pk) self.assertEqual(test_gmbh.num_employees, 36) # F expressions cannot be used to update attributes which are # foreign keys, or attributes which involve joins. test_gmbh.point_of_contact = None test_gmbh.save() self.assertEqual(test_gmbh.point_of_contact, None) self.assertRaises(ValueError, setattr, test_gmbh, 'point_of_contact', F('ceo')) test_gmbh.point_of_contact = test_gmbh.ceo test_gmbh.save() test_gmbh.name = F('ceo__last_name') self.assertRaises(FieldError, test_gmbh.save) # F expressions cannot be used to update attributes on objects # which do not yet exist in the database acme = Company(name='The Acme Widget Co.', num_employees=12, num_chairs=5, ceo=test_gmbh.ceo) acme.num_employees = F('num_employees') + 16 self.assertRaises(TypeError, acme.save)
def update_company(self, company): self.logger.debug("update_company") self.logger.info("saving company to database") self.logger.debug("company: %s", company) Company.update( code=company.code, name=company.name, nif=company.nif, address=company.address, city=company.city, state=company.state, zip_code=company.zip_code, phone=company.phone, contact_person=company.contact_person, alternative_phone=company.alternative_phone, fax=company.fax, email=company.email, iban=company.iban, bank_name=company.bank_name, payment_type=company.payment_type, expiration_days=company.expiration_days, first_payment_day=company.first_payment_day, second_payment_day=company.second_payment_day, third_payment_day=company.third_payment_day ).execute() self.logger.info("updating company to model") tree_iter = self.get_iter_from_selected_row(company.code) self.companies_model.set( tree_iter, [self.CODE_COLUMN, self.NAME_COLUMN, self.NIF_COLUMN], [company.code, company.name, company.nif] )
def _create_new_company(company_name): """ Adds a new record for the given company name to our database. The plan is to use the crunchbase/glassdoor API to query for the given company name. If the calls return info on the company, happily add it to our database. If not, we'll assume the given company name is faulty, and return an appropriate message to the user. """ crunchbase = CrunchbaseAPI() response = crunchbase.get_company_info(company_name) if response['exists']: # Create the company using the name returned by crunchbase # TODO: We should add our own checks.. so a search for '*' doesn't give # back 'Google' # Or maybe we shouldn't let crunchbase spellcheck? company_name = response['name'].lower() try: # If company already exists, redirect to it company = Company.objects.get(name=company_name) redirect('/companies/{}'.format(new_company.id)) except Company.DoesNotExist: new_company = Company(name=company_name) new_company.save() logger.info('New company created: {}'.format(company_name)) return redirect('/companies/{}'.format(new_company.id)) else: return redirect('/')
def create(self, request, **kwargs): serializer = CompanySerializer(data=request.data) if serializer.is_valid(): user = save_user(request.data, UserPermissions.IS_COMPANY) city = City(**request.data["city"]) company = Company( name=request.data["name"], type=UserType.COMPANY, address=request.data["address"], identityDoc=request.data["identityDoc"], phone=request.data["phone"], city=city, user=user, ) company.save() return Response( { "status": "SUCCESS", "msg_status": "Estabecimiento creado satisfactoriamente. Puede ingresar con su nueva contrasena.", } ) else: messages = [] make_error(serializer.errors.values(), messages) return Response({"status": "FAILURE", "msg_status": messages})
def create_company(jwt): request_value = request.get_json() if request_value is None: abort(400) properties = ['name', 'description', 'imageBase64', 'state', 'city'] for prop in properties: if prop not in request_value: abort(400) if request_value['name'] is None: abort(400) if request_value['state'] is None or request_value['city'] is None: abort(400) company = Company( request_value['name'], request_value['description'], request_value['imageBase64'], request_value['state'], request_value['city'] ) company.insert() return jsonify({ 'status_code': 200, 'company': company.format(), 'message': 'The company was successfully created', 'success': True, })
def enterpriseInfoSave(req): """ 企业信息保存! :param req: :return: """ enterpriseType = req.POST["enterpriseType"] enterprise = Company.objects.filter(companyName=enterpriseType) if req.POST["radioVal"] == "1": isShow = True else: isShow = False if enterprise.exists(): return HttpResponse(2) companyId = uuid.uuid1() # 生成36位随机数 try: with transaction.atomic(): # 事务的使用,包含的内容一旦出错,则保存的信息全部撤回 company = Company(companyId=companyId, companyName=enterpriseType, isShow=isShow) company.save() i = 0 while i < len(req.POST) - 2: enterpriseInfoNum = "enterprise" + str(i) enterpriseInfo = req.POST[enterpriseInfoNum] dataType = DataType(dataTypeId=uuid.uuid1(), companyId_id=companyId, dataName=enterpriseInfo) dataType.save() i = i + 1 print enterpriseInfo return HttpResponse(1) except Exception as err: print err return HttpResponse(0)
def admin_add_company(): data = json.loads(request.data) linkedin_id = data.get('linkedin_id') name = data.get('name') crunchbase_url = data.get('crunchbase_url') if not linkedin_id or linkedin_id == "": linkedin_id = gen_fake_linkedin_id() company = Company(linkedin_id, name) company.crunchbase_url = crunchbase_url db.session.add(company) db.session.commit() print ' adding new company for rescraping: ' + name.encode('utf8') + ' --> ' + linkedin_id print ' new company id = ' + str(company.id) if crunchbase_url and crunchbase_url != "": mode = FROM_URL_RESCRAPE_MODE else: mode = DEFAULT_RESCRAPE_MODE rescrape_companies_from_list([company], mode=mode) # TODO F**K F**K F**K F**K HEROKU for not allowing > 1 thread # WHAT IN THE F*****G F**K..... #times_left = 100 #while not company.is_feed_ready and times_left >= 0: # times_left -= 1 # this is to make sure this doesn't become an infinite for-loop # time.sleep(5) # db.session.commit() # Retarded way to start a new transaction because F**K sqlalchemy... # print ' ... waiting for scraper callback --> is_feed_ready ' + str(company.is_feed_ready) resp = json.dumps({'status': 'ok', 'company_id': company.id}) return resp
def add_company(payload): body = request.json # Need to have name and website keys in body if not all([x in body for x in ['name', 'website']]): abort(422) # Here we want this to return None otherwise another company has that name # If you don't handle this way, function works, but unittests catches the # print statement below in the 'except Exception as e' block and clutters # up the unittests output. duplicate_name = Company.query.filter_by( name=body['name'].strip()).one_or_none() if duplicate_name: abort(422) duplicate_website = Company.query.filter_by( website=body['website'].strip()).one_or_none() if duplicate_website: abort(422) try: new_co = Company(name=body['name'].strip(), website=body['website'].strip()) new_co.insert() except Exception as e: print(f'Exception in add_company(): {e}') abort(422) # Syntax is good, can't process for semantic reasons return jsonify({"id": new_co.id, "success": True})
def extract_company_info(self, company_url, company_name, position): if not company_url: return res = self.send_requests(company_url) document = get_document(res.text) name = get_simple_dom(document, './/div[@class="tHeader tHCop"]/div[contains(@class, "in")]/h1/@title') # 使用前程无忧搭建自己公司门户的公司 if not name: return Company.insert(name=company_name, uniqueId=self.get_company_unique_id(company_url), position=position, companyUrl=company_url, site=self.task.site).execute() # 普通公司 base_info = get_simple_dom(document, './/p[@class="ltype"]/text()', '') infos = base_info.split('|') nature = infos[0].strip() if len(infos) > 0 else '' scale = infos[1].strip() if len(infos) > 1 else '' category = infos[2].strip() if len(infos) > 2 else '' position_info = get_richtext(document, './/div[@class="inbox"]/p[@class="fp"]') position_info = position_info.strip().replace('公司地址:', '') position = re.sub('(\(邮编:.*\))', '', position_info) or '' position = position.strip() description = get_richtext(document, './/div[@class="con_txt"]') return Company.insert(name=name, nature=nature, scale=scale, category=category, position=position, description=description, uniqueId=self.get_company_unique_id(company_url), companyUrl=company_url, site=self.task.site).execute()
def create_company(request): if request.method == 'GET': template = 'new_position/new_company.html' form = CompanyForm() return render( request, template, { 'form': form } ) elif request.method == 'POST': form = CompanyForm(request.POST) if form.is_valid(): company = Company( name = form.cleaned_data.get('name'), tag_name = form.cleaned_data.get('tag_name'), product = form.cleaned_data.get('product'), why_us = form.cleaned_data.get('why_us'), traction = form.cleaned_data.get('traction'), funding = form.cleaned_data.get('funding'), size = form.cleaned_data.get('size'), last_active = datetime.now(), market = form.cleaned_data.get('market'), company_link = form.cleaned_data.get('company_link'), ) company.save() return render( request, 'landing_pages/index.html', )
def delete(id): entity = Company.get(id) if(entity is None): raise ValueError("Company does not exists") else: entity.active = False Company.save(entity)
class Simulation(object): def __init__(self, aggregator, strategy, reserved_resources, sleep_time=0): self.aggregator = aggregator self.strategy = strategy # 2 year warm up, and 10 year run self.stats = Stats(24, 120) self.company = Company(20, reserved_resources, self.strategy, self.stats) self.sleep_time = sleep_time self.reserved_resources = reserved_resources print(sim_to_key(self)) def run(self): for t in range(self.stats.runs): self.stats.start_month() projects = [generate_project() for _ in range(sample.project_count())] self.company.decide_projects(projects) used_resources = self.company.workflow.work() self.stats.end_month(used_resources, self.company.workflow.average_workload()) if self.sleep_time > 0: sleep(self.sleep_time) #print(self.stats.monthly_report()) self.aggregator.add_result(self)
def test_company_repr_1(self): """Test __repr__ methond of company class""" example1 = Company("id", "name", "summary", "people", "city", "finorgs", "twitter", "website", "logo") self.assertEqual(example1.__repr__(), "<Company 'name'>")
def test_company_repr_3(self): """Test __repr__ methond of company class""" example3 = Company("id3", "name3", "summary3", "people3", "city3", "finorgs3", "twitter3", "website3", "logo3") self.assertEqual(example3.__repr__(), "<Company 'name3'>")
def test_company_model_1(self): """Test adding Company objects to and deleting Company objects from the db""" with app.test_request_context(): example1 = Company("id", "name", "summary", "people", "city", "finorgs", "twitter", "website", "logo") example2 = Company("id2", "name2", "summary2", "people2", "city2", "finorgs2", "twitter2", "website2", "logo2") companies1 = db.session.query(Company).all() db.session.add(example1) db.session.add(example2) db.session.commit() companies2 = db.session.query(Company).all() self.assertTrue(example1 in companies2) self.assertTrue(example2 in companies2) self.assertEqual(len(companies2), len(companies1) + 2) db.session.delete(example1) db.session.delete(example2) db.session.commit() companies3 = db.session.query(Company).all() self.assertTrue(example1 not in companies3) self.assertTrue(example2 not in companies3) self.assertEqual(len(companies1), len(companies2) - 2)
def create(self, validated_data): company = Company( name=validated_data['name'], alias=validated_data['name'].lower(), active=validated_data['active'] ) company.save() return company
def test_callingGetEmployeesCallsFilterOnEmployeeWithCompanyPkAsArgumentAndIsMangerEqualToTrue(self): employee = Employee() company = Company() company.pk = 8 employee.company = company with mock.patch.object(Employee, 'objects') as query_mock: employee.getEmployees() query_mock.filter.assert_called_with(company__pk=company.pk, is_manager=True)
def createDB(): print "Creating the initial db" Company.create_table() NewsSource.create_table() News.create_table() Price.create_table() OpinionAPI.create_table() OpinionAPIResponse.create_table()
def create(): try: Company.create_company(request.json) company = Company.find_by_id(request.json["company_id"]) return jsonify(company), 201 except Exception as e: return jsonify(error="couldn't add game"), 400
def test_callingGetAvailableSchemesCallsFilterOnCompetenceFieldCollectionToCompanyRelationWithCompanyPkAsArgument( self): company = Company() company.pk = 8 with mock.patch.object(CompetenceFieldCollectionToCompanyRelation, 'objects') as query_mock: company.getAvailableSchemes() query_mock.filter.assert_called_with(company__pk=company.pk)
def setup_tbl(): Person.create_table(fail_silently=True) Company.create_table(fail_silently=True) Payment.create_table(fail_silently=True) Tariff.create_table(fail_silently=True) Point.create_table(fail_silently=True) Bike.create_table(fail_silently=True) ReservationState.create_table(fail_silently=True) Reservation.create_table(fail_silently=True)
def delete_company(self, company): self.logger.debug("delete_company") self.logger.info("deleting company from database") Company.delete().where(Company.code == company.code).execute() self.logger.info("removing company from model") tree_iter = self.get_iter_from_selected_row(company.code) self.companies_model.remove(tree_iter)
def get_ftse_companies(url): companies = [] table = pd.read_html(url)[3] for index, row in table.iterrows(): company = Company(row['EPIC']) company.name = row['Company'] company.sector = row[ 'FTSE Industry Classification Benchmark sector[12]'] companies.append(company) return companies
def company_insert(self, request): if request.from_datastore: my_company = request else: my_company = Company(parent=main.PARENT_KEY, name=request.name, bio=request.bio, logo=request.logo, jobs=request.jobs, majors=request.majors, table=request.table, website=request.website) my_company.put() return my_company
def test_callingGetEmployeesCallsFilterOnEmployeeWithCompanyPkAsArgumentAndIsMangerEqualToTrue( self): employee = Employee() company = Company() company.pk = 8 employee.company = company with mock.patch.object(Employee, 'objects') as query_mock: employee.getEmployees() query_mock.filter.assert_called_with(company__pk=company.pk, is_manager=True)
def test_company_dictionary_2(self): """Test dictionary method of company class""" example1 = Company("id", "name", "summary", "people", "city", "finorgs", "twitter", "website", "logo") dict_rep = example1.dictionary() self.assertEqual(dict_rep['financial_orgs'], "finorgs") self.assertEqual(dict_rep['twitter'], "twitter") self.assertEqual(dict_rep['website'], "website") self.assertEqual(dict_rep['logo_url'], "logo")
def post(self): if(Company.exists(request.form['name'])): return jsonify(message='Company exists') else: company = Company() company.name = request.form['name'] company.tin = request.form['tin'] company.address = request.form['address'] db.session.add(company) db.session.commit() return jsonify(company.serialize)
def setUp(self): self.region = Region(name="test_region") self.region.save() self.company = Company(code="1001", name="test_company", region=self.region) self.company.save() self.company2 = Company(code="1002", name="test_company_2", region=self.region) self.company2.save() self.store = Store(code="10011001", name="test_store", company=self.company) self.store.save()
def test_company_repr_2(self): """Test __repr__ methond of company class""" example2 = Company("id1", "Wetpaint", ("Wetpaint is a technology platform company that uses its proprietary" + " state-of-the-art technology and expertise in social media to build" + " and monetize audiences for digital publishers."), "person1", "San Francisco", "Youniversity Ventures", "wetpaint", "http://wetpaint.com", "logo_url") self.assertEqual(example2.__repr__(), "<Company 'Wetpaint'>")
def test_company_dictionary_1(self): """Test dictionary method of company class""" example1 = Company("id", "name", "summary", "people", "city", "finorgs", "twitter", "website", "logo") dict_rep = example1.dictionary() self.assertEqual(dict_rep['company_id'], "id") self.assertEqual(dict_rep['name'], "name") self.assertEqual(dict_rep['summary'], "summary") self.assertEqual(dict_rep['people'], "people") self.assertEqual(dict_rep['city'], "city")
def delete(self, request, company_id): """ Handling DELETE method. args request: Request to View. company_id: id of company to be deleted. :return: HttpResponse with code 201 if company is deleted. """ company = Company() company.delete_company(company_id) data = serializers.serialize("json", Company.get_company()) return HttpResponse(data)
def scrape_companies_data( company_names: List[str], use_cache: bool = False, n: int = 2147483647, skip_companies: Set[str] = set() ) -> Tuple[List[Company], List[FailedCompanyError]]: errors = [] output_data = [] for i, company_name in enumerate(company_names): if i >= n: break if company_name in skip_companies: print(f'[INFO] Skip scraping {company_name}') continue try: company_id = company_name.replace(' ', '_').lower() company = Company(id=company_id) overview_url, reviews_url = scraper.get_glassdoor_urls( company_name) print('[INFO]', company_name, overview_url, reviews_url) if overview_url is None or reviews_url is None: raise Exception( f'Cannot find both URLs for "{company_name}": {overview_url} {reviews_url}' ) reviews_data = scraper.scrape(reviews_url, f'{company_name}_reviews.html', scraper.get_reviews_data) overview_data = scraper.scrape(overview_url, f'{company_name}_overview.html', scraper.get_overview_data) data = { 'name': company_name, 'overview_url': overview_url, 'reviews_url': reviews_url, 'linkedin_url': scraper.get_linkedin_url(company_name), } data.update(reviews_data) data.update(overview_data) company.update_data(data) output_data.append(company) except Exception as e: print(f'[FAIL] caught exception when parsing "{company_name}"') errors.append( FailedCompanyError( company_name=company_name, exception=e, )) return output_data, errors
def test_company_model_creation_4(self): result = Company.Company.query.all() d = Company.Company(*TestModels.test_company) db.session.add(d) db.session.commit() e = Company.Company(*TestModels.test_company) try: db.session.add(e) db.session.commit() assert (False) except Exception as e: db.session.rollback()
def input_companies(): ''' Creates the company models for each S&P 500 company''' yahoo_data = yahooDataRetriever.get_dict_500_companies(name_file) for i in yahoo_data.keys(): if len(Company.objects.filter(ticker = i)) > 0: continue c = Company() c.name = yahoo_data.get(i) c.ticker = i c.save() print "## Saved Company Object for: " + i
def test_company_model_creation_3(self): # first record c = Company.Company(*TestModels.test_company) db.session.add(c) # second record new_comp = list(TestModels.test_company) new_comp[0] = 200 d = Company.Company(*new_comp) db.session.add(d) db.session.commit() result = Company.Company.query.all() # two records to Company so far assert (len(result) == 2)
def post(self): newCompany = Company() newCompany.name = self.request.get("name") newCompany.cgpa = self.request.get("cgpa") newCompany.desc = self.request.get("desc") newCompany.sal = self.request.get("sal") newCompany.date = self.request.get("date") newCompany.lastDate = self.request.get("lastDate"); newCompany.put() self.redirect("/admin")
def process_company_file(data_file): workbook = xlrd.open_workbook(settings.MEDIA_ROOT+'/'+data_file.uploaded_file.name) worksheets = workbook.sheet_names() data_file.number_of_sheets = len(worksheets) data_file.save() for worksheet_name in worksheets: worksheet = workbook.sheet_by_name(worksheet_name) num_rows = worksheet.nrows - 1 curr_row = 0 while curr_row < num_rows: curr_row += 1 company_name = worksheet.cell_value(curr_row, 0) industry = worksheet.cell_value(curr_row, 1) isin = worksheet.cell_value(curr_row, 2) bse_code = worksheet.cell_value(curr_row, 3) industry, created = Industry.objects.get_or_create(industry_name=industry) industry.created_by = data_file.uploaded_by industry.save() try: company = Company.objects.get(isin_code=isin) except: try: company = Company.objects.get(company_name=company_name) except: company = Company() company.isin_code = isin if bse_code: company.BSE_code = int(bse_code) company.industry = industry company.company_name = company_name company.created_by = data_file.uploaded_by company.save() data_file.processing_completed = True data_file.save()
def save_instance_form(self, form, user): company = Company() company.nit = form.cleaned_data['nit'] company.name = form.cleaned_data['name'] company.location = form.cleaned_data['location'] company.website = form.cleaned_data['website'] company.email = form.cleaned_data['email'] company.phone = form.cleaned_data['phone'] company.save() user.company = company user.save()
def test_poly_base(self): db.create_all() dealer1 = Dealer(1) dealer10 = Dealer(10) company1 = Company(id=1, dealer=dealer1) assert company1.dealer == dealer1 assert company1.dealer_id == 1 company1.dealer_id = 10 assert company1.dealer == dealer10 assert repr(company1) == '<Company id: 1>'
def get(self, request, company_id=None): """ Handling GET method. :args request: Request to View. company_id: id of company to be returned. :return: HttpResponse with company fields and values by id. If company_id is 'None' returns all companies with their fields and values. """ if not company_id: data = serializers.serialize("json", Company.get_company()) return HttpResponse(data) company = Company.get_company(company_id) company = model_to_dict(company) return HttpResponse(json.dumps({"company": company}))
def create_company(): search_form = SearchForm() form = CreateCompanyForm() if request.method == "POST": comp_obj = Company() comp_obj.name = form.company_name.data db.session.add(comp_obj) db.session.commit() flash("Create company success") return render_template("create_company.html", search_form=search_form, form=form) else: if current_user.role != 3: flash("You have no permission") return redirect(url_for("main")) return render_template("create_company.html", search_form=search_form, form=form)
def get_Company(): ##### Create a Company and retrieve all Companies ###### if request.method == 'POST': body = request.get_json() if body is None: raise APIException( "You need to specify the request body as a json object", status_code=400) if 'email' not in body or body['email'] == '': raise APIException('You need to specify the email', status_code=400) if 'password' not in body: body['password'] = None if "address" not in body: raise APIException('You need to specify the address', status_code=400) if "company_name" not in body: raise APIException('You need to specify the company_name', status_code=400) if "company_description" not in body: raise APIException('You need to specify the description', status_code=400) _company = Company.query.filter_by(email=body['email']).first() if (_company is not None): raise APIException("Company already registered") company1 = Company( email=body['email'], password=body['password'], address=body['address'], company_name=body['company_name'], company_description=body['company_description'], ) db.session.add(company1) db.session.commit() return jsonify(company1.serialize()), 200 #### GET REQUEST METHOD ##### if request.method == 'GET': all_company = Company.query.all() all_company = list(map(lambda x: x.serialize(), all_company)) return jsonify(all_company), 200 return "Invalid Method", 404
def on_row_activated(self, treeview, path, column): self.logger.debug("on_row_activated") tree_iter = self.companies_model.get_iter(path) code = self.companies_model.get_value(tree_iter, self.CODE_COLUMN) company = Company.select().where(Company.code == code).get() column_id = column.get_sort_column_id() if column_id == self.REMOVE_COLUMN: dialog = Gtk.MessageDialog( self.window, 0, Gtk.MessageType.QUESTION, Gtk.ButtonsType.YES_NO, "Are you sure to remove the company?" ) if dialog.run() == Gtk.ResponseType.YES: self.delete_company(company) dialog.destroy() elif column_id == self.DELIVERY_NOTES_COLUMN: DeliveryNotesManagerWindow(company=company, parent=self.window) elif column_id == self.INVOICE_COLUMN: InvoicesWindow(company=company, parent=self.window) else: self.populate_form(company)
def post(self): company_id = self.request.get('company_id') company = Company.get_by_id(int(company_id)) try: name = self.request.get('name') company.name = name except: pass try: address = self.request.get('address') company.address = address except: pass try: city = self.request.get('city') company.city = city except: pass try: country = self.request.get('country') company.country = country except: pass try: email = self.request.get('email') company.email = email except: pass try: phone_number = int(self.request.get('phone_number')) company.phone_number = phone_number except: pass try: owner_name = self.request.get('owner_name') owner = Owner() owner.name = owner_name owner.company = company.key() owner.put() except: pass try: director_name = self.request.get('director_name') director = Director() director.name = director_name director.company = company.key() director.put() except: pass company.put() self.redirect("company/" + str(company_id))
def company_new(): company = Company() if request.method == 'POST': form = CompanyForm(request.form, obj=company) if form.validate(): try: form.populate_obj(company) validate_company(company) company.save() flash('Yes, empresa cadastrada com sucesso.', 'success') return redirect(url_for('companies.company_index')) except AttributeError as e: flash(str(e), 'warning') else: form = CompanyForm(obj=company) return render_template('companies/new.html', form=form)
def post(self): parser = reqparse.RequestParser() parser.add_argument('company_name', required=True) parser.add_argument('email', required=True) parser.add_argument('password', required=True) parser.add_argument('address_1') parser.add_argument('address_2') parser.add_argument('city') parser.add_argument('province') parser.add_argument('zipcode') parser.add_argument('country') parser.add_argument('phone') parser.add_argument('state', default='NOT_VERIFIED' if os.environ.get('TEST') is None else 'ACTIVE') company_args = parser.parse_args() if company_args.get('state') not in COMPANY_STATES: error_dict = { 'error_message': f'Invalid state {company_args.get("state")}', } LOGGER.error(error_dict) return error_dict, 400 try: return Company.create(**company_args) except IntegrityError as e: error_dict = { 'error_message': e, } LOGGER.error(error_dict) return error_dict, 400 except Exception as e: LOGGER.error(e)
def get(self, url_name): company = Company.all().filter('urlname = ', url_name).fetch(1) company = company[0] reviews = company.GetReviews(); template_values = { 'reviews': reviews, 'company_name': company.name } self.Render(GetTemplate('company_reviews'), template_values)
def get(self, url_name): company = Company.all().filter('urlname = ', url_name).fetch(1) company = company[0] reviews = db.GqlQuery("SELECT * FROM Review ORDER BY date DESC") template_values = { 'reviews': reviews, 'company_name': company.name } self.Render(GetTemplate('review'), template_values)
def process_item(self, item, spider): if type(spider) is not spiders.info.InfoSpider: return item # process item company, created = CompanyModel.get_or_create(site_id=item['site_id']) company.update(**item).where(CompanyModel.site_id == company.site_id).execute() return created
def get(self,company_id): company = Company.get_by_id(int(company_id)) template_values = { 'company' : company } template = jinja_environment.get_template('owners.html') self.response.out.write(template.render(template_values))
def load_company(session, company_data): company = Company( name=company_data.get('name'), website=company_data.get('website'), description=company_data.get('description'), size=company_data.get('size'), remote=company_data.get('allows_remote') ) company.techs = [Tech(name=tech_name) for tech_name in company_data.get('stack', [])] company.location = Location( city=company_data['location'].get('city'), country=company_data['location'].get('country'), postcode=company_data['location'].get('postcode') ) session.add(company) session.commit()
def view(request, pk, slug): company = Company.goof(pk=pk) if not company.slug == slug: return HttpResponseRedirect(company.get_absolute_url() ) kmz_url = reverse("kml-company", kwargs={"pk": company.pk}) return locals()
def get(self,company_id): # apparently, it must be company_id, not something else. company = Company.get_by_id(int(company_id)) template_values = { 'company' : company } template = jinja_environment.get_template('company.html') self.response.out.write(template.render(template_values))
def post(self): company = Company() try: name = self.request.get('name') company.name = name except: pass try: address = self.request.get('address') company.address = address except: pass try: city = self.request.get('city') company.city = city except: pass try: country = self.request.get('country') company.country = country except: pass try: email = self.request.get('email') company.email = email except: pass try: phone_number = int(self.request.get('phone_number')) company.phone_number = phone_number except: pass company.put() company_id = company.key().id() self.redirect("company/" + str(company_id))
def companies(): q = Company.all() q.order("datetime") companies = q.fetch(1000) duplicates = [] for company in companies: if company.name in duplicates: db.delete(company) else: duplicates.append(company.name)