-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathuploadname.py
More file actions
51 lines (36 loc) · 1.14 KB
/
Copy pathuploadname.py
File metadata and controls
51 lines (36 loc) · 1.14 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from sqlalchemy import create_engine
from sqlalchemy import Column, String, Integer
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
import os
from dotenv import load_dotenv, find_dotenv
load_dotenv(find_dotenv())
db_string = os.getenv('DB_PASS')
db = create_engine(db_string)
base = declarative_base()
class Affiliate_INFO(base):
__tablename__ = 'affiliate_info'
id = Column(Integer, primary_key=True)
name = Column(String(50))
department = Column(String(100))
college = Column(String(50))
email = Column(String(50))
interests = Column(String(1000))
domains = Column(String(1000))
Session = sessionmaker(db)
session = Session()
base.metadata.create_all(db)
# load data
from algorithms.parseaffiliate import parse_affiliate as pa
data = pa.getXLS()
for n in range(len(data)):
item = Affiliate_INFO(
name=data[n]['name'].strip(),
department=data[n]['department'],
college=data[n]['college'],
email=data[n]['email'],
interests=data[n]['interests'],
domains=data[n]['domains']
)
session.add(item)
session.commit()