From b63b50e072b6e55e4566bf6692bed793998e0a91 Mon Sep 17 00:00:00 2001
From: Anton Bolshakov <blshkv@users.noreply.github.com>
Date: Sat, 19 Jan 2013 02:23:24 +0000
Subject: [PATCH] set: python2 fixes, cleanup

---
 net-analyzer/set/Manifest        |  3 ++-
 net-analyzer/set/set-9999.ebuild | 26 ++++++++++++++++++++++++--
 2 files changed, 26 insertions(+), 3 deletions(-)

diff --git a/net-analyzer/set/Manifest b/net-analyzer/set/Manifest
index af8f37b95..d0c9976c8 100644
--- a/net-analyzer/set/Manifest
+++ b/net-analyzer/set/Manifest
@@ -1,3 +1,4 @@
 AUX set 41 SHA256 61fa527c1906be500005f480446cdfe04e45e48e9ce1a7415e90570ff9f74a27 SHA512 c517eef81963a486b3ef4e7a2d07c88ab3f6e85dc00ead372ee4a207a9e98b084da8dcb723762927322e159a3593bd8a8269613d63bb6ea07e381cdb682276f1 WHIRLPOOL 7b4e2eb82436b0397d8686e9401fe78d501855dd6f282f995baab524680b8b3585d9c05c49aa66ff318ab087aec5a319f2f2aeb839ce3a79850180afdee8d81d
 AUX set-ssmtp.patch 4332 SHA256 6ad5057bb384667c8c01f9130a704f104b95f4f82bf6b7f7a1af7c2dd671c319 SHA512 5ba0923b5e3fad0a47dd2030505ee2ed21c58d645b108e576f6bcd025fb94808429fb6a79838b44ea7ddacb22cf7ca2a98b10c1a979aef33fe22f699fc5a1975 WHIRLPOOL 9215ecbe2fb6372dc8c0717fecc7ca6704c49c02557552a7f2874301732f0712e4ab9cb44b5f79d1fe2436e730fa565cf1742cb2697ec8e39883441f5d7a25b0
-EBUILD set-9999.ebuild 2411 SHA256 25267d1ebd9d679fda4064cd96cf8ec199b9bd3e4917c26863a3ffd5dd7ad7ae SHA512 65db0f8021d536ef1dc3f82d10cffe61203f3b1249f4934fe1e5bd25832fee88345c768637660c6e38d2074a0c7d038ed0e81d9a91f81e8f8f078c792566f1d6 WHIRLPOOL 511b7766af2c4b834e3ea4dbcd1a08cbfdbd75633c2a966bb911f9006b092a8acd4d0d7ba770cfc253441ffa6f3992242635a6b729f637d6aa236180b867edc3
+AUX set_config.py 2747 SHA256 c3543df7460d3b403dadb5f834896876c385f04c26e9ec0b86d4cd3ec91ae024 SHA512 b5cb359aacdd800d8a2219797d1d31cd6d8bb476f726fa0f4d94e152906dd3cae54eb8e5afe920acb06b9c20b5002cc723b14ceb644f98378f246069f801c448 WHIRLPOOL 713f51e89d56bb48fa89832a3564accbe844535e34f753041bbbe070befb136e98ddc193507b035b92a7ee4ec7203928362b3420766783436f391f415c8b05d6
+EBUILD set-9999.ebuild 2884 SHA256 7f8a20df6f5a720ace8a358c49e0ab2a7cd41c5368bab83022e3b4b13ccf3387 SHA512 279886c465d630afc10c931d9edd99960f38f7f7cabde2cab13ee847ab02aa9b5e6e54dd86fa8cd1ca0de5e4a6050b2bfe4d35b8ead354b22ddb41b7aeeb311d WHIRLPOOL 64aa03a3570f004857c6f8a816e44252136904f148266fcc017465d77a7d59ed92a75f37cddfd0ec8fdb43f348a44f7768369ab9c5a5201331c1d5d1dcf6bbf0
diff --git a/net-analyzer/set/set-9999.ebuild b/net-analyzer/set/set-9999.ebuild
index cde5662ee..0d1e59b58 100644
--- a/net-analyzer/set/set-9999.ebuild
+++ b/net-analyzer/set/set-9999.ebuild
@@ -5,7 +5,10 @@
 EAPI="4"
 MY_P=${PN/set/social_engineering_toolkit}
 
-inherit git-2 multilib eutils
+PYTHON_DEPEND="2"
+
+inherit git-2 multilib eutils python
+
 SRC_URI=""
 EGIT_REPO_URI="https://github.com/trustedsec/social-engineer-toolkit.git"
 DESCRIPTION="A social engineering framework"
@@ -27,6 +30,7 @@ RDEPEND="virtual/jdk
 	net-analyzer/metasploit
 	dev-python/pexpect
 	net-misc/wget
+	dev-python/pymssql
 	dev-python/pyopenssl
 	ettercap? ( net-analyzer/ettercap )
 	|| ( mail-mta/ssmtp
@@ -36,7 +40,14 @@ DEPEND=""
 
 S=${WORKDIR}/${MY_P}
 
-src_compile() {
+pkg_setup() {
+    python_set_active_version 2
+    python_pkg_setup
+}
+
+src_prepare() {
+	python_convert_shebangs -r 2 .
+
 	if has_version mail-mta/ssmtp
 	then
 		epatch "${FILESDIR}"/set-ssmtp.patch
@@ -56,6 +67,8 @@ src_install() {
 	# We have global agreement
 	touch "${S}"/src/agreement4
 
+	cp "${FILESDIR}"/set_config.py "${S}"/config/
+
 	# should be as simple as copying everything into the target...
 	dodir /usr/$(get_libdir)/${PN}
 	cp -R "${S}"/* "${D}"/usr/$(get_libdir)/${PN} || die "Copy files failed"
@@ -80,8 +93,17 @@ src_install() {
 }
 
 pkg_postinst() {
+#	python_mod_optimize /usr/$(get_libdir)/set/src/core/set.py \
+#		/usr/$(get_libdir)/set/config/update_config.py \
+#		/usr/$(get_libdir)/set/src/phishing/smtp/client/smtp_web.py
+
 	elog "If you wish to update ${PN} simply run:"
 	elog
 	elog "emerge ${PF}"
 	elog
 }
+
+pkg_postrm() {
+	#workaround to remove all pyc and pyo files
+	rm -rf "/usr/$(get_libdir)/set"
+}