python模块被import时获取caller的信息进行初始化

主要目的是将重复定义的模块独立出来，便于其他脚本直接引用，减少反复定义的问题 module1:

import os,sys

import inspect

if __name__ == '__main__':

pass

else:

# python3

if sys.version>'3':

frm_list = inspect.stack()

for item in frm_list:

if item.index is None:

# print(item)

continue

else:

if __file__ in item.filename:

# print(item)

continue

else:

# print(item)

frm = item

mod = inspect.getmodule(frm[0])

t_file = str(os.path.abspath(mod.__file__))

# python2

else:

frm_list = inspect.stack()

mod = inspect.getmodule(frm_list[1][0])

t_file = str(os.path.abspath(mod.__file__))

t_list = [1,2,3]

t_dict = {"k1":"v1"}

module2:

import time

from module1 import *

# print(t_name)

print(id(t_file),t_file)

print(id(t_list),t_list)

print(id(t_dict),t_dict)

time.sleep(10)

分别运行module2，返回结果如下：

# python2 module2.py

(139742777226544, '/home/software/boofuzz/ft_local/module2.py')

(139742781893304, [1, 2, 3])

(139742781927784, {'k1': 'v1'})

python3 module2.py

139720418778448 /home/software/boofuzz/ft_local/module2.py

139720418165064 [1, 2, 3]

139720421308600 {'k1': 'v1'}

参考： https: […]

kazoo操作zookeeper基本功能的记录

任何数字化的信息在若干年后都会变成废品

# coding: utf-8

from kazoo.client import KazooClient, NoNodeError

zk = KazooClient(hosts="192.168.1.1:2181")

zk.start()

node_path_tpl = "/test1/resources/rsip@{ip}/rsport@{port}"

# 从文件读取信息

# ip:port

# with open("ip.list", "r") as f:

# node_list = f.readlines()

# 直接从zk获取信息

node_list = []

rsip_list = zk.get_children("/test1/resources")

for rsip in rsip_list:

_, host = rsip.split("@")

rsport_list = zk.get_children("/test1/resources" + "/" + rsip)

for rsport in rsport_list:

_, port = rsport.split("@")

node_list.append("{}:{}".format(host, port))

for item in node_list:

sub_node_list = []

ip, port = item.replace("\n", "").split(":")

node_path = node_path_tpl.format(ip=ip, port=port)

print(node_path)

# 获取所有子节点

# 方法1: 直接返回异常

# try:

# if zk.exists(node_path):

# sub_node_list = zk.get_children(node_path)

# else:

# print("(ne){}".format(node_path))

# except NoNodeError:

# continue

# 方法2: 通过zstat获取

if zk.exists(node_path):

_, state = zk.get(node_path)

if state.numChildren != 0:

sub_node_list = zk.get_children(node_path)

else:

zk.ensure_path(node_path)

# 子节点不为空，删除所有子节点

for sub_item in sub_node_list:

sub_node_path = node_path + "/" + sub_item

print("\t" + sub_node_path)

# zk.delete(sub_node_path, recursive=True)

zk.stop()

zk.close()

python的dict对象update操作的一些收集

默认的dict.update()在处理dict中包含list/dict对象时候会直接进行替换，不能取并集，在某些场景下不满足需求。针对这一问题，网上有很多讨论的内容，stackoverflow这个帖子比较经典，可以参考 https://stackoverflow.com/questions/3232943/update-value-of-a-nested-dictionary-of-varying […]

python的map和zip

def map(func: Callable[[_T1], _S], iter1: Iterable[_T1])

map(func, *iterables) --> map object

Make an iterator that computes the function using arguments from each of the iterables. Stops when the shortest iterable is exhausted.

在使用map的时候，如果func只有一个参数，那么iterables直接传递一个list是可行的，但是如果func默认不止一个参数，那么这里传给map的参数需要额外关注，不能直接使用list或者tuple的组合，我们先通过一张图，来直观的看一下map参数的格式：假设func1是我们定义的参数，有4个固定参数。正常调用func1直接 […]

python3生成gb18030编码字符

代码：

#!/use/bin/env python3

#coding=gb18030

fmt="'{}','{}'\n"

f = open('character_gb18030.sql','w', encoding='gb18030')

# 单字节部分（0x80 不在gb18030范围）

f.write('--echo 单字节部分\n')

for i in range(0x01, 0x80):

b = bytes([i])

code = b

f.write(fmt.format(code.hex().upper(), code.decode('gb18030')))

# 双字节部分1

f.write('--echo 双字节部分1\n')

for i in range(0xB0, 0xF7+1):

w1 = bytes([i])

for j in range(0xA1, 0xFE+1):

w2 = bytes([j])

code = w1+w2

f.write(fmt.format(code.hex().upper(), code.decode('gb18030')))

# 双字节部分2

f.write('--echo 双字节部分2\n')

for i in range(0x81, 0xA0+1):

w1 = bytes([i])

for j in range(0x40, 0xFE+1):

if j in [0x7f]:

continue

else:

w2 = bytes([j])

code = w1+w2

f.write(fmt.format(code.hex().upper(), code.decode('gb18030')))

# 双字节部分3

f.write('--echo 双字节部分3\n')

for i in range(0xAA, 0xFE+1):

w1 = bytes([i])

for j in range(0x40, 0xA0+1):

if j in [0x7f]:

continue

else:

w2 = bytes([j])

code = w1+w2

f.write(fmt.format(code.hex().upper(), code.decode('gb18030')))

# 四字节部分1

f.write('--echo 四字节部分1\n')

for i in range(0x81, 0x82+1):

dw1 = bytes([i])

for j in range(0x30, 0x39+1):

dw2 = bytes([j])

for m in range(0x81, 0xFE+1):

dw3 = bytes([m])

for n in range(0x30, 0x39+1):

dw4 = bytes([n])

code = dw1+dw2+dw3+dw4

f.write(fmt.format(code.hex().upper(), code.decode('gb18030')))

# 四字节部分2

f.write('--echo 四字节部分2\n')

for i in range(0x95, 0x98+1):

dw1 = bytes([i])

for j in range(0x30, 0x39+1):

dw2 = bytes([j])

for m in range(0x81, 0xFE+1):

dw3 = bytes([m])

for n in range(0x30, 0x39+1):

dw4 = bytes([n])

code = dw1+dw2+dw3+dw4

f.write(fmt.format(code.hex().upper(), code.decode('gb18030')))

用来给数据库导入用的，基本上可以用，如果你的数据库语法有差异，将fmt格式化内容修改一下就好了。

openpyxl操作excel的基本方法

官方指南：https://openpyxl.readthedocs.io/en/stable/styles.html 基本文件操作 #创建新的xlsx文件 import openpyxl #创建工作簿 wb = openpyxl.Workbook() #工作簿总是会带有至少一个worksheet，通过Workbook.active可以选中第一个sheet ws = wb.active #保存文件（ […]

scapy用法整理

导入scapy： from scapy.all import * 常用的查看命令： ls() - 不带参数则查看所有可持的layer，也可以指定一个layer名称从而查询此layer的详细详细，例如ls(TCP) lsc() - 查看当前scapy的所有功能列表 help() - 查看功能的帮助，例如help(hexdump)可以查看hexdump的作用和参数介绍 conf - 查看当前的配置信息 […]

Python内部类通过外部类属性值传递参数一例

为了解决两个线程类通过global 变量传递数据的方法在并发情况下全局变量被覆盖的问题，不得已将两个代理类用一个class来包装，试验了许久才得出这个简单粗暴的方法，留存以备将来参考

# coding: utf-8

import threading

import queue

import time

class MainClass:

# class attribute to be used in subClass

queueTest = None

queueStopThread = None

def __init__(self):

MainClass.queueTest = queue.Queue()

MainClass.queueStopThread = queue.Queue()

def __del__(self):

MainClass.queueTest = None

MainClass.queueStopThread = None

class SubClass1(threading.Thread):

def __init__(self):

threading.Thread.__init__(self)

def run(self):

while MainClass.queueStopThread.empty():

if not MainClass.queueTest.empty():

print("{} - {}".format(threading.currentThread().ident, MainClass.queueTest.get()))

print("SubClass1: queueStopThread: {}".format(MainClass.queueStopThread.get()))

class SubClass2(threading.Thread):

def __init__(self):

threading.Thread.__init__(self)

def run(self):

for i in range(1, 10):

MainClass.queueTest.put(i)

time.sleep(0.1)

MainClass.queueStopThread.put(True)

@staticmethod

def run_main():

p_subclass1 = MainClass.SubClass1()

p_subclass1.daemon = True

p_subclass1.start()

p_subclass2 = MainClass.SubClass2()

p_subclass2.daemon = True

p_subclass2.start()

p_subclass1.join()

del p_subclass1

del p_subclass2

if __name__ == "__main__":

tM1 = MainClass()

tM2 = MainClass()

t1 = threading.Thread(target=tM1.run_main())

t1.daemon = True

t2 = threading.Thread(target=tM2.run_main())

t2.daemon = True

t1.start()

t2.start()

main_class_test = MainClass()

main_class_test.run_main()

运行结果如下：

F:\Python\classInClass\venv\Scripts\python.exe F:/Python/classInClass/classInClass.py

14596 - 1

14596 - 2

14596 - 3

14596 - 4

14596 - 5

14596 - 6

14596 - 7

14596 - 8

14596 - 9

SubClass1: queueStopThread: True

11812 - 1

11812 - 2

11812 - 3

11812 - 4

11812 - 5

11812 - 6

11812 - 7

11812 - 8

11812 - 9

SubClass1: queueStopThread: True

11120 - 1

11120 - 2

11120 - 3

11120 - 4

11120 - 5

11120 - 6

11120 - 7

11120 - 8

11120 - 9

SubClass1: queueStopThread: True

Process finished with exit code 0

mowblog

一群散乱的记忆

标签： python

python模块被import时获取caller的信息进行初始化

kazoo操作zookeeper基本功能的记录

python的dict对象update操作的一些收集

python的map和zip

python3生成gb18030编码字符

openpyxl操作excel的基本方法

scapy用法整理

Python内部类通过外部类属性值传递参数一例

2024 年 4 月
一	二	三	四	五	六	日
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