Skip to content

Add files via upload #1

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
rishabh214 wants to merge 1 commit into
base: temp
Choose a base branch
from
Open

Add files via upload #1

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
46 changes: 7 additions & 39 deletions generate.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,8 +10,6 @@
import gridfs
import shutil
import config


def retrieve_image_from_database(img_id):
# get an image based on it's 'certify code'
client = pymongo.MongoClient(config.MONGODB_URI, connectTimeoutMS=30000)
Expand All @@ -25,16 +23,12 @@ def retrieve_image_from_database(img_id):
return True
else:
return False


def add_to_image_database(image, img_id):
# Saves a given image to the database with the 'certify code' as its ID
client = pymongo.MongoClient(config.MONGODB_URI, connectTimeoutMS=30000)
database = client.certify
fs = gridfs.GridFS(database)
fs.put(image, _id=img_id)


def addToDatabase(listData):
client = pymongo.MongoClient(config.MONGODB_URI, connectTimeoutMS=30000)
#database certify
Expand All @@ -43,8 +37,6 @@ def addToDatabase(listData):
collection = database["userData"]
#inserting full list as many documents
collection.insert_many(listData)


class Create:
#defining variables
def __init__(self, template, spreadsheet, values, font, certify, ts):
Expand All @@ -54,79 +46,65 @@ def __init__(self, template, spreadsheet, values, font, certify, ts):
self.font = font
self.certify = certify
self.ts = ts

def resizeFontValues(self):
img = Image.open(self.template).convert('RGB')
width, height = img.size
heightNew = 500
ratio = height/heightNew

for coor in self.values:
self.values[coor] = [int((self.values[coor][0])*ratio), int((self.values[coor][1])*ratio) ]

self.font["size"]=int((self.font["size"])*ratio)

if self.certify["verify"] == True:
self.certify["coordinates"]=[ int((self.certify["coordinates"][0])*ratio),int((self.certify["coordinates"][1])*ratio) ]


def generate(self):
try:
data=pd.read_csv(self.spreadsheet)
except:
data=pd.read_excel(self.spreadsheet)
data.fillna("")

#updating data
print(self.values, self.font, self.certify)
self.resizeFontValues()
print(self.values, self.font, self.certify)
#getting max no. of rows
count = max(list(data.count()))

#update data with certify code if user checked it
datanew = self.addUnique(data,count)

#create directory for files
path = './static/temp/download/download_' + str(self.ts)
os.mkdir(path)
os.mkdir(path + '/images')

#generating certificate
self.imageProcessing(datanew,count)

#remove folder after creating zip
shutil.rmtree(path)

def imageProcessing(self,data,count):

for i in range(0,count):
#opening image and canvas
img = Image.open(self.template).convert('RGB')

#opening canvas
draw = ImageDraw.Draw(img)

#k for iterating through values{} index
for j,k in zip(self.values,range(0,len(self.values))):
fontSize = self.font['size']
if(j=='Certify'):
fontSize=int(fontSize/1.5)
#get font for that element
fontName = ImageFont.truetype(self.font['name'], fontSize)

#center align
h,w=draw.textsize(data[j][i])
if (j=='College' or j=='Position'):
draw.text([self.values[j][0]+31.5*3-(h/2)*3,self.values[j][1]], data[j][i],font=fontName, fill=self.font['color'])
if (h>65):
draw.text([self.values[j][0]+31.5*6-(h/2)*5.5,self.values[j][1]], data[j][i],font=fontName, fill=self.font['color'])
else:
draw.text([self.values[j][0]+31.5*3-(h/2)*3,self.values[j][1]], data[j][i],font=fontName, fill=self.font['color'])
elif(j=='Event'):
if (h>48):
draw.text([self.values[j][0]+23*6-(h/2)*6,self.values[j][1]], data[j][i],font=fontName, fill=self.font['color'])
else:
draw.text([self.values[j][0]+23*3-(h/2)*3,self.values[j][1]], data[j][i],font=fontName, fill=self.font['color'])
else:
draw.text([self.values[j][0]+24*6-(h/2)*6,self.values[j][1]], data[j][i],font=fontName, fill=self.font['color'])

#save image separately
#also save in database --> verified folder if verification is checked
if(self.certify['verify']):
Expand All @@ -140,43 +118,36 @@ def imageProcessing(self,data,count):
args=(database_image, data["Certify"][i]))
background_thread.start()




#save for downloading
folder = './static/temp/download/download_' + str(self.ts)
img.save(folder + '/images/' + str(i) + '.jpg')


#save dataframe as spreadsheet
data.to_csv(folder + '/data.csv', header=False, index=False)
shutil.make_archive(folder , 'zip', folder)

#generate unique id based on timestamp and salt
def unique(self):
#generating hashid, specifying salt and alphabets available
hashids = Hashids(salt = "certify to create certificates", alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvyxyz1234567890")

#generating unique id based on time stamp before and after decimal
uniqueId = hashids.encode(int(datetime.today().timestamp()))[-3:] + hashids.encode(int(str(datetime.today().timestamp()).split('.')[1]))

return uniqueId

#add unique id to data frame if certify verification is True
def addUnique(self,data,count):
if(self.certify["verify"]==True):
uIds=[]
#creating dataframe copy
data = data.copy()

#generate unique id for all values
for i in range(0,count):
uIds.append(self.unique())

#append column to dataframe
data['Certify'] = uIds

#creating empty list for database data
listData = []

#iterating through unique id array and 0 to count in same loop
for i,n in zip(uIds,range(0,count)):
#adding unique id to dictionary
Expand All @@ -192,8 +163,5 @@ def addUnique(self,data,count):
addToDatabase(listData)
#adding certify to value
self.values["Certify"]=self.certify["coordinates"]



#return new data frame
return data