52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
from sqlalchemy import create_engine, Column, Integer, String, LargeBinary
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.orm import Session
|
|
from pathlib import Path
|
|
import json
|
|
|
|
Base = declarative_base()
|
|
|
|
|
|
def db_config(path: Path) -> Session:
|
|
engine = create_engine(f"sqlite:///{path.resolve()}", native_datetime=True)
|
|
Base.metadata.create_all(engine)
|
|
session = Session(engine)
|
|
return session
|
|
|
|
|
|
class Disassembly(Base):
|
|
__tablename__ = "Disassembly"
|
|
id = Column(Integer, primary_key=True)
|
|
arch = Column(String, nullable=False)
|
|
checksum = Column(String, nullable=False)
|
|
count = Column(Integer, nullable=False)
|
|
mnemonic_rates = Column(String, nullable=False)
|
|
ret_rates = Column(String, nullable=False)
|
|
size = Column(Integer, nullable=False)
|
|
offset = Column(Integer, nullable=False)
|
|
opcodes = Column(String, nullable=False)
|
|
path = Column(String, nullable=False)
|
|
|
|
def __repr__(self):
|
|
return f"<Disassembly {json.dumps(self.values, indent=1)}>"
|
|
|
|
def __len__(self) -> int:
|
|
return self.count
|
|
|
|
def __lt__(self, other):
|
|
return len(self) < len(other)
|
|
|
|
@property
|
|
def values(self) -> dict:
|
|
values_dict = {
|
|
"id": self.id,
|
|
"arch": self.arch,
|
|
"count": self.count,
|
|
"mnemonic_rates": self.mnemonic_rates,
|
|
"ret_rates": self.ret_rates,
|
|
"size": self.size,
|
|
"offset": self.offset,
|
|
}
|
|
|
|
return values_dict
|