-
Notifications
You must be signed in to change notification settings - Fork 0
/
db.py
69 lines (61 loc) · 2.7 KB
/
db.py
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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
from sqlalchemy import Column, ForeignKey, Integer, String, DateTime, create_engine
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import relationship, sessionmaker
import datetime
Base = declarative_base()
class Certificate(Base) :
__tablename__ = 'tbl_certificates'
id = Column(Integer, primary_key=True)
sha1 = Column(String(40), nullable=False)
not_before = Column(String(40), nullable=False)
not_after = Column(String(40), nullable=False)
subjectdn = Column(String(250), nullable=False)
issuerdn = Column(String(250), nullable=False)
serial = Column(String(50), nullable=False)
class APK(Base) :
__tablename__ = 'tbl_apks'
id = Column(Integer, primary_key=True)
app_name = Column(String(250), nullable=False)
package_name = Column(String(250), nullable=False)
version_code = Column(String(50), nullable=False)
displayed_version = Column(String(50), nullable=False)
local_package = Column(String(250), nullable=False)
koodous_url = Column(String(250), nullable=False)
sha256 = Column(String(64), nullable=False)
certificate_id = Column(Integer, ForeignKey('tbl_certificates.id'))
certificate = relationship(Certificate)
class Submission(Base) :
__tablename__ = 'tbl_submissions'
id = Column(Integer, primary_key=True)
submitted_to_username = Column(String(250), nullable=True)
submitted_to_title = Column(String(250), nullable=True)
submitted_to_id = Column(String(250), nullable=False)
forwarded_from_username = Column(String(250), nullable=True)
forwarded_from_firstname = Column(String(250), nullable=True)
forwarded_from_lastname = Column(String(250), nullable=True)
forwarded_from_id = Column(String(250), nullable=True)
submitted_by_username = Column(String(250), nullable=True)
submitted_by_firstname = Column(String(250), nullable=True)
submitted_by_lastname = Column(String(250), nullable=True)
submitted_by_id = Column(String(250), nullable=False)
message_text = Column(String(1000), nullable=True)
filename = Column(String(250), nullable=False)
apk_id = Column(Integer, ForeignKey('tbl_apks.id'))
apk = relationship(APK)
date = Column(DateTime, default=datetime.datetime.utcnow)
engine = create_engine('sqlite:///telescam.db')
Base.metadata.create_all(engine)
# Create empty certificate for corrupted APKs
Base.metadata.bind = engine
DBSession = sessionmaker(bind=engine)
session = DBSession()
new_certificate = session.query(Certificate).filter(Certificate.sha1 == '-').first()
if new_certificate == None :
new_certificate = Certificate(sha1='-',
not_before='-',
not_after='-',
subjectdn='-',
issuerdn='-',
serial='-')
session.add(new_certificate)
session.commit()