async def search(poisk: str, response: Response):
users_collection = db.categories_items
list_of_index = []
list_of_cat = users_collection.find({"name": poisk}).distinct("id_name")
if list_of_cat:
for post in list_of_cat:
list_of_index.append(post)
else:
list_of_cat = users_collection.find({
"subtype": poisk
}).distinct("id_subtype")
if list_of_cat:
for post in list_of_cat:
list_of_index.append(post)
else:
list_of_cat = users_collection.find({
"lasttype": poisk
}).distinct("id_subtype")
if list_of_cat:
for post in list_of_cat:
list_of_index.append(post)
list_of_items = []
users_collection = db.items
for post in list_of_index:
current_items = users_collection.find({"categories": str(post)})
for p in current_items:
list_of_items.append(
models.items(id=p['_id'],
name=p['name'],
description=p['description'],
size=p['size'],
color=p['color'],
image=p['image'],
price=p['price'],
discount=p['discount'],
hit_sales=p['hit_sales'],
special_offer=p['special_offer'],
categories=p['categories'],
link_color=p['link_color']))
if not list_of_items:
items = users_collection.find({"name": {'$regex': poisk}})
for p in items:
list_of_items.append(
models.items(id=p['_id'],
name=p['name'],
description=p['description'],
size=p['size'],
color=p['color'],
image=p['image'],
price=p['price'],
discount=p['discount'],
hit_sales=p['hit_sales'],
special_offer=p['special_offer'],
categories=p['categories'],
link_color=p['link_color']))
return list_of_items
def add_item():
error = None
form = itemform()
if form.validate_on_submit():
try:
i = items(name=form.name.data, description=form.description.data)
db.session.add(i)
db.session.commit()
flash('Item added')
except Exception as e:
error = e
return render_template('add.html', form=form, error=error)
def sell(request, curr_user):
"Sell Page"
if request.method == 'POST':
form = SellForm(request.POST, instance=items(user=curr_user.user_obj))
if form.is_valid():
new_item = form.save()
# Redirect to the new item page
return HttpResponseRedirect('/item/?id=%d' % new_item.id)
else:
return render(request, 'sell.html', {'form': form})
else:
# Display the Form
return render(request, 'sell.html', {'form': SellForm()})
def add_to_cart(id):
already_in = items.query.filter_by(post_id=id).all()
if already_in:
return redirect(url_for('index'))
db_entry = items(post_id=id, email=current_user.email)
print(db_entry)
try:
db.session.add(db_entry)
db.session.commit()
return redirect(url_for('index'))
except SQLAlchemyError as e:
error = str(e.__dict__['orig'])
return error
def load_models(models,load_dir,prefix='best'):
'''
:param models: Either a single pytorch model, or a dict in {'model_name':model} format
:param load_dir: Directory to load from
:return: None
'''
if type(models) is not type(None):
for name,model in models.items():
load_file = os.path.join(load_dir, '{}_{}.pth'.fotmat(prefix,name))
model.load_state_dict(torch.load(load_file))
else:
load_file = os.path.join(load_dir,'{}_model.pth'.fotmat(prefix))
models.load_state_dict(torch.load(load_file))
return None
def save_models(models,save_dir,prefix='best'):
'''
:param models: Either a single pytorch model, or a dict in {'model_name':model} format
:param save_dir: Directory to save
:return: None
'''
if type(models) == dict:
for name,model in models.items():
save_file = os.path.join(save_dir, '{}_{}.pth'.fotmat(prefix,name))
torch.save(model.state_dict(), save_file)
else:
save_file = os.path.join(save_dir,'{}_model.pth'.fotmat(prefix))
torch.save(models.state_dict(),save_file)
return
def do_create(self, line):
"""Creates a new instance of BaseModel and saves it to a JSON file,
then prints the id of the newly created instance. """
if line == "":
print("** class name missing **")
return
if line not in models:
print("** class doesn't exist **")
return
for k, v in models.items():
if k == line:
value = v
new = value()
new.save()
print("{}".format(new.id))
async def get_items(id: int, response: Response):
users_collection = db.items
items = users_collection.find_one({"_id": id})
link_collection = db.link_color
if not items:
raise HTTPException(status_code=403)
response.status_code = status.HTTP_200_OK
return models.items(id=items['_id'],
name=items['name'],
description=items['description'],
size=items['size'],
color=items['color'],
image=items['image'],
price=items['price'],
discount=items['discount'],
hit_sales=items['hit_sales'],
special_offer=items['special_offer'],
categories=items['categories'],
link_color=link_collection.find_one(
{"_id": items['link_color']})['color_link'])
async def get_items(response: Response):
users_collection = db.items
link_collection = db.link_color
all_items = users_collection.find()
a_i = []
for post in all_items:
a_i.append(
models.items(id=post['_id'],
name=post['name'],
description=post['description'],
size=post['size'],
color=post['color'],
image=post['image'],
price=post['price'],
discount=post['discount'],
hit_sales=post['hit_sales'],
special_offer=post['special_offer'],
categories=post['categories'],
link_color=link_collection.find_one(
{"_id": post['link_color']})['color_link']))
response.status_code = status.HTTP_200_OK
return a_i
async def items_ind(index: list, response: Response):
items_collection = db.items
link_collection = db.link_color
list_of_items = []
for post in index:
items = items_collection.find_one({"_id": int(post)})
if items:
list_of_items.append(
models.items(id=items['_id'],
name=items['name'],
description=items['description'],
size=items['size'],
color=items['color'],
image=items['image'],
price=items['price'],
discount=items['discount'],
hit_sales=items['hit_sales'],
special_offer=items['special_offer'],
categories=items['categories'],
link_color=link_collection.find_one(
{"_id": items['link_color']})['color_link']))
return list_of_items
async def items_cat(id: str, response: Response):
list_of_items = []
items_collection = db.items
link_collection = db.link_color
items = items_collection.find({"categories": id}).sort("name", 1)
for post in items:
list_of_items.append(
models.items(id=post['_id'],
name=post['name'],
description=post['description'],
size=post['size'],
color=post['color'],
image=post['image'],
price=post['price'],
discount=post['discount'],
hit_sales=post['hit_sales'],
special_offer=post['special_offer'],
categories=post['categories'],
link_color=link_collection.find_one(
{"_id": post['link_color']})['color_link']))
return list_of_items
async def patch_items(patch_item: models.patch_items, response: Response):
users_collection = db.items
link_collection = db.link_color
items = users_collection.find_one({"_id": patch_item.id})
if not items:
raise HTTPException(status_code=403)
cat_collection = db.categories_items
cur_cat = cat_collection.find_one({
"name": patch_item.categories[0],
"subtype": patch_item.categories[1],
"lasttype": patch_item.categories[2]
})
if not cur_cat:
raise HTTPException(status_code=403)
current_item = users_collection.update_one({'_id': patch_item.id}, {
"$set": {
"name":
patch_item.name,
"description":
patch_item.description,
"size":
patch_item.size,
"color":
patch_item.color,
"image":
patch_item.image,
"price":
patch_item.price,
"discount":
patch_item.discount,
"hit_sales":
patch_item.hit_sales,
"special_offer":
patch_item.special_offer,
"categories": [
str(cur_cat['id_name']),
str(cur_cat['id_subtype']),
str(cur_cat['id_lasttype'])
],
"link_color":
patch_item.link_color
}
})
if not current_item:
raise HTTPException(status_code=403)
return models.items(id=patch_item.id,
name=patch_item.name,
description=patch_item.description,
size=patch_item.size,
color=patch_item.color,
image=patch_item.image,
price=patch_item.price,
discount=patch_item.discount,
hit_sales=patch_item.hit_sales,
special_offer=patch_item.special_offer,
categories=[
str(cur_cat['id_name']),
str(cur_cat['id_subtype']),
str(cur_cat['id_lasttype'])
],
link_color=patch_item.link_color)
def additem(cod,nom,fecha,cant):
produ = config.con.query(producto).filter_by(Nomb_producto=nom).first()
now = datetime.now()
nuevo = items(cod, produ.id_producto, now, fecha, "disponible", cant)
config.con.add(nuevo)
config.con.commit()
parser.add_argument('-s','--startdate',default = '2010-01-01 00:00:00',
help = 'specify a starting time stamp for relevant new data points')
parser.add_argument('-e','--enddate',default = '2020-01-01 00:00:00',
help = 'specify an ending time stamp for relevant new data points')
parser.add_argument('-c','--hydrocodes',
help = 'specify a hydrocode for all new data points being processed')
parser.add_argument('-d','--delimiter',default = ',',
help = 'specify a delimiter for outputting data points\nNote: use $\'\\t\' for tab character')
parser.add_argument('-w','--overwrite',action = 'store_true',
help = 'discard existing data if output file already exists')
args = parser.parse_args()
if not args.inputfiles:print('no input files provided!');quit()
elif not args.outputfile:print('no output file provided!');quit()
cfg,hcfg = convert.parse_config(args.configfile)
datapoints = collections.OrderedDict()
ifiles = args.inputfiles.split(',')
for ifile in ifiles:load_data(ifile,datapoints,hcfg,args)
risks = {}
for hydrocode in datapoints:
risks[hydrocode] = {}
for threat,model in models.items():
frisk = model.model(datapoints[hydrocode])
risks[hydrocode][threat] = frisk
save_data(args.outputfile,models,risks,hcfg,args)
if __name__ == '__main__':
for z in ['REG']:
# Construct the environment with three predictors and desirability functions
keys = ['A1', 'A2A', 'ERG']
A1 = utils.Predictor('output/env/RF_%s_CHEMBL226.pkg' % z, type=z)
A2A = utils.Predictor('output/env/RF_%s_CHEMBL251.pkg' % z, type=z)
ERG = utils.Predictor('output/env/RF_%s_CHEMBL240.pkg' % z, type=z)
mod1 = utils.ClippedScore(lower_x=4, upper_x=6.5)
mod2 = utils.ClippedScore(lower_x=9, upper_x=6.5)
mod3 = utils.ClippedScore(lower_x=7.5, upper_x=5)
objs = [A1, A2A, ERG]
models = {
'output/lstm_ligand.pkg':
'benchmark/FINE-TUNE_%s_%s.tsv' % (z, case),
'output/lstm_chembl.pkg':
'benchmark/PRE-TRAIN_%s_%s.tsv' % (z, case)
}
for case in ['OBJ1', 'OBJ3']:
if case == 'OBJ3':
mods = [mod1, mod1, mod3]
else:
mods = [mod2, mod1, mod3]
env = utils.Env(objs=objs, mods=mods, keys=keys)
for input, output in models.items():
df = pd.DataFrame()
df['Smiles'] = sampling(input, output)
scores = env(df['Smiles'], is_smiles=True)
df.to_csv(output, index=False, sep='\t')
def fit(models, X, Y):
result = {}
for key, model in models.items():
result[key] = model.fit(X, Y)
return result